In plain words: Instead of hand-picking training settings, this approach computes gradients through the whole training run so the settings can be tuned automatically by gradient descent. It beat existing ways of choosing training data and found learning-rate schedules as good as ones people design by hand.
Abstract · Optimizing ML Training with Metagradient Descent
A major challenge in training large-scale machine learning models is configuring the training process to maximize model performance, i.e., finding the best training setup from a vast design space. In this work, we unlock a gradient-based approach to this problem. We first introduce an algorithm for efficiently calculating metagradients -- gradients through model training -- at scale. We then introduce a "smooth model training" framework that enables effective optimization using metagradients. With metagradient descent (MGD), we greatly improve on existing dataset selection methods, outperform accuracy-degrading data poisoning attacks by an order of magnitude, and automatically find competitive learning rate schedules.
Logan Engstrom, Andrew Ilyas, Benjamin Chen, Axel Feldmann, William Moses, Aleksander Madry
arXiv:2503.13751 · stat.ML, cs.AI, cs.LG · submitted Mar 17, 2025
abstract · pdf · html
This is not to even mention more complex design decisions, like the architecture of the model, which can't be captured in a simple hyperparameter.