HyperAIHyperAI

Command Palette

Search for a command to run...

Reformer : Le Transformer efficace

Nikita Kitaev Łukasz Kaiser Anselm Levskaya

Résumé

Les modèles Transformer de grande taille atteignent régulièrement des résultats de pointe sur de nombreuses tâches, mais leur entraînement peut s'avérer prohibitivement coûteux, en particulier pour des séquences longues. Nous introduisons deux techniques visant à améliorer l'efficacité des Transformers. Premièrement, nous remplaçons l'attention par produit scalaire par une attention basée sur le hachage sensible à la localité, ce qui réduit sa complexité de O(L2L^2L2) à O(LlogLL\log LLlogL), où LLL désigne la longueur de la séquence. En outre, nous utilisons des couches résiduelles réversibles au lieu des résiduelles classiques, ce qui permet de stocker les activations une seule fois pendant l'entraînement, plutôt que NNN fois, où NNN est le nombre de couches. Le modèle résultant, appelé Reformer, atteint des performances comparables à celles des modèles Transformer tout en étant bien plus efficace en mémoire et nettement plus rapide sur des séquences longues.


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