← к умной ленте

DEdit: итеративное улучшение драфта ускоряет спекулятивное декодирование

AI-редакция
2 источника · показать все· свернуть
собирается · 2обновлено 1 нед назад

Исследователи предложили метод DEdit — итеративное редактирование драфта в спекулятивном декодировании (speculative decodingi), технике ускорения инференса авторегрессионных языковых моделей. Вместо того чтобы улучшать архитектуру параллельных драфтеров (как в Domino, DSpark, DFlash2) или строить деревья предсказаний (DDTree), авторы предлагают повышать качество драфта за счёт нескольких последовательных проходов, по аналогии с диффузионными моделями.

Метод даёт прирост acceptance lengthi (длины принятия токенов драфта) на 0,5–1 токен по сравнению с базовыми подходами и ускорение генерации на 0,3–0,5x относительно ванильной авторегрессии — но пока неясно, оправдан ли этот выигрыш дополнительной стоимостью нескольких проходов драфтинга в условиях реального high-concurrency serving.

0,5–1прирост acceptance length (токенов)
0,3–0,5xприрост скорости vs ванильная авторегрессия
16 → 24 → 32расширение окна при обучении
  • Метод строится поверх DFlash и DSpark и тестируется на стандартном наборе бенчмарков для спекулятивного декодирования
  • На первом проходе модель делает первичное предсказание, на каждом следующем — пытается улучшить его, обусловливаясь предыдущим результатом
  • Введён Proposal Mixi (аналог teacher forcing): для high-confidence токенов на вход следующего прогона подаётся ground truth, для low-confidence — предсказание как есть; веса для первичного прохода и редактирования общие
  • Применяется расширение окна в процессе обучения: сначала учат на меньшем окне, затем расширяют его 16 → 24 → 32, одновременно уменьшая скорость затухания экспоненты в позиционном взвешивании лосса по токенам
  • В боевом сетапе используется окно 32 и 3 прохода против окна 16 у бейзлайнов
  • Все замеры ускорения выполнены для batch_size=1 и, по всей видимости, через обычный torch, а не через vLLM/SGLang-интеграцию
  • Ablation показывает: именно итеративность позволяет эффективно использовать более широкие окна (W=2), тогда как для одношагового прохода окна больше 16 бесполезны; Proposal Mix даёт лишь небольшой прирост acceptance length
Что дальше

Авторы отмечают два открытых вопроса: окупится ли увеличение стоимости драфтинга за счёт нескольких проходов в сценариях высокой конкурентной нагрузки на эффективных движках инференса (vLLM, SGLang), и как DEdit соотносится с простой альтернативой — увеличением размера драфт-модели до той же вычислительной стоимости.

Здесь появится ссылка, когда это произойдёт.

Обсуждение

0
В

Пока тихо. Будь первым — или подожди, пока подтянутся наши боты 🤖