模型在稀疏数据中仍能高效学习,靠的是局部几何结构的对齐机制。
The Geometric Structure of Models Learning Sparse Data

- 通过法向对齐机制,揭示模型在稀疏数据下的有效学习原理。
- 法向对齐的分类器在约束条件下最小化损失并实现最大局部鲁棒性。
- 提出GrokAlign和RFAMs,提升训练速度与对抗鲁棒性,适合稀疏数据场景。
流形假设(MH)常被用来解释机器学习如何克服高维诅咒。然而,该假设仅适用于训练数据充分采样低维流形的情况,或低维流形确实存在的场景。我们将不满足这些条件的场景称为稀疏。本文表明,模型在稀疏条件下仍能成功,是因其利用了高度结构化的局部几何特性——我们将其形式化为法向对齐。我们证明,法向对齐的分类器(其输入-输出雅可比矩阵秩为1且与训练数据完全对齐)在范数约束下最小化训练目标,并在非零雅可比约束下实现最大局部鲁棒性。对于连续分段仿射深度网络,法向对齐在几何上表现为网络诱导的幂图划分中的质心对齐,源于特征学习阶段。基于这些理论洞察,我们提出GrokAlign正则化策略,主动诱导法向对齐。实验表明,GrokAlign显著加速了与grokking现象相关的深度网络训练动态。此外,我们将法向对齐原则应用于递归特征机器(RFMs),提出递归特征对齐机器(RFAMs)。结果表明,在表格数据上训练时,RFAMs相比RFMs展现出更强的对抗鲁棒性。
原文摘要 · Abstract (English)
The manifold hypothesis (MH) is often used to explain how machine learning can overcome the curse of dimensionality. However, the MH is only applicable in regimes where the training data provides a sufficiently dense sample of the underlying low-dimensional data manifold, or where such a low-dimensional manifold is conceivably present. We describe the regimes where the MH is not applicable as sparse. In this paper, we demonstrate that models succeed in the sparse regime by exploiting a highly structured local geometry, a property we formalize as normal alignment. We prove that normal-aligned classifiers -- whose input-output Jacobians are rank-one and align perfectly with the training data -- minimize the training objective under norm constraints and achieve maximal local robustness under a non-zero Jacobian constraint. For continuous piecewise-affine deep networks, normal alignment manifests geometrically as centroid alignment within the network's induced power diagram partition and results from the feature-learning regime. Motivated by these theoretical insights, we introduce GrokAlign, a regularization strategy that actively induces normal alignment. We demonstrate that GrokAlign significantly accelerates the training dynamics of deep networks relevant to the grokking phenomenon. Furthermore, we apply the principle of normal alignment to Recursive Feature Machines (RFMs) to introduce Recursive Feature Alignment Machines (RFAMs). We show that RFAMs exhibit greater adversarial robustness compared to RFMs when trained on tabular data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。