À suivreGuide suivant
Modèles d’attention clairsemés
IA linguistique
GUIDE Technique
L'attention fragmentée en blocs et native permet aux transformateurs de s'occuper uniquement des morceaux les plus pertinents d'une longue séquence au lieu de chaque jeton, réduisant ainsi le coût quadratique de l'attention standard.
Écoutez ce guideNarré par l'IA · ~2 min
C’est ce qui rend les modèles efficaces à contexte long pratiques sur du matériel réel.
L'auto-attention standard compare chaque jeton à chaque autre jeton, de sorte que le coût augmente quadratiquement avec la longueur de la séquence, devenant prohibitif pour les documents très longs. Une attention éparse restreint chaque jeton à un sous-ensemble d’autres. Les approches par blocs divisent la séquence en blocs et calculent l'attention uniquement pour les paires de blocs sélectionnées, qui sont mappées efficacement sur les cœurs tenseurs du GPU. Native Sparse Attention (NSA), de DeepSeek, va plus loin : il est entraînable de bout en bout et aligné sur le matériel, combinant trois branches, une compression de jetons à gros grain, une sélection fine des blocs les plus importants et une fenêtre coulissante pour le contexte local. Étant donné que le modèle de parcimonie est appris pendant le pré-entraînement plutôt que intégré par la suite, la NSA préserve la précision tout en offrant des accélérations importantes sur de longues séquences.
Exécuter un modèle sur l'intégralité d'une base de code ou sur un long contrat juridique où toute l'attention épuiserait la mémoire du GPU.
La NSA de DeepSeek accélère à la fois le pré-entraînement et l'inférence à long contexte tout en correspondant ou en dépassant la précision de la pleine attention.
Résumer des documents de la longueur d'un livre en consultant des résumés de blocs compressés ainsi que des passages pertinents au niveau local.
Accélérer les assistants de discussion à contexte long dont l'étape de décodage est limitée en mémoire en limitant chaque jeton aux blocs les mieux classés.
La parcimonie entraînable et sensible au matériel devient la voie vers un contexte d'un million de jetons sans exploser les coûts. Attendez-vous à ce qu’une attention éparse soit co-conçue avec des noyaux et des accélérateurs, mélangée à des idées d’attention linéaire et d’espace d’état, et adoptée dans des modèles de raisonnement et de contexte long frontière. À mesure que les modèles deviennent apprenables et dynamiques, les modèles alloueront un budget d'attention de manière adaptative à chaque requête, et les benchmarks mesureront de plus en plus le débit de décodage sur de longues séquences, et pas seulement la qualité brute.
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
L'attention fragmentée en blocs et native permet aux transformateurs de s'occuper uniquement des morceaux les plus pertinents d'une longue séquence au lieu de chaque jeton, réduisant ainsi le coût quadratique de l'attention standard. C’est ce qui rend les modèles efficaces à contexte long pratiques sur du matériel réel.
Chaque jeton s'occupe de tous les autres jetons, donc le calcul et la mémoire évoluent avec le carré de la longueur de la séquence.
La séquence est divisée en blocs et l'attention est calculée uniquement pour les paires de blocs choisies, qui correspondent également bien aux cœurs tenseurs du GPU.
La NSA apprend son modèle de parcimonie lors du pré-entraînement et est conçue pour s'aligner sur le matériel, de sorte qu'elle préserve la précision tout en fonctionnant rapidement.
La NSA fusionne une compression grossière de jetons, une sélection de blocs à granularité fine et une fenêtre coulissante locale à l'aide de portes apprises.
Les modèles d'accès au niveau des blocs conviennent au matériel GPU, de sorte que les économies théoriques de FLOP se transforment en véritables accélérations d'horloge murale.
Continuez à apprendre
Plus de guides sélectionnés pour ce sujet
À suivreGuide suivant
Modèles d’attention clairsemés
IA linguistique