用随机损失集成防过拟合,提升分类模型泛化能力
EnsLoss: Stochastic Calibrated Loss Ensembles for Preventing Overfitting in Classification
- 将损失函数融合为随机集成,通过梯度双重随机化实现
- 在14个表格数据集和46个图像数据集上显著降低过拟合
- 无需显式定义损失函数,自动保持校准性,适合深度学习场景
经验风险最小化(ERM)结合计算可行的代理损失是分类任务的主流方法。损失函数的凸性和校准性(CC)特性确保了ERM在最大化准确率时的一致性,从而提供了丰富的代理损失选择。本文提出一种新型集成方法EnsLoss,将集成学习思想扩展至ERM框架中的损失函数组合。其关键在于保持组合损失的“合法性”,即维持CC性质。具体地,将损失的CC条件转化为损失导数,绕过显式损失函数的定义,直接生成校准的损失导数。受Dropout启发,EnsLoss通过一次训练过程实现损失集成,采用双重随机梯度下降(即随机样本批次与随机校准损失导数)。我们从理论上建立了该方法的统计一致性,并揭示其优势。实验在14个OpenML表格数据集和46个图像数据集上,使用多种深度学习架构验证了EnsLoss相比固定损失方法的数值有效性。代码已开源于GitHub。
原文摘要 · Abstract (English)
Empirical risk minimization (ERM) with a computationally feasible surrogate loss is a widely accepted approach for classification. Notably, the convexity and calibration (CC) properties of a loss function ensure consistency of ERM in maximizing accuracy, thereby offering a wide range of options for surrogate losses. In this article, we propose a novel ensemble method, namely EnsLoss, which extends the ensemble learning concept to combine loss functions within the ERM framework. A key feature of our method is the consideration on preserving the "legitimacy" of the combined losses, i.e., ensuring the CC properties. Specifically, we first transform the CC conditions of losses into loss-derivatives, thereby bypassing the need for explicit loss functions and directly generating calibrated loss-derivatives. Therefore, inspired by Dropout, EnsLoss enables loss ensembles through one training process with doubly stochastic gradient descent (i.e., random batch samples and random calibrated loss-derivatives). We theoretically establish the statistical consistency of our approach and provide insights into its benefits. The numerical effectiveness of EnsLoss compared to fixed loss methods is demonstrated through experiments on a broad range of 14 OpenML tabular datasets and 46 image datasets with various deep learning architectures. Python repository and source code are available on GitHub at https://github.com/statmlben/ensloss.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。