Трансформер

Трансформер (англ. transformer) — архитектура глубоких нейронных сетей, основанная на механизме внимания без использования рекуррентных нейронных сетей (сокр. RNN). Самое большое преимущество трансформеров по сравнению с RNN заключается в их высокой эффективности в условиях параллелизации. Впервые модель трансформера была предложена в статье Attention is All You Need от разработчиков Google в 2017 году.

Архитектура трансформера

Архитектура трансформера

Устройство трансформера состоит из кодирующего и декодирующего компонентов. На вход принимается некая последовательность, создается ее векторное представление (англ. embedding), векторизованная последовательность поступает в кодирующий компонент, а затем декодирующий компонент получает на вход часть этой последовательности и выход кодирующего. В результате получается новая выходная последовательность.

Кодирующий компонент – это стек кодировщиков (англ. encoders), а декодирующий компонент – это стек декодировщиков (англ. decoders). Каждый кодировщики последовательно передает результат своей работы следующему кодировщику на вход. Декодировщики последовательно передают друг другу на вход результат работы вместе с результатом кодирующего компонента.

Ниже рассмотрим архитектуру кодировщика и декодировщика подробнее.

Архитектура трансформера-кодировщика

Архитектура трансформера-кодировщика

Рассмотрим последовательно шаг за шагом этапы кодировщика:

1. На вход поступает последовательность элементов [math]w_i[/math], по ней создается последовательность эмбедингов, где каждый [math]x_i[/math] это векторное представление элемента [math]w_i[/math].

2. Добавляются позиционные векторы [math]p_i[/math]: [math]h_i = x_i + p_i[/math], [math]H = (h_1,...,h_n)[/math]. Это необходимо для того, чтобы отобразить информацию о позиции элемента в исходной последовательности. Основное свойство позиционного кодирования — чем дальше два вектора будут стоять друг от друга в последовательности, тем больше между ними будет расстояние. Более подробное устройство позиционного кодирования будет рассмотрено ниже.

3. Полученный вектор [math]h_i[/math] подается на вход в блок многомерного самовнимания (англ. multi-headed self-attention). [math]h^j_i = Attn(W^j_q h_i, W^j_k H, W^j_v H)[/math], где [math]W[/math] - обучаемые матрицы, [math]W_q[/math] для запроса, [math]W_k[/math] для ключа, [math]W_v[/math] для значения. Подробное объяснения работы механизма self-attention будет разобрано ниже.

4. Затем необходима конкатенация, чтобы вернуться в исходную размерность: [math] h'_i = M H_j (h^j_i) = [h^1_i...h^J_i] [/math]

5. Добавим сквозные связи (англ. skip connection) - по факту просто добавление из входного вектора к выходному ([math]h'_i + h_i[/math]). После делаем нормировку уровня (англ. layer normalization): [math]h''_i = LN(h'_i + h_i; \mu_1, \sigma_1)[/math]. У нее два обучаемых параметра, для каждой размерности вектора вычисляется среднее и дисперсия.

6. Теперь добавим преобразование, которое будет обучаемым - полносвязную 2х-слойную нейронную сеть: [math] h'''_i = W_2 ReLU (W_1 h''_i + b_1) + b_2 [/math]

7. Повторим пункт 5 еще раз: добавим сквозную связь и нормировку уровня: [math]z_i = LN(h'''_i + h''_i; \mu_2, \sigma_2)[/math]

После, в кодирующем компоненте пункты кодировщика 3-7 повторяются еще несколько раз, преобразовывая друг за другом из контекста контекст. Тем самым мы обогащаем модель и увеличиваем в ней количество параметров.

Позиционное кодирование

Визуализация работы позиционного кодирования

Позиционное кодирование (англ. positional encoding) - позволяет модели получить информацию о порядке элементов в последовательности путем прибавления специальных меток к вектору входных элементов. Позиции элементов [math]i[/math] кодируются векторами [math]p_i[/math], [math]i = 1, 2, ..., n[/math], так, что чем больше [math]|i - j|[/math], тем больше [math]||p_i - p_j||[/math], и [math]n[/math] не ограничено: [math]p_{(i, s)} = \sin(i \cdot 10000^{\frac{-2s}{d_{model}}})[/math], [math]p_{(i, s + \frac{d}{2})} = \cos(i \cdot 10000^{\frac{-2s}{d_{model}}})[/math]

Self-attention

Self-Attention — разновидность механизма внимания, задачей которой является выявление закономерности между входными данными.

Будем для каждого элемента [math]x_i[/math] обучать три вектора:

• Запрос (query) [math]q_i = W_Q x_i[/math]
• Ключ (key) [math]k_i = W_K x_i[/math]
• Значение (value) [math]v_i = W_V x_i[/math]

Векторы [math]q_i[/math] и [math]k_i[/math] будем использовать, что посчитать важность элемента [math]x_j[/math] для элемента [math]x_i[/math]. Чтобы понять, насколько для пересчета вектора элемента [math]x_i[/math] важен элемент [math]x_j[/math] мы берем [math]k_j[/math] (вектор ключа элемента [math]x_j[/math]) и умножаем на [math]q_i[/math] (вектор запроса элемента [math]x_i[/math]). Так мы скалярно перемножаем вектор запроса на все векторы ключей, тем самым понимаем, насколько каждый входной элемент нам нужен, чтобы пересчитать вектор элемента [math]x_i[/math].

Далее считаем важность влияния элемента [math]x_j[/math] для кодирования элемента [math]x_i[/math]: [math]w_{ji}=\frac{ \exp(\frac{\langle q_i, k_j \rangle}{\sqrt{d}}) }{ \sum_{p=1}^n \exp(\frac{\langle q_i, k_p \rangle}{\sqrt{d}}) }[/math], где [math]d[/math] - размерность векторов [math]q_i[/math] и [math]k_j[/math], а [math]n[/math] - число элементов во входной последовательности.

Таким образом, новое представление элемента [math]x_i[/math] считаем как взвешенную сумму векторов значения: [math]z_i = Attn(W_q x_i, W_k X, W_v X) = \sum_{p=1}^n w_{p i} v_p[/math], где [math]X = (x_1, x_2, ..., x_n)[/math] - входные векторы. По итогу мы перемешиваем все входные векторы, чтобы получить новые векторы всех элементов, где каждый элемент зависит от всех входных элементов.

Multi-headed self-attention

Multi-headed self-attention — улучшенная модификация self-attention.

Слой внимания снабжается множеством «подпространств представлений» (англ. representation subspaces). Теперь у нас есть не один, а множество наборов матриц запроса/ключа/значения. Каждый из этих наборов создается случайным образом. Далее после обучения каждый набор используется для отображения входящих векторов в разных подпространствах представлений. Также появляется способность модели фокусироваться на разных аспектах входной информации.

То есть параллельно независимо несколько раз делаем attention. Потом результат каждого attention по элементам конкатенируем, затем сжимаем получившуюся матрицу и получаем для каждого элемента свой вектор той же размерности.

[math]с^j = Attn(W^j_q q, W^j_k X, W^j_v X)[/math], где [math]j = 1...J[/math], [math]J[/math] - количество разных моделей внимания, [math]X = (x_1, x_2, ..., x_n)[/math] - входные векторы, а [math]W[/math] - обучаемые матрицы.

Архитектура трансформера-декодировщика

Архитектура трансформера-декодировщика

На вход декодировщику подается выход кодировщика. Главное отличие архитектуры декодировщика заключается в том, что дополнительно имеется attention к вектору, который получен из последнего блока кодирующего компонента. Компонент декодировщика тоже многослойный и каждому блоку компонента на вход подается вектор именно с последнего блока кодирующего компонента. Разберем по порядку этапы работы декодировщика:

1. Для того, чтобы распараллелить декодировщик и уйти от рекуррентности, но тем не менее генерировать элементы друг за другом, используется прием маскирования данных из будущего. Идея в том, что мы запрещаем себе подглядывать в те элементы, которые еще не сгенерированы с учетом порядка. Когда генерируем элемент под номером [math]t[/math], имеем право смотреть только первые [math]t-1[/math] элементов: [math]h_t = y_{t-1} + p_t[/math]; [math]H_t=(h_1, ...,h_t)[/math]

2. Далее идет этап многомерного самовнимания: линейная нормализация и multi-headed self-attention. Особенность в том, что в attention ключи и значения применяются не ко всем векторам, а только к тем, значения которых уже синтезировали ([math]H_t[/math]): [math] h'_t = LN \circ M H_j \circ Attn(W^j_q h_t, W^j_k H_t, W^j_v H_t) [/math]

3. На следующем этапе мы делаем многомерное внимание на кодировку [math]Z[/math], результат работы компонента кодировщика: [math] h''_t = LN \circ M H_j \circ Attn(W^j_q h_t, W^j_k Z, W^j_v Z) [/math]

4. Линейная полносвязная сеть (по аналогии, как и в кодировщике): [math] y_t = LN \circ FNN(h''_t) [/math]

5. Линейный предсказывающий слой: [math] p(\tilde w|t) = SoftMax_{\tilde w}(W_y y_t + b_y) [/math], [math] W_y [/math] - обучаемая матрица весовых коэффицентов, [math] b_y [/math] - обучаемый вектор смещения

Последний этап выполняется только после того, когда повторились пункты 1-4 для всех декодировщиков. На выходе получаем вероятности классов, по факту решаем для каждого класса задачу многоклассовой классификации, для того, чтобы понять какие элементы лучше поставить на каждые позиции. В самом конце происходит генерация самих элементов: [math] {\tilde w}_t = arg max_{\tilde w} p(\tilde w|t) [/math], пока не сгенерируется слово обозначающее конец последовательности.