Fast Inference from Transformers via Speculative Decoding

Mots clés générés par l'IA : Décodage spéculatif Transformers YaGuang Li Blake Hechtman XLA

Points clés générés par l'IA

  • Remerciements à YaGuang Li et Blake Hechtman pour leur aide avec LaMDA et XLA
  • Remerciements à Asaf Aharoni, Sasha Goldshtein, Nadav Sherman, Eyal Segalis, Eyal Molad, Dani Valevski, Daniel Wasserman, Valerie Nygaard et les équipes LaMDA et Theta Labs chez Google ainsi que nos familles
  • Méthode de décodage spéculatif pour accélérer l'inférence à partir de modèles autorégressifs sans changer les sorties
  • Utilisation d'exécution spéculative et d'une nouvelle méthode d'échantillonnage pour exécuter en parallèle sur les sorties des modèles d'approximation générés potentiellement plusieurs jetons simultanés sans changer la distribution
  • Pas besoin de réentrainement ou modifications architecturales
  • Accélération 2X-3X par rapport à la mise en œuvre standard de T5X avec des sorties identiques
  • Études antérieures sur l'efficacité de l'inférence à partir de grands modèles (Dehghani et al., 2021)
  • Article intitulé "Fast Inference from Transformers via Speculative Decoding" présentant une méthode pour accélérer l'inférence à partir des transformers via un décodage spéculatif.
Accédez également à nos autres résultats générés par IA : Résumé complet, Résumé vulgarisé, Article de type blog; ou posez des questions sur cet article à notre Assistant IA.

Auteurs : Yaniv Leviathan, Matan Kalman, Yossi Matias

Licence : CC BY 4.0

Résumé : Inference from large autoregressive models like Transformers is slow - decoding K tokens takes K serial runs of the model. In this work we introduce speculative decoding - an algorithm to sample from autoregressive models faster without any changes to the outputs, by computing several tokens in parallel. At the heart of our approach lie the observations that (1) hard language-modeling tasks often include easier subtasks that can be approximated well by more efficient models, and (2) using speculative execution and a novel sampling method, we can make exact decoding from the large models faster, by running them in parallel on the outputs of the approximation models, potentially generating several tokens concurrently, and without changing the distribution. Our method supports existing off-the-shelf models without retraining or architecture changes. We demonstrate it on T5-XXL and show a 2X-3X acceleration compared to the standard T5X implementation, with identical outputs.

Soumis à arXiv le 30 Nov. 2022

Posez des questions sur cet article à notre assistant IA

Vous pouvez aussi discutez avec plusieurs papiers à la fois ici.

Instructions pour utiliser l'assistant IA ?

Résultats du processus de synthèse de l'article arXiv : 2211.17192v1

Nous tenons à remercier chaleureusement YaGuang Li pour son aide concernant LaMDA et le calcul des chiffres de LaMDA dans l'article ainsi que Blake Hechtman pour ses excellentes idées et son aide avec XLA. Nous souhaitons également remercier Asaf Aharoni, Sasha Goldshtein, Nadav Sherman, Eyal Segalis, Eyal Molad, Dani Valevski, Daniel Wasserman, Valerie Nygaard et les équipes LaMDA et Theta Labs chez Google ainsi que nos familles. Dans cette étude nous présentons une méthode de décodage spéculatif - un algorithme qui permet d'accélérer l'inférence à partir de modèles autorégressifs sans changer les sorties en calculant plusieurs jetons en parallèle. Notre approche est basée sur les observations suivantes : (1) les tâches de modélisation linguistique difficiles incluent souvent des sous-tâches plus faciles qui peuvent être bien approximés par des modèles plus efficaces et (2) en utilisant une exécution spéculative et une nouvelle méthode d'échantillonnage nous pouvons accélérer le décodage exact à partir des grands modèles en les exécutant en parallèle sur les sorties des modèles d'approximation générés potentiellement plusieurs jetons simultanés sans changer la distribution. Notre mise en œuvre ne requiert pas de réentrainement ou modifications architecturales. Nous montrons l'efficacité de notre mise en œuvre sur T5-XXL obtenant une accélération 2X-3X par rapport à la mise en œuvre standard de T5X avec des sorties identiques. En ce qui concerne les travaux connexes l’efficacité de l’infèrence à partir de grands modèles a était largement étudiée (Dehghani et al., 2021). De nombreux travaux interessants ont étés réalisés dans ce domaine. Enfin le document source est un extrait d'une requête ArXiv contenant des informations sur l'article intitulée "Fast Inference from Transformers via Speculative Decoding". Cet article présente une methode pour accerler l’inferece à partir des transformers via un décodage speculatif.
Créé le 21 Sep. 2023

Évaluez la qualité du contenu généré par l'IA en votant

Note : 0

Pourquoi avons-nous besoin de votes ?

Les votes sont utilisés pour déterminer si nous devons réexécuter nos outils de synthèse. Si le compte atteint -10, nos outils peuvent être redémarrés.

Le résumé précédent a été créé il y a plus d'un an et peut être réexécuté (si nécessaire) en cliquant sur le bouton Exécuter ci-dessous.

Articles similaires résumés avec nos outils d'IA

Naviguez à travers encore plus d'articles similaires en utilisant une

représentation arborescente

Recherchez des articles similaires (en version bêta)

En cliquant sur le bouton ci-dessus, notre algorithme analysera tous les articles de notre base de données pour trouver le plus proche en fonction du contenu des articles complets et pas seulement des métadonnées. Veuillez noter que cela ne fonctionne que pour les articles pour lesquels nous avons généré des résumés et que vous pouvez le réexécuter de temps en temps pour obtenir un résultat plus précis pendant que notre base de données s'agrandit.

Avertissement : Notre outil de synthèse basé sur l'IA et l'assistant virtuel fournis sur ce site Web peuvent ne pas toujours fournir des résumés complets ou des réponses exactes. Nous vous encourageons à examiner attentivement et à évaluer le contenu généré pour vous assurer de sa qualité et de sa pertinence par rapport à vos besoins.