发现深度模型局部遗忘现象,提出可修复的训练优化方法
On Local Overfitting and Forgetting in Deep Neural Networks
- 定义遗忘率评分,量化模型在数据特定区域的性能退化
- 实验证明局部过拟合独立于传统过拟合存在,且在多种架构中普遍
- 新集成方法利用单个网络训练历史恢复遗忘知识,无额外推理开销
深度神经网络中过拟合罕见令人困惑:理论预期模型越大越易过拟合,但实践中增大模型常提升性能。然而,若过拟合仅局限在数据空间的特定子区域呢?本文提出一种新评分,用于衡量深度模型在验证数据上的遗忘率。我们假设该评分可量化局部过拟合——即性能下降局限于数据空间的某些区域。实证表明,局部过拟合即使在无传统过拟合时也存在。基于深度过参数线性模型框架,我们对遗忘知识给出理论刻画,并证明其与真实深度模型的遗忘行为相关。最后,我们设计了一种新集成方法,仅依赖单个网络的训练历史来恢复遗忘知识。结合自蒸馏,该方法可在不增加推理成本的前提下提升任意训练模型的表现。大量实验证明其在多个数据集、现代神经网络架构和训练协议下的有效性。
原文摘要 · Abstract (English)
The infrequent occurrence of overfitting in deep neural networks is perplexing: contrary to theoretical expectations, increasing model size often enhances performance in practice. But what if overfitting does occur, though restricted to specific sub-regions of the data space? In this work, we propose a novel score that captures the forgetting rate of deep models on validation data. We posit that this score quantifies local overfitting: a decline in performance confined to certain regions of the data space. We then show empirically that local overfitting occurs regardless of the presence of traditional overfitting. Using the framework of deep over-parametrized linear models, we offer a certain theoretical characterization of forgotten knowledge, and show that it correlates with knowledge forgotten by real deep models. Finally, we devise a new ensemble method that aims to recover forgotten knowledge, relying solely on the training history of a single network. When combined with self-distillation, this method enhances the performance of any trained model without adding inference costs. Extensive empirical evaluations demonstrate the efficacy of our method across multiple datasets, contemporary neural network architectures, and training protocols.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。