arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2512.19728cs.LG

增强的DPO后训练用于小型语言模型

Hard Negative Sample-Augmented DPO Post-Training for Small Language Models

Computer Science NYU Shanghai(纽约大学上海学院)

Haocheng Lu, Minjun Zhu, Henry Yu

AI总结:

本文提出一种轻量级后训练方法,通过MathVerifier检测结构化错误,改进DPO以提升小型模型在数学推理中的表现。

摘要:

大型语言模型(LLMs)在数学推理上仍存在困难,传统后训练流程常将生成的解决方案简化为二元结果:正确或错误。这种观点在实践中受限,因为链式推理(CoT)中的失败通常是结构化的;解决方案可能看似有说服力,但包含细微的逻辑、代数或数值错误。同时,依赖大型奖励模型或LLM-as-a-judge信号的强化学习从人类反馈(RLHF)变体往往昂贵、难以扩展且迭代不稳定。本文提出一种轻量级且务实的后训练流程,旨在在现实计算预算下针对此类结构化错误。从MetaMathQA样式的CoT数据进行监督微调(SFT)开始,我们引入了一个紧凑的MathVerifier,将候选解决方案分解为六维错误配置文件,并将其汇总为可解释的错误性和荒谬性评分。这些验证器信号起到两个作用:(i)挖掘接近正确但结构上错误的硬负样本,(ii)定义每样本的重要性权重,强调最有信息量的偏好对。我们将两者整合到一个离线直接偏好优化(DPO)目标中,通过验证器引导的加权公式。在1.5B参数Qwen2.5模型上的实验表明,验证器引导的加权DPO比普通SFT和无权DPO产生更精确的改进,特别是在解决方案在数值上接近正确但逻辑不一致的问题上,同时避免了训练大型奖励模型或依赖外部评判者的开销。

英文摘要:

Large language models (LLMs) continue to struggle with mathematical reasoning, and common post-training pipelines often reduce each generated solution to a binary outcome: correct or incorrect. This perspective is limiting in practice, as failures in chain-of-thought (CoT) reasoning are frequently structured; solutions may appear convincing while containing subtle logical, algebraic, or numerical flaws. Meanwhile, reinforcement learning from human feedback (RLHF) variants that rely on large reward models or LLM-as-a-judge signals are often expensive, difficult to scale, and unstable to iterate. We propose a lightweight and pragmatic post-training pipeline that targets such structured errors under realistic compute budgets. Starting from supervised fine-tuning (SFT) on MetaMathQA-style CoT data, we introduce a compact MathVerifier that decomposes a candidate solution into a six-dimensional error profile and aggregates it into interpretable wrongness and absurdity scores. These verifier signals serve two roles: (i) mining hard negatives that are near-correct yet structurally flawed, and (ii) defining per-sample importance weights that emphasize the most informative preference pairs. We integrate both into an offline Direct Preference Optimization (DPO) objective via a verifier-guided weighted formulation. Experiments on a 1.5B-parameter Qwen2.5 model show that verifier-guided, weighted DPO yields more targeted improvements than vanilla SFT and unweighted DPO, particularly on problems where solutions are numerically close to correct but logically inconsistent, while avoiding the overhead of training large reward models or relying on external judges.

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

↑