El momento que acelera el entrenamiento de redes neuronales exprés

Original: "Critical Damping as a Momentum Schedule: Multi-Seed Validation, a Hybrid Recipe, and an Exhaustive Negative Result on Surgical Layer Selection"
· Ivan Pasichnyk
arXiv:2603.28921 · 2026-03-30 · CC BY 4.0 · 2 min
Un momento bien elegido en el optimizador acelera el entrenamiento de la red neuronal en 2,4 veces sin perder precisión.

Entrenar una red neuronal es como bajar una montaña en trineo: necesitas la velocidad adecuada para no quedarte atascado ni salirte de la pista. En el popular método de entrenamiento con momento (como la inercia), normalmente se usa un valor constante de 0,9. Los autores del artículo recordaron la física de las oscilaciones amortiguadas y propusieron cambiar el momento durante el entrenamiento: primero es grande, luego disminuye según una fórmula donde el momento es igual a uno menos dos veces la raíz cuadrada de la tasa de aprendizaje.

La fórmula se deriva de la condición de que el sistema está "críticamente amortiguado" — como un amortiguador que elimina las oscilaciones sin balanceos innecesarios.

Se probó en una tarea estándar: la red neuronal ResNet-18 aprende a reconocer imágenes del conjunto CIFAR-10. Con el nuevo programa de momento, el modelo alcanza el 90% de precisión 2,34 veces más rápido que con el constante de 0,9. Sin embargo, la precisión final es ligeramente menor: 0,46 puntos porcentuales menos. Entonces se aplicó un enfoque híbrido: primero amortiguamiento crítico hasta el 90%, luego cambio a momento constante. Esto elimina la pérdida de precisión y mantiene la aceleración: la precisión final es 95,45%, y al 90% el modelo llega 2,4 veces más rápido.

La aceleración también funciona para otras arquitecturas: VGG-16 es 1,72 veces más rápido, y en el conjunto más complejo CIFAR-100 la ganancia es más modesta, pero aún presente.

En la segunda parte del artículo, los autores probaron la idea de seleccionar capas para el ajuste fino según los gradientes de error. Resultó que la selección por gradientes no es mejor que la aleatoria. Pero hay una señal útil: si una capa tiene el gradiente más bajo, definitivamente no vale la pena tocarla — intervenir en ella puede causar mucho daño. Así que los gradientes ayudan a evitar malas decisiones, no a encontrar las buenas.

El momento en el optimizador se llama así por analogía con el momento de inercia físico: ayuda a superar los mínimos locales, como un trineo que pasa sobre pequeños baches gracias a la velocidad acumulada.

\mu(t) = 1 - 2\sqrt{\alpha(t)}
Aquí μ(t) es el momento en el paso t, α(t) es la tasa de aprendizaje en el mismo paso. La fórmula define cómo debe cambiar el momento durante el entrenamiento para que el sistema esté críticamente amortiguado.
Conceptos
tasa de aprendizaje xgboost