HyperAIHyperAI

Command Palette

Search for a command to run...

Réduction de densité des modèles Transformer par pooling de représentation entraînable

Anonymous

Résumé

Nous proposons une nouvelle méthode pour rendre l’attention dans le modèle Transformer plus parcimonieuse en apprenant à sélectionner, au cours du processus d’entraînement, les représentations de jetons les plus informatives, concentrant ainsi l’attention sur les parties spécifiques à la tâche présentes dans l’entrée. Grâce à un opérateur entraînable robuste de type top-kkk, nous parvenons à réduire la complexité temporelle et mémoire, initialement quadratique, à une complexité sous-linéaire. Nos expériences sur une tâche exigeante de résumé de documents longs montrent que même notre modèle de base simple atteint des performances comparables à l’état de l’art actuel, et qu’en utilisant un pooling entraînable, nous conservons cette qualité élevée tout en étant 1,8 fois plus rapide pendant l’entraînement, 4,5 fois plus rapide pendant l’inférence, et jusqu’à 13 fois plus efficace sur le plan computationnel dans le décodeur.


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
Réduction de densité des modèles Transformer par pooling de représentation entraînable | Articles | HyperAI