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