HyperAIHyperAI

Command Palette

Search for a command to run...

Qualité du Transformer en temps linéaire

Weizhe Hua Zihang Dai Hanxiao Liu Quoc V. Le

Résumé

Nous reprenons l'analyse des choix architecturaux des Transformers, et proposons des méthodes visant à surmonter leurs limites dans la gestion des séquences longues. Tout d'abord, nous introduisons une couche simple appelée unité d'attention à portes, qui permet d'utiliser une attention à un seul head plus faible avec une perte de qualité minimale. Nous proposons ensuite une méthode d'approximation linéaire complémentaire à cette nouvelle couche, qui est compatible avec les accélérateurs matériels et offre une qualité très compétitive. Le modèle résultant, nommé FLASH, atteint un perplexité équivalente à celle des Transformers améliorés, tant pour des longueurs de contexte courtes (512) que longues (8K), tout en offrant des accélérations d'apprentissage allant jusqu'à 4,9× sur Wiki-40B et 12,1× sur PG-19 pour le modèle de langage auto-régressif, ainsi que 4,8× sur C4 pour le modèle de langage masqué.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp