arXiv:2410.08898cs.LG2024-10ICML被引 2

揭示长度泛化难题本质,提出可实现泛化的关键条件。

Low-Dimension-to-High-Dimension Generalization And Its Implications for Length Generalization

  • 从低维到高维泛化视角解析长度泛化问题
  • 证明仅靠梯度下降无法实现泛化,需先验知识引导
  • 提出新位置编码方案,提升对数据格式的鲁棒性

低维到高维(LDHD)泛化是分布外(OOD)泛化的一种特殊形式,训练数据局限于高维测试空间的低维子空间。若每个实例由潜在变量生成,且潜在变量维度反映问题规模,则长度泛化中的固有缩放挑战可在潜在空间中通过LDHD泛化建模。我们理论证明:在无先验知识提供合适归纳偏置的情况下,LDHD泛化通常不可实现。具体研究布尔函数时发现,不同架构经(S)GD训练后收敛至相对于不同独立集的最小次数插值器。只有当目标函数与该归纳偏置一致时,泛化才可能实现。将此洞见应用于长度泛化,解释了思维链(CoT)通过改变潜在空间结构以促进更好泛化的机制。据此提出位置编码设计原则,以应对内在的LDHD泛化及数据格式等干扰因素。基于该原则,提出新型位置编码RPE-Square,有效缓解了数据格式带来的扰动。

原文摘要 · Abstract (English)

Low-Dimension-to-High-Dimension (LDHD) generalization is a special case of Out-of-Distribution (OOD) generalization, where the training data are restricted to a low-dimensional subspace of the high-dimensional testing space. Assuming that each instance is generated from a latent variable and the dimension of the latent variable reflects the problem scale, the inherent scaling challenge in length generalization can be captured by the LDHD generalization in the latent space. We theoretically demonstrate that LDHD generalization is generally unattainable without exploiting prior knowledge to provide appropriate inductive bias. Specifically, we explore LDHD generalization in Boolean functions. We verify that different architectures trained with (S)GD converge to \emph{min-degree interpolators w.r.t. different independent sets}. LDHD generalization is achievable if and only if the target function coincides with this inductive bias. Applying the insights from LDHD generalization to length generalization, we explain the effectiveness of CoT as changing the structure latent space to enable better LDHD generalization. We also propose a principle for position embedding design to handle both the inherent LDHD generalization and the nuisances such as the data format. Following the principle, we propose a novel position embedding called RPE-Square that remedies the RPE for dealing with the data format nuisance.

长度泛化归纳偏置位置编码潜在空间

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