Получить доступ

Как LLM генерирует текст и что такое KV-кэш?

MiddleНейросети и трансформеры
Ещё спрашивают так:
  • Почему первый токен ответа появляется дольше остальных?
  • Что такое prefill и decode?
Коротко

Генерация авторегрессивна: токен за токеном, каждый новый зависит от всех предыдущих. Prefill — параллельная обработка всего промпта (определяет время до первого токена, TTFT). Decode — по одному токену за шаг (определяет скорость печати). KV-кэш хранит ключи и значения внимания уже обработанных токенов, чтобы не пересчитывать их на каждом шаге.

  • Без кэша каждый шаг пересчитывал бы весь контекст — квадратичная работа
  • Кэш занимает много памяти GPU и растёт с длиной контекста и числом параллельных запросов
  • Decode упирается в пропускную способность памяти, а не в вычисления

А ещё Серверы инференса (vLLM, TGI) используют paged attention и continuous batching

Так подсказку видит пользователь sotto на живом собеседовании — за пару секунд, пока вы думаете вслух.

Подробно

Два режима:

  • prefill — все токены промпта проходят через модель за один параллельный проход; длинный промпт → дольше ждать первый токен;
  • decode — модель выдаёт один токен, добавляет его в контекст, повторяет. Здесь каждый шаг читает из памяти все веса модели ради одного токена — узкое место в пропускной способности памяти.

KV-кэш. Для внимания каждому новому токену нужны K и V всех предыдущих. Их сохраняют после первого вычисления и на следующем шаге считают только для нового токена. Объём кэша = слои × головы × размерность × длина контекста × батч — гигабайты на длинных контекстах.

Оптимизации в продакшене:

  • continuous batching — новые запросы встраиваются в батч на лету, GPU не простаивает;
  • paged attention (vLLM) — кэш хранится страницами, без фрагментации памяти;
  • prefix caching / prompt caching — общий префикс (системный промпт) обрабатывается один раз;
  • speculative decoding — маленькая модель предлагает несколько токенов, большая проверяет их за один шаг;
  • квантизация весов и кэша.

Метрики: TTFT (время до первого токена) и токенов в секунду на генерации — это разные характеристики, и оптимизируются они разными средствами.

Что хотят услышать

  • Авторегрессивная генерация, prefill и decode.
  • Что хранит KV-кэш и почему он ускоряет.
  • Цена кэша и оптимизации серверов инференса.

Связанные вопросы

Закрытая бета

Подсказки на собеседовании — в реальном времени

sotto слышит вопрос интервьюера и показывает короткий ответ и план, как в блоке «Коротко». Оставьте контакт — пришлём доступ и промокод.

  • Промокод на 2 часа — сверх пробных 30 минут
  • Программа для Windows — скачать сразу, войти — через Яндекс ID или VK ID
  • Прямая связь с разработчиком: что улучшить — сделаем