arXiv:2603.19865cs.LG2026-03被引 1

发现模型在记忆数据时,潜在泛化能力早期就达到峰值。

On the Dynamics & Transferability of Latent Generalization during Memorization

  • 用线性探针验证潜在泛化可被线性解码
  • 潜在泛化能力在训练初期达峰值,后下降
  • 通过权重编辑可将潜在泛化转移至模型性能

深度网络在标签随机打乱的训练数据上仍能实现高准确率,这一现象称为记忆。尽管此时模型整体泛化能力下降,但其内部表示仍保留显著的潜在泛化能力。我们此前证明可通过简单探针(MASC探针)从层间表示中恢复这种能力。本文通过实证追踪训练过程,发现潜在泛化能力在训练早期即达峰值,并随训练逐渐减弱。进一步分析表明,原MASC探针为非线性二次分类器,因此我们设计了一种新线性探针以检验线性可解码性。最终提出一种权重编辑方法,将最后一层表示中的潜在泛化直接转移到模型本身,实现了泛化能力的有效迁移。

原文摘要 · Abstract (English)

Deep networks have been known to have extraordinary generalization abilities, via mechanisms that aren't yet well understood. It is also known that upon shuffling labels in the training data to varying degrees, deep networks, trained with standard methods, can still achieve perfect or high accuracy on this corrupted training data. This phenomenon is called memorization, and typically comes at the cost of poorer generalization to true labels. Our recent work has demonstrated, that the internal representations of such models retain significantly better latent generalization abilities than is directly apparent from the model. In particular, it has been shown that such latent generalization can be recovered via simple probes (called MASC probes) on the layer-wise representations of the model. However, the origin and dynamics over training of this latent generalization during memorization is not well understood. Here, we track the training dynamics, empirically, and find that latent generalization abilities largely peak early in training, with model generalization. Next, we investigate to what extent the specific nature of the MASC probe is critical for our ability to extract latent generalization from the model's layerwise outputs. To this end, we first examine the mathematical structure of the MASC probe and show that it is a quadratic classifier, i.e. is non-linear. This brings up the question of the extent to which this latent generalization might be linearly decodable from layerwise outputs. To investigate this, we designed a new linear probe for this setting. Next, we consider the question of whether it is possible to transfer latent generalization to model generalization by directly editing model weights. To this end, we devise a way to transfer the latent generalization present in last-layer representations to the model using the new linear probe.

深度学习泛化能力模型记忆探针分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。