Исследователи представили DominoTree — новый метод ускорения инференса больших языковых моделей, объединяющий преимущества блочной диффузии и древовидного поиска. В отличие от стандартных подходов, Domino использует GRU-коррекцию для учета зависимостей между токенами, что позволяет создавать более точные последовательности для параллельной верификации. Метод значительно повышает эффективность генерации, сохраняя при этом качество выходных данных модели.

Спекулятивное декодирование традиционно опирается на «черновые» модели (drafters), которые генерируют кандидатов для быстрой проверки основной моделью. Однако классические методы часто сталкиваются с ограничением: либо они генерируют токены независимо, теряя контекст, либо требуют слишком больших вычислительных затрат на построение дерева вариантов. DominoTree решает эту проблему за счет введения причинно-следственной коррекции на уровне GRU-блока, что позволяет лучше предсказывать вероятные токены в рамках одной итерации.

Использование древовидной структуры позволяет модели рассматривать несколько путей генерации одновременно. Благодаря интеграции Domino, система эффективно балансирует между глубиной поиска и точностью предсказаний. Это снижает количество обращений к основной модели, которая является наиболее «тяжелым» компонентом системы, и сокращает общую задержку при генерации длинных текстов.

Ключевые факты

  • DominoTree объединяет блочную диффузию с поиском по дереву для оптимизации спекулятивного декодирования.
  • Внедрение GRU-коррекции позволяет учитывать зависимости между токенами внутри одного блока генерации.
  • Метод минимизирует количество вызовов основной модели за счет повышения точности черновиков.
  • Подход позволяет эффективно масштабировать параллельную верификацию токенов без потери качества ответов.