По порядку Middle Attention и трансформеры обновлено

Прямой проход decoder-only модели

Вопрос

Расставьте шаги прямого прохода GPT-подобной модели для предсказания следующего токена.

Попробуйте сами

Расставьте шаги по порядку стрелками.

  1. 1
    Токенизировать текст в id
  2. 2
    Финальная нормализация
  3. 3
    Взять эмбеддинги токенов (позиции добавятся через RoPE в attention)
  4. 4
    Линейная проекция в логиты словаря
  5. 5
    Softmax по логитам последней позиции и выбор следующего токена
  6. 6
    Пройти N блоков: нормализация, маскированный self-attention, residual, нормализация, MLP, residual
Показать ответ и разбор

Коротко

Так устроены современные LLM с pre-norm. В обучении логиты считаются для всех позиций сразу, и потеря это кросс-энтропия со сдвинутыми на один токен целями.

Правильный порядок

  1. 1
    Токенизировать текст в id
  2. 2
    Взять эмбеддинги токенов (позиции добавятся через RoPE в attention)
  3. 3
    Пройти N блоков: нормализация, маскированный self-attention, residual, нормализация, MLP, residual
  4. 4
    Финальная нормализация
  5. 5
    Линейная проекция в логиты словаря
  6. 6
    Softmax по логитам последней позиции и выбор следующего токена

Разбор

Так устроены современные LLM с pre-norm. В обучении логиты считаются для всех позиций сразу, и потеря это кросс-энтропия со сдвинутыми на один токен целями.

Ответьте сами, а не прочитайте

В тренажёре вопрос попадёт в повторения и вернётся, когда начнёт забываться. Плюс билеты с зачётом: 20 вопросов, не больше 2 ошибок.

Термины по теме