À suivreGuide suivant
Attention dispersée en blocs et native
Technique
GUIDE IA du langage
Une attention éparse rend les Transformers moins chers en laissant chaque jeton s'occuper uniquement d'un sous-ensemble soigneusement choisi d'autres jetons plutôt que de tous.
Écoutez ce guideNarré par l'IA · ~2 min
Cela échange une petite portée mondiale contre de grosses économies de mémoire et de calcul sur de longues séquences.
Une attention personnelle totale compare chaque jeton à chaque autre jeton, de sorte que le coût augmente avec le carré de la longueur de la séquence, ce qui devient pénible pour les documents longs. Une attention éparse remplace le modèle dense par un modèle structuré. Les conceptions courantes incluent une attention à fenêtre coulissante (locale), où chaque jeton ne voit que les voisins proches ; des modèles striés ou dilatés qui avancent pour atteindre un contexte distant à moindre coût ; et des jetons globaux, quelques positions spéciales qui s'occupent de tout et auxquelles tout s'occupe, agissant comme des centres d'information. Des modèles tels que Longformer, BigBird et Sparse Transformer les combinent de sorte que le nombre total de connexions augmente à peu près linéairement au lieu de quadratiquement, permettant des contextes de milliers à dizaines de milliers de jetons.
Longformer traite des articles scientifiques ou des documents juridiques entiers en un seul passage à l'aide d'une fenêtre coulissante et d'une attention globale
BigBird gère les réponses aux questions et les séquences génomiques de longs documents avec une attention à échelle linéaire
Résumer un texte de la longueur d'un livre où toute l'attention épuiserait la mémoire du GPU
Systèmes de récupération et de discussion en contexte long qui utilisent des jetons de hub global pour acheminer les informations clés entre des milliers de jetons
L'attention éparse reste au cœur de la modélisation de contexte long, de plus en plus associée à des noyaux optimisés comme FlashAttention et à une parcimonie apprise ou dynamique qui sélectionne les jetons à prendre en compte par entrée. À mesure que les fenêtres contextuelles s'étendent vers des millions de jetons, les piles hybrides mélangent des couches clairsemées, denses et d'espace d'état. Attendez-vous à ce que des noyaux clairsemés sensibles au matériel et une attention basée sur le routage continuent de réduire le coût de lecture d'entrées très longues.
Newsletter gratuite
Trois histoires d'IA vérifiées chaque matin de la semaine, écrites dans un anglais simple. Gratuit pour toujours, sans publicité.
Un e-mail chaque jour de la semaine. Désabonnez-vous en un clic. Nous ne vendons ni ne partageons jamais votre adresse.
Testez-vous
Commentaires instantanés sur chaque réponse et un certificat partageable avec une pièce d'identité vérifiable une fois que vous avez réussi un cours.
Soutenez l’enseignement gratuit de l’IA. AI Understanding est une organisation à but non lucratif 501(c)(3) : pas de publicité, pas de paywall, jamais. Faire un don
Une attention éparse rend les Transformers moins chers en laissant chaque jeton s'occuper uniquement d'un sous-ensemble soigneusement choisi d'autres jetons plutôt que de tous. Cela échange une petite portée mondiale contre de grosses économies de mémoire et de calcul sur de longues séquences.
Une attention totale compare chaque jeton à chaque autre jeton, ce qui donne un coût O (N au carré) en temps et en mémoire.
L'attention locale limite la vue de chaque jeton à une fenêtre fixe de jetons environnants, ce qui réduit considérablement les coûts.
Quelques jetons globaux se connectent à toutes les positions, permettant aux informations de circuler à moindre coût sur toute la séquence.
BigBird a montré que son modèle clairsemé est un approximateur universel des fonctions de séquence tout en évoluant à peu près linéairement.
En limitant les connexions aux fenêtres locales et à quelques liens globaux, le nombre total de connexions augmente de manière linéaire.
Continuez à apprendre
Plus de guides sélectionnés pour ce sujet
À suivreGuide suivant
Attention dispersée en blocs et native
Technique