让优化器自己学会正则化,无需显式添加规则
Learning Regularizers: Learning Optimizers that can Regularize
- 用元学习训练优化器,让它自动内化正则化效果
- 在多个数据集上,带正则化的优化器测试准确率更高
- 新任务中仍能迁移正则化能力,适合想提升泛化的研究者
学习型优化器(LOs)作为一类元学习方法,因其可参数化且可训练的特性,在高效优化中备受关注。传统基于梯度的方法通过显式正则化技术(如SAM、GAM、GSAM)提升泛化性和收敛性。本文探讨一个根本问题:能否让正则化被学习?我们通过大量实验验证,即使不显式在目标函数中应用正则化,LOs也能被训练出内化传统正则化效果的能力。在标准基准(包括MNIST、FMNIST、CIFAR及MLP、MLP-Relu、CNN等神经网络)上的对比表明,经过正则化训练的LOs在测试准确率和泛化性能上始终优于未正则化版本。此外,这些优化器能将正则化效应迁移到新优化任务中,本质是寻找与正则化目标相似的极小值点。结果表明,优化器可内在学习正则属性,挑战了显式优化目标正则化的必要性。
原文摘要 · Abstract (English)
Learned Optimizers (LOs), a type of Meta-learning, have gained traction due to their ability to be parameterized and trained for efficient optimization. Traditional gradient-based methods incorporate explicit regularization techniques such as Sharpness-Aware Minimization (SAM), Gradient-norm Aware Minimization (GAM), and Gap-guided Sharpness-Aware Minimization (GSAM) to enhance generalization and convergence. In this work, we explore a fundamental question: \textbf{Can regularizers be learned?} We empirically demonstrate that LOs can be trained to learn and internalize the effects of traditional regularization techniques without explicitly applying them to the objective function. We validate this through extensive experiments on standard benchmarks (including MNIST, FMNIST, CIFAR and Neural Networks such as MLP, MLP-Relu and CNN), comparing LOs trained with and without access to explicit regularizers. Regularized LOs consistently outperform their unregularized counterparts in terms of test accuracy and generalization. Furthermore, we show that LOs retain and transfer these regularization effects to new optimization tasks by inherently seeking minima similar to those targeted by these regularizers. Our results suggest that LOs can inherently learn regularization properties, \textit{challenging the conventional necessity of explicit optimizee loss regularization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。