Исследователи представили FreqDepthKV — метод оптимизации инференса LLM, решающий проблему нехватки памяти при работе с длинными контекстами. Технология разделяет KV-состояния на низкочастотные общие компоненты и разреженные высокочастотные детали. Это позволяет значительно сократить объем кэша без потери точности при извлечении информации и выполнении многошаговых логических задач, что критично для производительности современных моделей.
Традиционные методы сжатия KV-кэша часто приводят к деградации ответов, так как удаляют важные для модели признаки, распределенные по разным слоям. FreqDepthKV использует частотный анализ для идентификации избыточных данных между соседними слоями. Метод позволяет динамически перераспределять ресурсы, сохраняя критически важную информацию для сложных запросов, требующих глубокого понимания контекста.
Применение данного подхода снижает требования к пропускной способности памяти и объему видеопамяти (VRAM) во время генерации. Это открывает возможности для запуска моделей с контекстным окном в сотни тысяч токенов на менее мощном оборудовании, сохраняя при этом качество генерации на уровне исходных моделей без сжатия.
Ключевые факты
- Метод FreqDepthKV фокусируется на факторизации KV-состояний соседних слоев для уменьшения избыточности.
- Технология разделяет данные на низкочастотные (общие) и высокочастотные (уникальные) компоненты.
- Сжатие направлено на решение проблемы ограничений памяти и пропускной способности при работе с длинными контекстами.
- Метод сохраняет способность модели к точному поиску (retrieval) и многошаговому рассуждению, которые обычно страдают при агрессивном сжатии кэша.