VKВлад КуликовОткрытые лекции
Современные LLM12 / 20

Модуль 12. Диффузионные LLM

Сначала попробуйте решить упражнение. Ответы, подробные решения и проектные ориентиры раскрываются отдельно и сохраняют своё состояние при переключении RU/EN.

Скачать файлы курса (.zip) 2588,1 КБ

Архив содержит файлы на обоих языках и вспомогательные материалы с сохранением структуры каталогов.

У1. Паритет плотной 8B-модели

Выведите максимальное целое SS, при котором dLLM с полным холстом в учебном сценарии быстрее AR-границы при пакете из одного запроса.

Показать ответ и решение

Решение. AR генерирует один токен за проход, поэтому его верхняя оценка равна 1/0.004776119=209.3751/0.004776119=209.375 ток/с. dLLM за SS проходов раскрывает холст из 1024 токенов; её средняя скорость в этой модели равна 1024/(S⋅0.016549495)1024/(S\cdot0.016549495). Для преимущества требуется

1024S⋅0.016549495>209.375.\frac{1024}{S\cdot0.016549495}>209.375.

Отсюда S<295.52S<295.52. Максимальное целое — 295; при 296 dLLM уже не быстрее в этой арифметической модели.

У2. Плотный сценарий и реальная DiffusionGemma

Для условной плотной 8B-модели восстановите нижние границы блока 256 при bf16 и четырёхбитных весах. Затем объясните, почему их нельзя перенести на DiffusionGemma.

Показать ответ и решение

Решение. Плотная 8B-модель: объём bf16-весов 16 GB и нижняя граница чтения 4.776 мс; четырёхбитный объём весов 4 GB и граница чтения 1.194 мс. Арифметическая граница блока — 4.137 мс, поэтому bf16-вариант ограничен памятью, а четырёхбитный — вычислениями. DiffusionGemma — разреженная MoE-модель 25.2B/3.8B: разные позиции выбирают разные наборы экспертов, численная точность тензоров неоднородна, фактический трафик определяется объединением выбранных экспертов и средой исполнения. Её результат на H100 измерен в FP8, а NVFP4 относится к Blackwell.

У3. Паспорт скорости

Сравните три заявления: Mercury 2 — 1009 ток/с на Blackwell; Gemini Diffusion — 1479 ток/с без учёта 0.84 с фиксированных накладных расходов; Seed Diffusion — 2146 ток/с на H20 в задачах программирования. Можно ли назвать победителя?

Показать ответ и решение

Решение. Нет. Оборудование, нагрузка, качество модели и граница измерения различаются. Особенно показателен Gemini: для NN выходных токенов заявленная модель даёт приближённое время 0.84+N/14790.84+N/1479 с при условном переносе опубликованных средних на эту длину, так что 1479 ток/с не является end-to-end скоростью короткого ответа. Сначала нужно выровнять качество на общей задаче, распределение входных/выходных длин, размер пакета, семплер, точность и границы измерения задержки; затем отдельно сравнить время до первого токена, полное время и скорость установившейся генерации.

У4. Маска внимания блочной диффузии

Постройте маску для трёх завершённых токенов и текущего блока длины четыре. Покажите, какие строки можно кэшировать.

Показать ответ и решение

Решение. В порядке строк/столбцов P1 P2 P3 B1 B2 B3 B4 маска разрешённых связей равна

text
P1  1 0 0 | 0 0 0 0
P2  1 1 0 | 0 0 0 0
P3  1 1 1 | 0 0 0 0
    -------+--------
B1  1 1 1 | 1 1 1 1
B2  1 1 1 | 1 1 1 1
B3  1 1 1 | 1 1 1 1
B4  1 1 1 | 1 1 1 1

В выбранной маске префикс каузален и не зависит от текущего блока, поэтому K/V первых трёх токенов можно сохранять между шагами демаскирования. Каждая строка активного блока читает весь префикс и весь блок; при изменении масок его четыре состояния нужно пересчитать.

У5 [проект]

Обучите маленький символьный Transformer с masked diffusion. Сравните случайное раскрытие, раскрытие по уверенности и раскрытие по энтропии при одинаковом числе вызовов модели.

Демо и критерии проекта

Демо к этому проектному упражнению: русскоязычный демонстрационный ноутбук модуля 12, раздел «Демо 1». Он показывает компактный воспроизводимый пример на малых данных, а не готовое решение полномасштабного проекта.

У6 [проект]

Начните с журнала работы в демо: меняйте длину блока и число шагов раскрытия, учитывая отдельный проход по готовому блоку. Сопоставьте число вызовов и обработанных позиций; для условной плотной модели рассчитайте байты весов при допущении одного полного чтения на вызов. Затем измерьте качество и задержку маленькой обученной модели на CPU, указав число потоков, прогрев и повторения. Объясните, почему расчётные байты не являются измеренным трафиком. Аппаратное профилирование обращений к памяти — необязательное расширение.

Демо и критерии проекта

Демо к этому проектному упражнению: русскоязычный демонстрационный ноутбук модуля 12, раздел «Демо 2». Он показывает компактный воспроизводимый пример на малых данных, а не готовое решение полномасштабного проекта.