提出高阶正则化方法,提升神经网络泛化能力与可解释性。
High-order Regularization for Machine Learning and Learning-based Control
- 基于逆映射视角构建高阶正则化,理论可计算误差边界。
- 证明正则化使模型逼近具有收缩性质,显著增强泛化性能。
- 适用于任意映射矩阵的神经网络,适合需要可解释学习的场景。
本文提出一种新型高阶正则化(HR)方法,为机器学习提供新视角。该方法建立正则化与可解释学习之间的理论联系,证明正则化可视为逆映射近似,且其 $L_2$ 正则为特例。文中给出了 HR 解的上下界误差,确保逼近算法的可证明收敛性。进一步证明正则化具收缩性,最优正则化矩阵能最大化神经网络泛化能力,且对任意映射矩阵均适用。结合极限学习机理论,本方法可更好解释网络输出。通过基于正则化极限学习机的案例研究,验证了增量式 HR 解法的有效性,并在经典强化学习控制问题中展现显著泛化性能提升。
原文摘要 · Abstract (English)
The paper proposes a novel regularization procedure for machine learning. The proposed high-order regularization (HR) provides new insight into regularization, which is widely used to train a neural network that can be utilized to approximate the action-value function in general reinforcement learning problems. The proposed HR method ensures the provable convergence of the approximation algorithm, which makes the much-needed connection between regularization and explainable learning using neural networks. The proposed HR method theoretically demonstrates that regularization can be regarded as an approximation in terms of inverse mapping with explicitly calculable approximation error, and the $L_2$ regularization is a lower-order case of the proposed method. We provide lower and upper bounds for the error of the proposed HR solution, which helps build a reliable model. We also find that regularization with the proposed HR can be regarded as a contraction. We prove that the generalizability of neural networks can be maximized with a proper regularization matrix, and the proposed HR is applicable for neural networks with any mapping matrix. With the theoretical explanation of the extreme learning machine for neural network training and the proposed high-order regularization, one can better interpret the output of the neural network, thus leading to explainable learning. We present a case study based on regularized extreme learning neural networks to demonstrate the application of the proposed HR and give the corresponding incremental HR solution. We verify the performance of the proposed HR method by solving a classic control problem in reinforcement learning. The result demonstrates the superior performance of the method with significant enhancement in the generalizability of the neural network.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。