Un transformador es una arquitectura de aprendizaje profundo desarrollada por Google y basada en el mecanismo de atención de múltiples cabezales , propuesto en un artículo de 2017 " Attention Is All You Need ". [1] El texto se convierte en representaciones numéricas llamadas tokens , y cada token se convierte en un vector buscando en una tabla de incrustación de palabras . [1] En cada capa, cada token se contextualiza dentro del alcance de la ventana de contexto con otros tokens (desenmascarados) a través de un mecanismo de atención paralelo de múltiples cabezales que permite amplificar la señal de los tokens clave y disminuir los tokens menos importantes. El artículo de Transformer, publicado en 2017, se basa en el mecanismo de atención basado en softmax propuesto por Bahdanau et al. en 2014 para traducción automática . [2] [3]
Los transformadores tienen la ventaja de no tener unidades recurrentes y, por lo tanto, requieren menos tiempo de entrenamiento que las arquitecturas neuronales recurrentes (RNN) anteriores, como la memoria a corto plazo (LSTM). [4] Se han adoptado ampliamente variaciones posteriores para entrenar modelos de lenguaje grandes (LLM) en conjuntos de datos grandes (de lenguaje), como el corpus de Wikipedia y Common Crawl . [5]
Los transformadores se utilizan actualmente en el procesamiento del lenguaje natural a gran escala , visión por computadora ( transformadores de visión ), audio, [6] procesamiento multimodal, robótica, [7] e incluso en el juego de ajedrez . [8] También ha llevado al desarrollo de sistemas pre-entrenados , como los transformadores generativos pre-entrenados (GPT) [9] y BERT [10] (Representaciones de codificador bidireccional de transformadores).
El modelado y la generación de secuencias se habían realizado durante muchos años con redes neuronales recurrentes simples. Un ejemplo temprano bien citado fue la red Elman (1990). En teoría, la información de un token puede propagarse arbitrariamente hacia abajo en la secuencia, pero en la práctica el problema del gradiente evanescente deja el estado del modelo al final de una larga oración sin información precisa y extraíble sobre los tokens anteriores.
Un componente clave del mecanismo de atención es incluir neuronas que multiplican las salidas de otras neuronas. Estas neuronas se denominaron unidades multiplicativas , [11] y las redes neuronales que utilizaban unidades multiplicativas se denominaron redes sigma-pi [12] o redes de orden superior , [13] pero se enfrentaban a una alta complejidad computacional. [4] Un avance clave fue LSTM (1995), [nota 1] que incorporó unidades multiplicativas en una red recurrente, así como otras innovaciones que evitaron el problema del gradiente de desaparición y permitieron un aprendizaje eficiente del modelado de secuencias largas. Se convirtió en la arquitectura estándar para el modelado de secuencias largas hasta la publicación de Transformers en 2017.
Sin embargo, LSTM no resolvió un problema general que suelen tener las redes recurrentes [nota 2] , que es que no pueden operar en paralelo con todos los tokens en una secuencia. Debe operar uno a la vez desde el primer token hasta el último. The Fast Weight Controller (1992) [14] fue uno de los primeros intentos de sortear esta dificultad. Utilizó la arquitectura de pesos rápidos (1987), [15] donde una red neuronal genera los pesos de otra red neuronal. Más tarde se demostró que era equivalente al Transformador lineal sin normalización. [16] [17] Schmidhuber utilizó la terminología "aprendizaje de focos internos de atención" en 1993, [18] y ahora afirma que fue un precursor de lo que ahora se conoce como el mecanismo de atención, pero Geoffrey Hinton cuestiona esta afirmación de prioridad. [19]
La idea de la transducción de secuencia codificador-decodificador se desarrolló a principios de la década de 2010 (consulte [20] [21] para artículos anteriores). Los artículos más comúnmente citados como los creadores que produjeron seq2seq son dos artículos de 2014. [20] [21]
(Sutskever et al, 2014) [21] , un modelo seq2seq de 380M de parámetros para traducción automática que utiliza dos memorias largas a corto plazo (LSTM). La arquitectura consta de dos partes. El codificador es un LSTM que toma una secuencia de tokens y la convierte en un vector. El decodificador es otro LSTM que convierte el vector en una secuencia de tokens. Otro modelo de parámetros 130M utilizó unidades recurrentes cerradas (GRU). [20] Investigaciones posteriores demostraron que los GRU no son ni mejores ni peores que los LSTM para seq2seq. [22] [23]
Estos primeros modelos seq2seq no tenían ningún mecanismo de atención y el vector de estado es accesible solo después de que se procesó la última palabra del texto fuente. Aunque en teoría dicho vector retiene la información sobre toda la oración original, en la práctica la información se conserva deficientemente, ya que la entrada es procesada secuencialmente por una red recurrente en un vector de salida de tamaño fijo , que luego fue procesado por otra red recurrente en una salida. Si la entrada es larga, entonces el vector de salida no podrá contener toda la información relevante y la calidad de la salida se degradará. Como prueba, revertir la oración de entrada mejoró la traducción seq2seq. [24]
(Bahdanau et al, 2014) [2] introdujo un mecanismo de atención en seq2seq para traducción automática para resolver el problema del cuello de botella, permitiendo que el modelo procese dependencias de larga distancia más fácilmente. Llamaron a su modelo RNNsearch , ya que "emula la búsqueda a través de una oración fuente durante la decodificación de una traducción".
(Luong et al, 2015) [25] compararon el rendimiento relativo de las arquitecturas de modelos de atención global (la de (Bahdanau et al, 2014)) y local (ventana deslizante) para la traducción automática, y descubrieron que una arquitectura de atención mixta tenía mayor calidad. que la atención global, mientras que el uso de una arquitectura de atención local redujo el tiempo de traducción.
En 2016, Google Translate se renovó a Google Neural Machine Translation , que sustituyó al modelo anterior basado en la traducción automática estadística . El nuevo modelo era un modelo seq2seq donde el codificador y el decodificador eran ambos de 8 capas de LSTM bidireccional. [26] Su desarrollo tardó nueve meses y logró un mayor nivel de rendimiento que el enfoque estadístico, que tardó diez años en desarrollarse. [27]
Los modelos Seq2seq con atención todavía sufrieron el mismo problema con las redes recurrentes, que es que son difíciles de paralelizar, lo que impidió que se aceleraran en las GPU. En 2016, la atención descomponible aplicó un mecanismo de atención a la red feedforward , que es fácil de paralelizar. [28] Uno de sus autores, Jakob Uszkoreit, sospechaba que la atención sin recurrencia es suficiente para la traducción de idiomas, de ahí el título "la atención es todo lo que necesitas". [29]
En 2017, el modelo de transformador codificador-decodificador original (tamaño 100 M) se propuso en el documento " La atención es todo lo que necesita ". En ese momento, el objetivo de la investigación era mejorar seq2seq para la traducción automática , eliminando su recurrencia para procesar todos los tokens en paralelo, pero preservando su mecanismo de atención al producto punto para mantener su rendimiento de procesamiento de texto. [1] Su paralelización fue un factor importante para su uso generalizado en grandes redes neuronales. [30]
Se utilizaron transformadores en muchos modelos que contribuyeron al auge de la IA .
En modelado de lenguaje, ELMo (2018) era un LSTM bidireccional que produce incrustaciones de palabras contextualizadas , mejorando la línea de investigación de bag of Words y word2vec . Le siguió BERT (2018), un modelo Transformer solo con codificador. [31] En octubre de 2019, Google comenzó a utilizar BERT para procesar consultas de búsqueda. [32] En 2020, Google Translate reemplazó el modelo anterior de codificador-RNN-decodificador RNN por un modelo de codificador-RNN-decodificador Transformer. [33]
A partir de 2018, la serie OpenAI GPT de Transformers solo decodificadores se convirtió en lo último en generación de lenguaje natural . En 2022, un chatbot basado en GPT-3, ChatGPT , se hizo inesperadamente popular, [34] provocando un auge en torno a los grandes modelos de lenguaje . [35] [36]
Desde 2020, los Transformers se han aplicado a más modalidades además del texto, incluido el transformador de visión , [37] el reconocimiento de voz, [38] la robótica, [39] y el multimodal [40] . El transformador de visión, a su vez, estimuló nuevos desarrollos en las redes neuronales convolucionales . [41] Los generadores de imágenes y videos como DALL-E (2021), Stable Diffusion 3 (2024) [42] y Sora (2024) se basaron en la arquitectura Transformer.
La arquitectura simple del transformador tuvo dificultades para converger. En el artículo original [1], los autores recomendaron utilizar el calentamiento con tasa de aprendizaje. Es decir, la tasa de aprendizaje debe aumentar linealmente desde 0 hasta el valor máximo para la primera parte del entrenamiento (generalmente se recomienda que sea el 2% del número total de pasos de entrenamiento), antes de disminuir nuevamente.
Un artículo de 2020 encontró que el uso de la normalización de capas antes (en lugar de después) de la atención de múltiples cabezas y las capas de avance estabiliza el entrenamiento, sin requerir un calentamiento de la tasa de aprendizaje. [43]
Los transformadores suelen someterse a un aprendizaje autosupervisado que implica un entrenamiento previo no supervisado seguido de un ajuste fino supervisado . El entrenamiento previo generalmente se realiza en un conjunto de datos más grande que el ajuste fino, debido a la disponibilidad limitada de datos de entrenamiento etiquetados . Las tareas de preentrenamiento y ajuste comúnmente incluyen:
El informe del transformador T5 [44] documenta una gran cantidad de tareas de preentrenamiento. Algunos ejemplos son:
Thank you <X> me to your party <Y> week.
-> <X> for inviting <Y> last <Z>
donde <Z>
significa "fin de salida".translate English to German: That is good.
-> Das ist gut.
.The course is jumping well.
-> not acceptable
.En general, existen cuatro clases de tareas de modelado de lenguaje: "enmascaradas", "autoregresivas" y "prefixLM". [45] [46] Estas clases son independientes de la elección arquitectónica. No es necesario utilizar un Transformer para entrenar modelos de lenguaje para realizar estas tareas. Sin embargo, a menudo se analizan en el contexto de Transformer.
En una tarea enmascarada, uno o más tokens están enmascarados y el modelo produciría una distribución de probabilidad que predice cuáles son los tokens enmascarados según el contexto. La función de pérdida para la tarea suele ser la suma de las perplejidades logarítmicas de los tokens enmascarados: y el modelo está entrenado para minimizar esta función de pérdida. BERT está entrenado para dos tareas, una de las cuales era la predicción de tokens enmascarados.
En una tarea autorregresiva, al principio toda la secuencia está enmascarada y el modelo produciría una distribución de probabilidad que predice cuál es el primer token. Luego se revela la primera ficha y el modelo predeciría la segunda ficha, y así sucesivamente. La función de pérdida de la tarea suele ser la misma.
En una tarea prefixLM, la secuencia se divide en dos partes. La primera parte se presenta como contexto y el modelo predeciría el primer token de la segunda parte. Luego eso se revelaría y el modelo predeciría el segundo token, y así sucesivamente. La función de pérdida de la tarea suele ser la misma.
Tenga en cuenta que "enmascarado" como en "modelado de lenguaje enmascarado" no es "enmascarado" como en "atención enmascarada" y "prefixLM" (modelado de lenguaje de prefijo) no es "prefixLM" (modelo de lenguaje de prefijo).
El transformador ha tenido un gran éxito en el procesamiento del lenguaje natural (PLN), por ejemplo en tareas de traducción automática y predicción de series temporales . Muchos modelos de lenguaje grandes como GPT-2 , GPT-3 , GPT-4 , Claude , BERT , XLNet, RoBERTa y ChatGPT demuestran la capacidad de los transformadores para realizar una amplia variedad de tareas relacionadas con la PNL y tienen el potencial de encontrar aplicaciones del mundo real. Estos pueden incluir:
Además de las aplicaciones de PNL, también ha tenido éxito en otros campos, como la visión por ordenador o las aplicaciones de plegado de proteínas (como AlphaFold ). También se aplicó para evaluar las posiciones del tablero de ajedrez. Utilizando únicamente la evaluación estática (es decir, sin búsqueda Minimax ) pudo alcanzar un Elo de 2895, colocándolo al nivel de gran maestro . [8]
El modelo transformador se ha implementado en marcos estándar de aprendizaje profundo como TensorFlow y PyTorch .
Transformers es una biblioteca producida por Hugging Face que proporciona arquitecturas basadas en transformadores y modelos previamente entrenados. [9]
Todos los transformadores tienen los mismos componentes primarios:
Como la arquitectura Transformer procesa de forma nativa datos numéricos, no texto, debe haber una traducción entre el texto y los tokens. Un token es un número entero que representa un carácter o un segmento corto de caracteres. En el lado de entrada, el texto de entrada se analiza en una secuencia de token. De manera similar, en el lado de la salida, los tokens de salida se analizan nuevamente en texto. El módulo que realiza la conversión entre secuencias de tokens y textos es un tokenizador .
El conjunto de todos los tokens es el vocabulario del tokenizador. Cuando se enfrentan a tokens fuera del vocabulario, normalmente se utiliza un token especial, escrito como "[UNK]" para "desconocido".
Algunos tokenizadores de uso común son la codificación de pares de bytes , WordPieza y SentencePieza.
Cada token se convierte en un vector de incrustación mediante una tabla de búsqueda . Luego, los vectores de codificación posicional se suman a sus respectivos vectores de incrustación de tokens, produciendo la secuencia de vectores de entrada.
Una codificación posicional es una representación vectorial de tamaño fijo de las posiciones relativas de los tokens dentro de una secuencia: proporciona al modelo del transformador información sobre dónde están las palabras en la secuencia de entrada. Sin codificación posicional, el modelo no podría procesar la secuencia de entrada como más que una bolsa de palabras , como por ejemplo, tanto " hombre muerde a perro " como "perro muerde a hombre" se procesarían exactamente de la misma manera.
La codificación posicional se define como una función del tipo , donde es un número entero par positivo . La codificación posicional completa, tal como se define en el artículo original, viene dada por la ecuación: donde .
Aquí hay un parámetro libre que debería ser significativamente mayor que el mayor que se ingresaría en la función de codificación posicional. En el artículo original, [1] los autores eligieron .
La función tiene una forma más simple cuando se escribe como una función compleja de tipo donde .
La razón principal por la que los autores eligieron esto como función de codificación posicional es que permite realizar cambios como transformaciones lineales: ¿ dónde está la distancia que se desea cambiar? Esto permite que el transformador tome cualquier posición codificada y encuentre la codificación de la posición n pasos adelante o n pasos atrás, mediante una multiplicación de matrices.
Al tomar una suma lineal, cualquier convolución también se puede implementar como transformaciones lineales: para cualquier constante . Esto permite que el transformador tome cualquier posición codificada y encuentre una suma lineal de las ubicaciones codificadas de sus vecinos. Esta suma de posiciones codificadas, cuando se introduce en el mecanismo de atención, crearía pesos de atención en sus vecinos, de forma muy parecida a lo que sucede en un modelo de lenguaje de red neuronal convolucional . En palabras del autor, "presumimos que permitiría al modelo aprender fácilmente a atender por posición relativa".
En implementaciones típicas, todas las operaciones se realizan con números reales, no con números complejos, pero dado que la multiplicación compleja se puede implementar como una multiplicación real de matrices 2 por 2 , esta es una mera diferencia de notación.
Al igual que los modelos seq2seq anteriores , el modelo de transformador original utilizaba una arquitectura codificador-decodificador . El codificador consta de capas de codificación que procesan los tokens de entrada de forma iterativa, una capa tras otra, mientras que el decodificador consta de capas de decodificación que procesan de forma iterativa la salida del codificador, así como los tokens de salida del decodificador hasta el momento.
La función de cada capa codificadora es generar representaciones de tokens contextualizadas, donde cada representación corresponde a un token que "mezcla" información de otros tokens de entrada mediante un mecanismo de autoatención. Cada capa de decodificador contiene dos subcapas de atención: (1) atención cruzada para incorporar la salida del codificador (representaciones de tokens de entrada contextualizadas) y (2) atención propia para "mezclar" información entre los tokens de entrada al decodificador (es decir, la tokens generados hasta ahora durante el tiempo de inferencia). [48] [49]
Tanto la capa codificadora como la capa decodificadora tienen una red neuronal de retroalimentación para el procesamiento adicional de las salidas y contienen conexiones residuales y pasos de normalización de capas. [49] Estas capas de retroalimentación contienen la mayoría de los parámetros en un modelo Transformer.
Los bloques de construcción del transformador son unidades de atención de productos escalados . Para cada unidad de atención, el modelo transformador aprende tres matrices de ponderaciones: ponderaciones de consulta , ponderaciones de clave y ponderaciones de valor . Para cada token , la representación del token de entrada se multiplica por cada una de las tres matrices de peso para producir un vector de consulta , un vector clave y un vector de valor . Los pesos de atención se calculan utilizando los vectores clave y de consulta: el peso de atención de un token a otro es el producto escalar entre y . Los pesos de atención se dividen por la raíz cuadrada de la dimensión de los vectores clave, que estabiliza los gradientes durante el entrenamiento, y se pasan a través de un softmax que normaliza los pesos. El hecho de que y sean matrices diferentes permite que la atención sea no simétrica: si el token atiende al token (es decir, es grande), esto no significa necesariamente que el token atenderá al token (es decir, podría ser pequeño). La salida de la unidad de atención para el token es la suma ponderada de los vectores de valor de todos los tokens, ponderada por la atención del token a cada token.
El cálculo de atención para todos los tokens se puede expresar como un cálculo matricial grande utilizando la función softmax , que es útil para el entrenamiento debido a las optimizaciones de operaciones matriciales computacionales que calculan rápidamente las operaciones matriciales. Las matrices , y se definen como las matrices donde las filas enésimas son vectores , y respectivamente. Entonces podemos representar la atención como
donde el softmax se aplica sobre cada una de las filas de la matriz.
Un conjunto de matrices se llama cabeza de atención y cada capa en un modelo de transformador tiene múltiples cabezas de atención. Si bien cada cabezal de atención atiende a los tokens que son relevantes para cada token, múltiples cabezales de atención permiten que el modelo haga esto para diferentes definiciones de "relevancia". Además, el campo de influencia que representa la relevancia puede dilatarse progresivamente en capas sucesivas. Muchas cabezas de atención transformadoras codifican relaciones de relevancia que son significativas para los humanos. Por ejemplo, algunas cabezas de atención pueden prestar atención principalmente a la siguiente palabra, mientras que otras prestan atención principalmente de los verbos a sus objetos directos. [50] Los cálculos para cada cabeza de atención se pueden realizar en paralelo , lo que permite un procesamiento rápido. Las salidas de la capa de atención se concatenan para pasar a las capas de la red neuronal de retroalimentación .
Concretamente, dejemos que las múltiples cabezas de atención estén indexadas por , entonces tenemos donde la matriz es la concatenación de incrustaciones de palabras, y las matrices son "matrices de proyección" propiedad de la cabeza de atención individual , y es una matriz de proyección final propiedad de todo el conjunto de múltiples cabezas de atención. cabeza de atención dirigida.
Puede que sea necesario eliminar los vínculos de atención entre algunos pares de palabras. Por ejemplo, el decodificador, al decodificar la posición del token , no debe tener acceso al token en la posición . Esto se puede lograr antes de la etapa softmax agregando una matriz de máscara que esté en las entradas donde se debe cortar el vínculo de atención y en otros lugares: Por ejemplo, la siguiente matriz de máscara se usa en el modelado autorregresivo: En palabras, significa que cada El token puede prestar atención a sí mismo y a todos los tokens anteriores a él, pero no a los posteriores.
Un codificador consta de una capa de incrustación, seguida de varias capas de codificador.
Cada capa de codificador consta de dos componentes principales: un mecanismo de autoatención y una capa de retroalimentación. Toma una entrada como una secuencia de vectores de entrada, aplica el mecanismo de autoatención para producir una secuencia intermedia de vectores y luego aplica la capa de avance para cada vector individualmente. Esquemáticamente tenemos:
donde significa "red de retroalimentación". Podemos escribirlo de manera más sucinta con la convención implícita de que se aplica a cada fila de la matriz individualmente.
Las capas del codificador están apiladas. La primera capa codificadora toma la secuencia de vectores de entrada de la capa de incrustación, produciendo una secuencia de vectores. Esta secuencia de vectores es procesada por el segundo codificador, y así sucesivamente. Luego, el decodificador utiliza la salida de la capa del codificador final.
A medida que el codificador procesa toda la entrada a la vez, cada token puede atender a todos los demás tokens (atención de todos a todos), por lo que no hay necesidad de enmascaramiento causal.
Un decodificador consta de una capa de incrustación, seguida de varias capas de decodificador, seguidas de una capa de desincrustación.
Cada decodificador consta de tres componentes principales: un mecanismo de autoatención enmascarado causalmente, un mecanismo de atención cruzada y una red neuronal de retroalimentación. El decodificador funciona de manera similar al codificador, pero se inserta un mecanismo de atención adicional que, en cambio, extrae información relevante de las codificaciones generadas por los codificadores. Este mecanismo también puede denominarse atención codificador-decodificador . [1] [49]
Al igual que el primer codificador, el primer decodificador toma como entrada información posicional e incrustaciones de la secuencia de salida, en lugar de codificaciones. El transformador no debe utilizar la salida actual o futura para predecir una salida, por lo que la secuencia de salida debe estar parcialmente enmascarada para evitar este flujo de información inverso. [1] Esto permite la generación de texto autorregresivo . Para la decodificación, la atención general es inapropiada, porque un token no puede atender a los tokens que aún no se han generado. Por lo tanto, el módulo de autoatención en el decodificador está causalmente enmascarado.
Por el contrario, el mecanismo de atención cruzada atiende a los vectores de salida del codificador, que se calcula antes de que el decodificador comience a decodificar. En consecuencia, no hay necesidad de enmascarar el mecanismo de atención cruzada.
Esquemáticamente tenemos: ¿Dónde está la matriz con filas que son los vectores de salida del codificador?
Al último decodificador le sigue una capa final de desincrustación, es decir, una transformación lineal seguida de un softmax , para producir las probabilidades de salida sobre el vocabulario. Luego, se muestrea uno de los tokens según la probabilidad y el decodificador se puede ejecutar nuevamente para producir el siguiente token, etc., generando texto de salida de forma autorregresiva.
Las capas de codificador y decodificador son como se describen. Se componen múltiples capas de codificador y decodificador en una configuración de codificador-decodificador, como se ilustra.
Los puntos finales de detalle son las conexiones residuales y la normalización de capas (LayerNorm o LN), que si bien conceptualmente son innecesarias, son necesarias para la estabilidad numérica y la convergencia. Es decir, la salida de cada subcapa es donde está la función implementada por la propia subcapa.
Cada capa de codificador contiene dos subcapas: la red de autoatención y la red de avance. Cada capa decodificadora contiene cuatro subcapas: la autoatención causalmente enmascarada, la red de avance, la atención cruzada y la red de avance.
La arquitectura Transformer al ser modular permite variaciones. Aquí se describen varias variaciones comunes. [51]
Un transformador "solo codificador" aplica el codificador para asignar un texto de entrada a una secuencia de vectores que representan el texto de entrada. Esto generalmente se usa para la incrustación de texto y el aprendizaje de representación para aplicaciones posteriores. BERT es solo codificador. Se ha descubierto que no son significativamente mejores que entrenar un transformador codificador-decodificador y luego tomar solo el codificador. [52]
Un transformador "sólo decodificador" no es literalmente sólo decodificador, ya que sin un codificador, el mecanismo de atención cruzada no tiene nada que atender. Por lo tanto, las capas de decodificador en un Transformer solo decodificador se componen de solo dos subcapas: la autoatención causalmente enmascarada, la red de alimentación directa. Esto generalmente se usa para generar texto y seguir instrucciones . Los modelos de la serie GPT son únicamente decodificadores.
Un transformador "codificador-decodificador" es generalmente el mismo que el transformador original, con 2 subcapas por capa codificadora y 4 subcapas por capa decodificadora, etc. Es posible que tengan mejoras arquitectónicas menores, como funciones de activación alternativas, cambio de ubicación de normalización, etc. Esto también se usa generalmente para la generación de texto y el seguimiento de instrucciones . Los modelos de la serie T5 son codificadores-decodificadores. [51]
Un "prefixLM" (modelo de lenguaje de prefijo) es una arquitectura de solo decodificador, pero con enmascaramiento de prefijo, que es diferente del enmascaramiento causal. Específicamente, tiene una máscara de la forma [51] donde las primeras columnas corresponden al "prefijo" y las columnas posteriores corresponden al texto generado autorregresivamente en función del prefijo. Se parecen a los modelos de codificador-decodificador, pero tienen menos "escasez". Estos modelos rara vez se utilizan, aunque se citan como posibilidades teóricas y comparaciones comparativas. [52]
El transformador original utiliza la función de activación ReLU . Se desarrollaron otras funciones de activación, como SwiGLU. [53]
Los transformadores pueden utilizar otros métodos de codificación posicional distintos de los sinusoidales. [54]
RoPE (incrustación posicional rotativa), [55] se explica mejor considerando una lista de vectores bidimensionales . Ahora elige algún ángulo . Entonces la codificación RoPE es equivalente, si escribimos los vectores bidimensionales como números complejos , entonces la codificación RoPE es simplemente una multiplicación por un ángulo: para una lista de vectores bidimensionales, un codificador RoPE se define por una secuencia de ángulos . Luego se aplica la codificación RoPE a cada par de coordenadas.
El beneficio de RoPE es que el producto escalar entre dos vectores depende únicamente de su ubicación relativa:
para cualquier número entero .
ALiBi (Atención con polarizaciones lineales) [56] no reemplaza el codificador posicional del transformador original. En cambio, es un codificador posicional adicional que se conecta directamente al mecanismo de atención. Específicamente , el mecanismo de atención de ALiBi es un número real ("escalar") y es la matriz de sesgo lineal definida por , en otras palabras, .
ALiBi permite el entrenamiento previo en ventanas de contexto cortas y luego el ajuste en ventanas de contexto más largas. Dado que se conecta directamente al mecanismo de atención, se puede combinar con cualquier codificador posicional que esté conectado al "fondo" de toda la red (que es donde se encuentra el codificador sinusoidal del transformador original, así como RoPE y muchos otros). Están localizados).
Codificaciones de posición relativa [57] es similar a ALiBi, pero más genérica: dónde está una matriz de Toeplitz , es decir, cuando sea .
En 2020, las dificultades con la convergencia del transformador original se resolvieron aplicando la normalización de capas antes (en lugar de después) de la atención multicabezal. Esto se llama transformador anterior a LN . [43]
FlashAttention [58] es un algoritmo que implementa el mecanismo de atención del transformador de manera eficiente en una GPU. Realiza multiplicaciones de matrices en bloques , de modo que cada bloque cabe dentro del caché de una GPU y, mediante una gestión cuidadosa de los bloques, minimiza la copia de datos entre cachés de GPU (ya que el movimiento de datos es lento).
Se desarrolló una versión mejorada, FlashAttention-2, [59] [60] [61] para satisfacer la creciente demanda de modelos de lenguaje capaces de manejar contextos de mayor longitud. Ofrece mejoras en la partición del trabajo y el paralelismo, lo que le permite alcanzar hasta 230 TFLOP/s en GPU A100 ( FP16 / BF16 ), un aumento de velocidad 2 veces mayor que el FlashAttention original.
Los avances clave en FlashAttention-2 incluyen la reducción de FLOP no matmul, paralelismo mejorado en la dimensión de longitud de secuencia, mejor partición del trabajo entre deformaciones de GPU y soporte agregado para dimensiones de cabeza de hasta 256 y atención de consultas múltiples (MQA) y agrupación. atención de consultas (GQA).
Los puntos de referencia revelaron que FlashAttention-2 es hasta 2 veces más rápido que FlashAttention y hasta 9 veces más rápido que una implementación de atención estándar en PyTorch. Los desarrollos futuros incluyen la optimización para nuevo hardware como las GPU H100 y nuevos tipos de datos como FP8.
La atención de consultas múltiples cambia el mecanismo de atención de múltiples cabezas. [62] Mientras que normalmente,
con Atención Multiconsulta, solo hay una , por lo tanto:
Esto tiene un efecto neutral sobre la calidad del modelo y la velocidad de entrenamiento, pero aumenta la velocidad de inferencia.
Cuando se utiliza un transformador autorregresivo para inferencia, como generar texto, el vector de consulta es diferente en cada paso, pero los vectores clave y de valor ya calculados son siempre los mismos. El método de almacenamiento en caché KV guarda los vectores de clave y valor calculados en cada bloque de atención, de modo que no se vuelvan a calcular en cada nuevo token. PagedAttention aplica paginación de memoria al almacenamiento en caché de KV. [63] [64] [65]
Si se utiliza un transformador con un mensaje integrado, como ["Usted es un agente de atención al cliente..."], entonces los vectores clave y de valor se pueden calcular para el mensaje y guardarse en el disco. El ahorro en computación es significativo cuando el modelo se utiliza para muchas interacciones breves, como en los chatbots en línea.
Los transformadores se utilizan en modelos de lenguaje grandes para la generación de secuencias autorregresivas: generar un flujo de texto, un token a la vez. Sin embargo, en la mayoría de los entornos, la decodificación de modelos de lenguaje está ligada a la memoria, lo que significa que tenemos potencia de cálculo disponible. La decodificación especulativa [66] [67] utiliza esta potencia de cálculo adicional calculando varios tokens en paralelo. De manera similar a la ejecución especulativa en las CPU, los tokens futuros se calculan simultáneamente, especulando sobre el valor de los tokens anteriores, y luego se descartan si resulta que la especulación fue incorrecta.
Específicamente, considere un modelo de transformador como GPT-3 con un tamaño de ventana de contexto de 512. Para generar una ventana de contexto completa de forma autorregresiva con decodificación codiciosa, se debe ejecutar 512 veces, generando cada vez un token . Sin embargo, si tuviéramos alguna suposición fundamentada sobre los valores de estos tokens, podríamos verificarlos todos en paralelo, en una ejecución del modelo, verificando que cada uno sea de hecho el token con la mayor probabilidad logarítmica en la -ésima salida. .
En la decodificación especulativa, se utiliza un modelo más pequeño o alguna otra heurística simple para generar algunos tokens especulativos que posteriormente son verificados por el modelo más grande. Por ejemplo, supongamos que un modelo pequeño generó cuatro tokens especulativos: . Estos tokens se ejecutan en el modelo más grande y solo se aceptan y . La misma ejecución del modelo grande ya generó un nuevo token para reemplazar y está completamente descartado. Luego, el proceso se repite (a partir del cuarto token) hasta que se generen todos los tokens.
Para la decodificación no codiciosa, se aplican ideas similares, excepto que los tokens especulativos se aceptan o rechazan estocásticamente, de una manera que garantiza que la distribución de salida final sea la misma que si no se utilizara la decodificación especulativa. [66] [68]
Entrenar arquitecturas basadas en transformadores puede resultar costoso, especialmente para entradas largas. [69] Las arquitecturas alternativas incluyen el Reformer (que reduce la carga computacional de a [69] ), o modelos como ETC/BigBird (que puede reducirla a ) [70] donde es la longitud de la secuencia. Esto se hace utilizando hash sensible a la localidad y capas reversibles. [71] [72]
Los transformadores ordinarios requieren un tamaño de memoria que sea cuadrático en el tamaño de la ventana contextual. Los transformadores sin atención [73] reducen esto a una dependencia lineal y al mismo tiempo conservan las ventajas de un transformador al vincular la clave al valor.
Long Range Arena (2020) [74] es un punto de referencia estándar para comparar el comportamiento de arquitecturas de transformadores en entradas largas.
Atención de características aleatorias (2021) [75] utiliza características aleatorias de Fourier : donde hay muestras independientes de la distribución normal . Esta elección de parámetros satisface , o En consecuencia, la atención unidireccional, con una consulta, se puede escribir como donde . Lo mismo ocurre con consultas múltiples y con atención de múltiples cabezas.
Esta aproximación se puede calcular en tiempo lineal, ya que primero podemos calcular la matriz y luego multiplicarla por la consulta. En esencia, hemos logrado obtener una versión más precisa de
Performer (2022) [76] utiliza la misma atención de características aleatorias, pero primero se muestrean de forma independiente de la distribución normal y luego se procesan según Gram-Schmidt .
Los transformadores también se pueden usar/adaptar para modalidades (entrada o salida) más allá del simple texto, generalmente encontrando una manera de "tokenizar" la modalidad.
Los transformadores de visión [37] adaptan el transformador a la visión por computadora descomponiendo las imágenes de entrada como una serie de parches, convirtiéndolas en vectores y tratándolas como tokens en un transformador estándar.
Conformer [38] y posteriormente Whisper [77] siguen el mismo patrón para el reconocimiento de voz , primero convirtiendo la señal de voz en un espectrograma , que luego se trata como una imagen, es decir, se divide en una serie de parches, se convierte en vectores y se trata como fichas en un transformador estándar.
Perceptores de Andrew Jaegle et al. (2021) [78] [79] pueden aprender de grandes cantidades de datos heterogéneos.
Con respecto a la salida de imágenes , Peebles et al introdujeron un transformador de difusión (DiT) que facilita el uso de la arquitectura del transformador para la producción de imágenes basada en difusión . [80] Además, Google lanzó un generador de imágenes centrado en transformadores llamado "Muse" basado en decodificación paralela y tecnología de transformadores generativos enmascarados. [81] (Los transformadores desempeñaron un papel menos central con las tecnologías de producción de imágenes anteriores, [82] aunque sigue siendo importante. [83] )
{{cite journal}}
: Citar diario requiere |journal=
( ayuda ){{cite news}}
: |archive-date=
/ |archive-url=
la marca de tiempo no coincide; 20 de marzo de 2024 sugerido ( ayuda ){{cite journal}}
: Citar diario requiere |journal=
( ayuda ){{cite web}}
: CS1 maint: multiple names: authors list (link)