Échauffement du taux d'apprentissage
Données
| Taux d'apprentissage de pic | 0,0003 |
|---|---|
| Taux d'apprentissage minimal | 3e-5 |
| Étapes d'échauffement | 2 000 |
| Étapes totales | 100 000 |
| Étape courante | 1 000 |
Échauffement du taux d'apprentissage
Calculez le taux d'apprentissage à n'importe quelle étape selon le calendrier standard des transformeurs : un échauffement linéaire jusqu'à un pic suivi d'une décroissance en cosinus vers un plancher, ainsi que la fraction d'échauffement.
Données
Calendrier
Résultats
Saisissez une valeur pour afficher les résultats.
Détails
Échauffement du taux d'apprentissage
Le taux d'apprentissage est l'hyperparamètre le plus déterminant de l'entraînement des réseaux de neurones, et les transformeurs modernes le maintiennent rarement constant. La recette standard le fait monter de zéro vers un pic sur les premières étapes, puis le fait décroître selon une courbe en cosinus vers un plancher modeste pour le reste de l'entraînement. Ce calcul renvoie le taux d'apprentissage à n'importe quelle étape choisie selon ce calendrier, ainsi que la part de l'entraînement consacrée à l'échauffement.
Pourquoi le taux évolue dans le temps
Au début de l'entraînement, les poids sont aléatoires et les estimations courantes des statistiques de gradient tenues par l'optimiseur ne sont pas fiables : de grandes mises à jour peuvent faire diverger la perte. Un court échauffement, durant lequel le taux monte linéairement de zéro, maintient les premières mises à jour faibles pendant que l'optimiseur se stabilise. Une fois l'échauffement terminé, un taux élevé accélère les progrès, mais le conserver élevé jusqu'à la fin laisse le modèle osciller autour d'un minimum sans s'y poser. Faire décroître le taux — ici selon une courbe en cosinus — permet au modèle de faire de grands pas tôt, puis des pas de plus en plus fins à mesure qu'il converge.
Le calendrier
Deux phases régissent le taux. Pendant l'échauffement, tant que l'étape courante ne dépasse pas la longueur d'échauffement , le taux monte linéairement jusqu'au pic . Après l'échauffement, il suit une décroissance en cosinus du pic vers le plancher sur les étapes restantes jusqu'au total .
ηη=ηmaxws=ηmin+21(ηmax−ηmin)(1+cosS−wπ(s−w))(s≤w)(s>w)La fraction d'échauffement vaut simplement . À la moitié de la décroissance, le terme en cosinus est nul : le taux se situe alors exactement à mi-chemin entre le pic et le plancher.
Exemple chiffré
Soit un pic de 0,0003, un plancher de 0,00003, 2 000 étapes d'échauffement et 100 000 étapes au total. À l'étape 1 000 — encore dans l'échauffement — et à l'étape 51 000 — la moitié de la décroissance —, les taux valent
η1000η51000=0.0003×20001000=0.00015=0.00003+21(0.0003−0.00003)(1+cos2π)=0.000165À mi-échauffement, le taux vaut la moitié du pic. À la moitié de la décroissance, où le cosinus est nul, il se place exactement entre le pic et le plancher. L'échauffement couvre ici 2 000 des 100 000 étapes, soit 2 % de l'entraînement.
Choisir les valeurs
Le taux de pic domine le calendrier et se détermine en général empiriquement, souvent par un test de plage qui augmente le taux jusqu'à ce que la perte cesse de diminuer. Les pics publiés pour les grands transformeurs se situent couramment entre environ 1e-4 et 6e-4, et le pic tend à varier avec la taille de lot : changer la taille de lot effective appelle donc généralement un nouveau réglage. Un échauffement de quelques centaines à quelques milliers d'étapes, souvent un petit pourcentage de l'entraînement, est usuel. Le plancher est fréquemment fixé à environ un dixième du pic, ou à zéro lorsqu'une décroissance complète est souhaitée.
Limites
Les cadres logiciels diffèrent dans des détails que ce modèle ne capte pas : certains mesurent le calendrier en jetons plutôt qu'en étapes, d'autres paramètrent le plancher comme un rapport du pic, et d'autres ajoutent une queue finale constante ou linéaire. La forme cosinus avec échauffement retenue ici est le réglage par défaut courant, mais les bons hyperparamètres dépendent du problème et doivent être confirmés par l'expérience. Le calcul que représente un nombre d'étapes donné est estimé par le Temps et coût d'entraînement, et la facture matérielle correspondante par le Calculateur de coût GPU cloud.
Questions fréquentes (FAQ)
Pourquoi échauffer le taux d'apprentissage plutôt que démarrer au pic ?
Au début de l'entraînement, l'optimiseur dispose de peu d'informations fiables : les méthodes adaptatives ont des estimations bruitées des statistiques de gradient, et les poids initialisés au hasard produisent des mises à jour amples et erratiques. Partir d'emblée au pic dans cet état peut déstabiliser l'entraînement ou le faire diverger.
Faire monter le taux sur les premières étapes laisse l'optimiseur se stabiliser tant que les mises à jour restent faibles, après quoi le pic peut être appliqué sans risque. L'échauffement est particulièrement important pour les grands lots et les transformeurs profonds, où l'instabilité initiale est plus marquée.
Quel calendrier ce calcul modélise-t-il ?
Il modélise le calendrier utilisé par la plupart des entraînements modernes de transformeurs : un échauffement linéaire de zéro au pic sur les étapes d'échauffement, suivi d'une décroissance en cosinus du pic vers le minimum sur les étapes restantes. La forme en cosinus passe plus de temps près du taux élevé en début de décroissance et rejoint doucement le plancher à la fin, ce qui s'entraîne bien en pratique.
D'autres calendriers existent — constant, décroissance linéaire, racine carrée inverse, décroissance par paliers —, mais l'échauffement linéaire suivi d'une décroissance en cosinus est le réglage par défaut courant que cet outil reproduit.
Comment choisir le taux d'apprentissage de pic ?
Le taux de pic est l'hyperparamètre le plus important du calendrier et se détermine en général empiriquement, souvent par un court test de plage qui augmente le taux jusqu'à ce que la perte cesse de diminuer, puis revient en arrière.
Les valeurs publiées pour les grands transformeurs se situent couramment entre environ 1e-4 et 6e-4 pour les plus grands modèles, les modèles plus petits tolérant des taux un peu plus élevés. Le pic varie en général avec la taille de lot : un changement de taille de lot effective appelle donc souvent un nouveau réglage.
Mentions légales
Ce calcul reproduit le calendrier courant d'échauffement linéaire puis de décroissance en cosinus ; les cadres logiciels réels diffèrent dans le détail, par exemple selon que la décroissance se mesure en étapes ou en jetons et selon la façon dont le minimum est paramétré. Les hyperparamètres optimaux dépendent du problème et doivent être confirmés empiriquement.