提出新方法缓解模型局部过拟合,提升泛化能力。
Forget Me Not: Fighting Local Overfitting with Knowledge Fusion and Distillation
- 通过分析验证集遗忘率,识别局部过拟合现象。
- 结合知识融合与蒸馏,恢复遗忘知识并提升性能。
- 适合存在标签噪声或需高效推理的场景。
深度神经网络中的过拟合现象比预期少见,这令人困惑——理论认为模型容量越大越易过拟合,但实践中却很少发生。我们提出一种新指标,衡量模型在验证数据上的遗忘速率,揭示了局部过拟合:性能退化仅局限于输入空间的特定子区域。即使未出现全局过拟合,局部过拟合仍可能发生,且与双下降现象密切相关。基于此,我们提出两阶段方法:首先聚合单个模型训练过程中的多个检查点形成集成,再将知识蒸馏回原始大小的单个模型,从而在不增加推理成本的前提下增强性能。在多个数据集、现代架构和训练策略下实验验证有效。尤其在标签噪声存在时,该方法(知识融合+知识蒸馏)优于原模型及独立训练的集成,实现训练与推理复杂度降低的罕见双赢。
原文摘要 · Abstract (English)
Overfitting in deep neural networks occurs less frequently than expected. This is a puzzling observation, as theory predicts that greater model capacity should eventually lead to overfitting -- yet this is rarely seen in practice. But what if overfitting does occur, not globally, but in specific sub-regions of the data space? In this work, we introduce a novel score that measures the forgetting rate of deep models on validation data, capturing what we term local overfitting: a performance degradation confined to certain regions of the input space. We demonstrate that local overfitting can arise even without conventional overfitting, and is closely linked to the double descent phenomenon. Building on these insights, we introduce a two-stage approach that leverages the training history of a single model to recover and retain forgotten knowledge: first, by aggregating checkpoints into an ensemble, and then by distilling it into a single model of the original size, thus enhancing performance without added inference cost. Extensive experiments across multiple datasets, modern architectures, and training regimes validate the effectiveness of our approach. Notably, in the presence of label noise, our method -- Knowledge Fusion followed by Knowledge Distillation -- outperforms both the original model and independently trained ensembles, achieving a rare win-win scenario: reduced training and inference complexity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。