Cross-regularization: Adaptive Model Complexity through Validation Gradients
Carlos Stein Brito
Abstract
Model regularization requires extensive manual tuning to balance complexity against overfitting. Crossregularization resolves this tradeoff by directly adapting regularization parameters through validation gradients during training. The method splits parameter optimization -training data guides feature learning while validation data shapes complexity controls -converging provably to cross-validation optima. When implemented through noise injection in neural networks, this approach reveals striking patterns: unexpectedly high noise tolerance and architecture-specific regularization that emerges organically during training. Beyond complexity control, the framework integrates seamlessly with data augmentation, uncertainty calibration and growing datasets while maintaining single-run efficiency through a simple gradient-based approach.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on3
- Sharpness-aware Minimization for Efficiently Improving GeneralizationPierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam NeyshaburICLR 2021 · 1,861 citations
- Deep Double Descent: Where Bigger Models and More Data HurtPreetum Nakkiran, Gal Kaplun, Yamini Bansal, Tristan Yang et al.ICLR 2020 · 1,108 citations
- Weighted Training for Cross-Task LearningShuxiao Chen, Koby Crammer, Hangfeng He, Dan Roth et al.ICLR 2022 · 30 citations
Related papers
- The Generalization-Stability Tradeoff In Neural Network PruningBrian R. Bartoldson, Ari S. Morcos, Adrian Barbu, Gordon ErlebacherNeurIPS 2020 · 97 citations
- On the Interpretability of Regularisation for Neural Networks Through Model Gradient SimilarityVincent Szolnoky, Viktor Andersson, Balázs Kulcsár, Rebecka JörnstenNeurIPS 2022 · 6 citations
- Noise Injection Node Regularization for Robust LearningNoam Levi, Itay M. Bloch, Marat Freytsis, Tomer VolanskyICLR 2023 · 2 citations
- Neural Complexity MeasuresYoonho Lee, Juho Lee, Sung Ju Hwang, Eunho Yang et al.NeurIPS 2020 · 13 citations
- Improving Robustness with Adaptive Weight DecayAmin Ghiasi, Ali Shafahi, Reza ArdekaniNeurIPS 2023 · 18 citations
