arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.08302cs.LGcs.AI
收藏

线性模型中逻辑回归的grokking现象

On the Dynamics of Delayed Generalization in Linear Models

Department of XXX, University of YYY, Location, Country(YYY大学XXX系);School of ZZZ, Institute of WWW, Location, Country(WWW研究所ZZZ学院);Indian Institute of Technology Madras, Chennai(马德拉斯印度理工学院)

Atreya Vedantam, Nataraj Das, Chandrashekar Lakshminarayanan

AI总结:

研究揭示了线性模型中逻辑回归的grokking现象,通过分析梯度下降隐含偏置的三阶段学习过程,发现延迟泛化现象可通过支撑向量分布和数据不对称性产生,无需深度或表示学习。

摘要:

grokking现象,即延迟泛化现象,通常归因于深度神经网络的深度和组合结构。我们研究了最简单的可能设置之一:在数据线性(且最大边缘)可分的情况下,学习一个具有逻辑损失的线性模型进行二分类。我们研究了三种测试模式:(1)测试数据从与训练数据相同的分布中抽取,在这种情况下不会观察到grokking;(2)测试数据集中在边缘附近,在这种情况下会观察到grokking;(3)通过投影梯度下降(PGD)攻击生成对抗性测试数据,在这种情况下也会观察到grokking。我们理论上证明了梯度下降的隐含偏置诱导了一个三阶段的学习过程——群体主导阶段、支持向量主导的反向学习阶段和支持向量主导的泛化阶段,在此过程中可能会出现延迟泛化。我们的分析进一步将grokking的出现与数据中的不对称性联系起来,包括每类样本的数量和支撑向量在各类中的分布,从而得出grokking时间的特征。我们通过种植不同的群体点和支撑向量分布,并分析准确率曲线和超平面动态来验证我们的理论。总体而言,我们的结果表明,grokking不需要深度或表示学习,甚至在通过偏置项动态在线性模型中也能出现。

英文摘要:

Delayed generalization, where training accuracy becomes nearly perfect substantially before test accuracy, has often been studied in the context of the depth and compositional structure of deep networks. In this paper, we ask whether delayed generalization can occur in simpler settings. We show that it can arise in a linear model trained with gradient descent and an exponential-tail loss on linearly separable data. We characterize the finite-time hyperplane dynamics into three phases: a population phase, a support vector dominated unlearning phase and a generalization phase. The convergence of the residual beyond the maximum-margin direction is sufficient for the unlearning phase to begin. In the support vector dominated phase, we characterize the hyperplane dynamics. The time scale in these dynamics causes delayed generalization. We validate our theory through controlled experiments varying population and support-vector distributions, and further verify the predicted bias behavior in a deep model with a single output neuron on MNIST. Our results provide a unified characterization of the asymptotic and finite-time dynamics underlying delayed generalization in linear models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑