SPINAL揭示大模型对齐时关键修正集中在最后几层。
SPINAL -- Scaling-law and Preference Integration in Neural Alignment Layers
- 通过逐层追踪结构变化,量化对齐过程中的几何演化
- 对齐模型在最后几层出现收缩增强与传输平滑,表明策略稳定
- 适合关注模型对齐机制、训练异常检测的研究者
直接偏好优化(DPO)是一种从成对偏好中对齐大语言模型的可扩展方法,但其内部几何特征仍不清晰,限制了审计、检查点对比和故障预测。我们提出SPINAL(神经对齐层中的缩放律与偏好整合),一种诊断工具,通过逐层追踪结构变化,测量对齐如何重塑表示。在多个模型族中,DPO导致最终解码器块(通常为第21-30层)出现集中式层间校准效应,偏好梯度在此最直接影响下一个词分布。SPINAL将每个检查点编码为深度轨迹:(层索引, 收缩得分, 传输得分)。收缩得分反映层谱尾部衰减速率(小模式消失速度);得分越高,表示向更少有效方向收缩越强。传输得分基于有界重叠度量,反映相邻层间分布转移程度;得分越低,表示路径越短越平滑。对齐检查点显示后期层收缩上升、传输下降,符合策略质量收紧与稳定化;未对齐模型则呈现高曲率、高熵且几何不一致的深度路径。总体上,对齐具有几何局部性:最终层编码主导的偏好修正。SPINAL将此局部性转化为实用审计信号,量化对齐集中位置、强度及训练中失稳时间。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) is a principled, scalable alternative to RLHF for aligning large language models from pairwise preferences, but its internal geometric footprint remains undercharacterized, limiting audits, checkpoint comparisons, and failure prediction. We introduce SPINAL (Scaling-law and Preference Integration in Neural Alignment Layers), a diagnostic that measures how alignment reshapes representations across depth by tracing localized structural change layer by layer. Across model families, DPO produces a layerwise calibration effect concentrated in the final decoder blocks (often layers 21-30), where preference gradients most directly affect the next-token distribution. SPINAL encodes each checkpoint as a depth trace over (layer index, contraction score, transport score). The contraction score summarizes how quickly the tail of a layer's spectrum decays (how fast small modes vanish); higher values indicate stronger contraction into fewer effective directions. The transport score summarizes how much the token distribution shifts between adjacent layers using a bounded overlap measure; lower values indicate shorter, smoother steps through representation space. Aligned checkpoints show a late-layer ramp-up in contraction and a smooth reduction in transport, consistent with tightened and stabilized policy mass, while unaligned models trace higher-curvature, more entropic, and geometrically incoherent depth paths. Overall, alignment is geometrically localized: the final layers encode the dominant preference-induced corrections. SPINAL turns this localization into a practical audit signal, quantifying where alignment concentrates, how strongly it manifests, and when it begins to destabilize during training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。