À suivreGuide suivant
Agrégation de mélange d'agents
IA linguistique
GUIDE IA du langage
Le mélange de profondeurs (MoD) permet à un transformateur de dépenser différentes quantités de calcul sur différents jetons, en acheminant uniquement les jetons « importants » à travers le calcul lourd de chaque couche.
Écoutez ce guideNarré par l'IA · ~2 min
Il réduit le coût de traitement des jetons simples tout en conservant un budget de calcul fixe et prévisible.
Les transformateurs standard appliquent chaque couche à chaque jeton, même les plus triviaux comme la ponctuation. Mixture of Depths, introduit par Google DeepMind en 2024, ajoute un petit routeur à chaque bloc qui sélectionne une fraction top-k fixe de jetons pour subir l'auto-attention complète et le calcul MLP ; le reste saute le bloc via une connexion résiduelle. Étant donné que seuls k jetons sont traités par couche, le calcul total (FLOP) est plafonné et connu à l'avance, contrairement aux méthodes de profondeur dynamique antérieures qui variaient de manière imprévisible. Cela rend le traitement par lots et l’utilisation du matériel efficaces. Les modèles formés par le MoD peuvent correspondre à la qualité d'un transformateur de base en utilisant moins de FLOP par passage avant, ou atteindre une qualité supérieure avec le même calcul, et l'idée se compose naturellement avec un mélange d'experts pour donner des modèles « MoDE » qui acheminent à la fois la profondeur et la largeur.
Réduire les FLOP nécessaires au traitement de documents longs en ignorant les calculs approfondis sur les jetons de remplissage
Entraîner un modèle qui correspond à la qualité de base avec un calcul inférieur, réduisant ainsi le coût de diffusion
Combinaison avec le mélange d'experts (MoDE) pour tracer à la fois la profondeur des couches et le choix des experts
Conserver une latence prévisible et fixe par jeton, car le budget de calcul par couche est fixé à l'avance
Le calcul conditionnel est un levier majeur d’efficacité à mesure que les modèles évoluent, et le MoD en est un exemple précoce et clair. Attendez-vous à une intégration plus approfondie avec un mélange d'experts (routage à la fois en profondeur et en experts), à des budgets adaptatifs qui diminuent pour faciliter les entrées et à des routeurs appris qui identifient mieux les jetons qui nécessitent réellement un traitement approfondi. Alors que les coûts d'inférence dominent l'économie du déploiement, les techniques permettant aux modèles de « réfléchir plus sérieusement » uniquement lorsque cela est nécessaire, tout en conservant une latence prévisible, sont susceptibles de devenir la norme dans les architectures à grande échelle.
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
Le mélange de profondeurs (MoD) permet à un transformateur de dépenser différentes quantités de calcul sur différents jetons, en acheminant uniquement les jetons « importants » à travers le calcul lourd de chaque couche. Il réduit le coût de traitement des jetons simples tout en conservant un budget de calcul fixe et prévisible.
Le MoD n'achemine que quelques jetons à travers les calculs lourds de chaque couche, de sorte que différents jetons obtiennent effectivement une profondeur différente.
Le choix d'un top-k fixe de jetons par bloc limite les FLOP et maintient les formes de tenseurs constantes, ce qui est respectueux du matériel.
Les jetons non sélectionnés contournent le calcul du bloc et sont reportés inchangés par le chemin résiduel.
Le mélange de profondeurs a été introduit par Google DeepMind dans un article de 2024 sur le calcul des transformateurs dynamiques.
La combinaison du routage en profondeur avec le routage expert donne des modèles « MoDE » qui conditionnent le calcul sur les couches et les experts.
Continuez à apprendre
Plus de guides sélectionnés pour ce sujet
À suivreGuide suivant
Agrégation de mélange d'agents
IA linguistique