Description
UL2 20B représente une avancée significative dans le pré-entraînement des modèles de langage, offrant un cadre unifié qui améliore les performances sur diverses tâches et configurations. Développelé par Google Research, ce modèle open-source aborde les limitations des paradigmes existants, qui excellent souvent soit dans le fine-tuning, soit dans l'apprentissage few-shot en contexte, mais peinent avec l'autre.
Les modèles de langage traditionnels entrent généralement dans deux catégories : les architectures autorégressives de type décodeur seul (comme GPT-3) qui prédisent le mot suivant, et les architectures encodeur-décodeur basées sur la corruption de spans (comme T5) qui récupèrent du texte masqué. Alors que les modèles autorégressifs sont aptes à la génération ouverte et à l'apprentissage basé sur des prompts, ils peuvent sous-performer sur les tâches de fine-tuning. Inversement, les modèles de type T5 obtiennent de bons résultats en fine-tuning supervisé mais sont moins efficaces dans les scénarios few-shot. UL2 comble cet écart en unifiant ces approches.
L'innovation principale d'UL2 réside dans son nouveau paradigme de pré-entraînement, l'Unified Language Learner (UL2). Ce cadre formule différents objectifs d'entraînement comme des tâches de débruitage, où le modèle apprend à reconstruire des sous-séquences manquantes du texte d'entrée. Pendant le pré-entraînement, UL2 emploie un mélange unique de dénoisers, échantillonnant à partir d'une variété d'objectifs avec différentes configurations. Cette approche permet au modèle de bénéficier simultanément des forces de multiples stratégies d'entraînement, atténuant les faiblesses individuelles.
Le mélange de dénoisers d'UL2 comprend trois tâches principales : le R-denoising (corruption de span standard), le X-denoising (corruption de span extrême) et le S-denoising (PrefixLM séquentiel). En échantillonnant ces tâches en fonction de ratios spécifiés par l'utilisateur et en ajoutant un token de paradigme (par exemple, [R], [X], [S]) pour indiquer la tâche, UL2 entraîne un modèle de langage plus polyvalent et robuste. Cette approche unifiée conduit à des performances améliorées en génération, compréhension du langage, récupération d'informations, compréhension de textes longs et réponse aux questions.
Le modèle UL2 20B, avec ses 20 milliards de paramètres, démontre des capacités exceptionnelles en matière de prompting few-shot et de raisonnement en chaîne de pensée (CoT). Il surpasse d'autres modèles de pointe comme PaLM et T5 sur des tâches telles que la résumé en 1 shot (XSUM), atteignant des scores ROUGE supérieurs. De plus, UL2 20B permet le prompting et le raisonnement CoT à une échelle accessible, rendant les techniques de raisonnement avancées disponibles à une communauté de recherche plus large. La publication publique des checkpoints UL2 20B vise à accélérer les progrès dans le développement de meilleurs modèles de langage au sein de la communauté du machine learning.
Points forts de UL2 20B
Paradigme unifié d'apprentissage du langage
Objectif de pré-entraînement par mélange de dénoisers
Prend en charge les tâches R-denoising, X-denoising et S-denoising
Améliore les performances sur les tâches de fine-tuning
Renforce les capacités d'apprentissage few-shot en contexte
Excelle dans la génération ouverte
Fortes performances en compréhension du langage
Efficace pour les tâches de récupération d'informations
Capable de comprendre des textes longs
Aide aux tâches de réponse aux questions
Checkpoints de modèle de 20 milliards de paramètres publiés publiquement
Permet le prompting en chaîne de pensée (CoT)
Facilite le raisonnement à une échelle accessible
Premiers pas avec UL2 20B
Accéder au modèle : Obtenir les checkpoints du modèle UL2 20B.
Configurer l'environnement : Configurer les bibliothèques et l'infrastructure nécessaires.
Intégrer via API : Charger le modèle et préparer les données d'entrée.
Définir la tâche : Spécifier la tâche de langage souhaitée (par exemple, résumé, QA).
Exécuter l'inférence : Traiter les données d'entrée à travers le modèle.
Évaluer les résultats : Analyser les sorties du modèle pour les performances.
Cas d'utilisation de UL2 20B
- Apprentissage Few-Shot
- Génération de Texte
- Compréhension du Langage
- Réponse aux Questions
- Résumé
- Tâches de Raisonnement
- Récupération d'Informations





