跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

离线推理训练的权重空间几何

文章摘要

该论文研究了六种离线强化学习损失函数(SFT、RFT、DFT、RIFT、Offline GRPO、DPO)在推理蒸馏中的权重更新几何特性。实验基于Qwen3-4B模型和相同数学数据,发现SFT、RFT和RIFT的权重增量近乎共线,DFT偏离较大,Offline GRPO增加了正交分量,而DPO位于近乎正交的子空间且准确率最高,但存在模式连接障碍。

来源arXiv Machine Learning作者: Aleksandr Nikolich, Igor Kiselev, Vladimir Platonov, Karina Romanova
离线推理训练的权重空间几何
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

离线强化学习损失函数(如RFT、RIFT、DFT、Offline GRPO和DPO)被广泛用于将大型教师模型的推理能力蒸馏到较小的学生模型中。然而,这些方法通常仅通过下游准确率进行比较,其机制上的差异尚不明确。一项新的研究直接比较了六种方法(SFT、RFT、DFT、RIFT、Offline GRPO和DPO)在相同数学数据上的权重更新几何特性。

研究人员使用Qwen3-4B作为基础模型,采用仅注意力的LoRA适配器,对所有方法使用相同的数学推理数据(来自教师模型生成)。他们通过余弦相似度、主角度子空间分析、线性模式连接性和CKA(Centered Kernel Alignment)等方法分析了训练前后的权重增量。结果揭示了显著的结构差异。

首先,SFT(监督微调)、RFT(拒绝微调)和RIFT(奖励加权拒绝微调)的权重增量几乎共线,余弦相似度超过0.97,中位主角度约7度(在144个模块中)。这三种方法在GSM8K数学基准测试上的准确率相当(约87-88%),且配对McNemar检验的p值≥0.15,表明无显著差异。这表明RFT和RIFT可能并未通过奖励加权引入独立的机制。

相比之下,尽管使用相同的训练数据,DFT(直接偏好微调)的更新方向明显偏离奖励加权方法。Offline GRPO(群体相对策略优化的一种离线变体)则表现出独特的性质:它在保持SFT损失盆地的同时,增加了大量与SFT方向正交的分量(全局约67%,在后期层中高达86%),但不破坏损失收敛性。

最引人注目的是DPO(直接偏好优化)。其权重增量几何与其他方法几乎正交,在后期层中CKA降至约0.46,甚至出现模式连接障碍。尽管几何上隔离,DPO在实验中取得了最高准确率:GSM8K上93.5%,AIME26上30.0%(其他方法为3.3-10.0%)。但值得注意的是,DPO使用了比其他方法小10倍的学习率(标准惯例),因此准确率差异可能同时来自损失函数和优化器选择。作者将学习率匹配后的比较留作未来工作。

这项研究强调了离线推理训练方法在权重空间中的显著几何差异,其中一些差异可能源于损失函数设计,而非纯粹的监督信号。理解这些几何性质有助于更好地选择蒸馏策略和优化超参数。

展开要点与分析

文章情报

工程师进阶

要点

  • SFT、RFT和RIFT的权重余弦相似度≥0.97,GSM8K准确率约87-88%。
  • DFT的更新方向比任何奖励加权方法都更发散。
  • Offline GRPO在保持SFT损失盆地的同时,增加了约67%的正交分量(晚期层达86%)。
  • DPO位于近乎正交的子空间,GSM8K准确率达93.5%,AIME26达30%,但存在模式连接障碍。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。