发现深度网络在记忆数据时仍保留泛化能力,可从中解码出提升性能的关键信息。
Decoding Generalization from Memorization in Deep Neural Networks
- 通过扰动标签实验,发现模型内部仍存有真实标签的泛化线索
- 提出方法从网络中间层解码出显著提升泛化的信息
- 适用于研究模型泛化机制或改进训练策略的研究者
近年来,过参数化的深度神经网络虽能良好泛化,但其原因尚不明确。当训练标签被不同程度打乱时,网络仍可达到完美训练准确率,但对真实标签的泛化能力严重下降——这一现象称为记忆。以往不清楚这种泛化失败是因网络所有层彻底重构表示,还是某些层仍保留泛化潜力却未被有效利用。本文通过实验证明,此类模型在内部表示中仍包含显著提升泛化能力的信息,且可通过简单技术从网络内部解码。该方法在多个标准数据集上的多种模型中均取得有效结果。代码已公开于:https://github.com/simranketha/MASC_DNN。
原文摘要 · Abstract (English)
Overparameterized deep networks that generalize well have been key to the dramatic success of deep learning in recent years. The reasons for their remarkable ability to generalize are not well understood yet. When class labels in the training set are shuffled to varying degrees, it is known that deep networks can still reach perfect training accuracy at the detriment of generalization to true labels -- a phenomenon that has been called memorization. It has, however, been unclear why the poor generalization to true labels that accompanies such memorization, comes about. One possibility is that during training, all layers of the network irretrievably re-organize their representations in a manner that makes generalization to true labels difficult. The other possibility is that one or more layers of the trained network retain significantly more latent ability to generalize to true labels, but the network somehow "chooses" to readout in a manner that is detrimental to generalization to true labels. Here, we provide evidence for the latter possibility by demonstrating, empirically, that such models possess information in their representations for substantially-improved generalization to true labels. Furthermore, such abilities can be easily decoded from the internals of the trained model, and we build a technique to do so. We demonstrate results on multiple models trained with standard datasets. Our code is available at: https://github.com/simranketha/MASC_DNN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。