Applez Nous: +41 22 518 70 00
Lorsqu’un réseau possède plus de paramètres libres que d’exemples dans le jeu de données, le risque de mémorisation par cœur (Overfitting) est omniprésent. Les techniques de régularisation contraignent la complexité effective du réseau :
Alors que SGD utilise un taux d’apprentissage uniforme, Adam (Adaptive Moment Estimation) calcule des taux individuels pour chaque paramètre à partir des estimations des moments d’ordre 1 (moyenne) et d’ordre 2 (variance non centrée) des gradients. La variante AdamW découple correctement le weight decay du gradient pour une régularisation optimale.
# Stratégie de Cosine Annealing avec Warmup linéaire
scheduler = optim.lr_scheduler.CosineAnnealingWarmRestarts(
optimizer,
T_0=10,
T_mult=2,
eta_min=1e-6
)
Cette approche permet au modèle de franchir les zones chaotiques en début d’apprentissage puis d’atterrir en douceur dans les minima locaux les plus plats et généralisables.