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

从自生成响应中学习:用于推理的可靠性-不确定性自蒸馏

Learning from Self-Generated Responses: Reliability--Uncertainty Self-Distillation for Reasoning

George Mason University(乔治梅森大学)

Jinhao Pan, Bowen Wei, Yuqing Zhou, Ziwei Zhu

AI总结:

该研究提出RUSD方法,利用大语言模型自身的解题尝试,通过结合判定可靠性与学生预测不确定性加权蒸馏损失,在11个推理基准上显著提升了Qwen3系列模型的推理准确率。

摘要:

参考解决方案和特定问题的评分规则并非总是可用于指导自蒸馏中的教师。大语言模型自身的解题尝试提供了一种替代方案,但在开放式任务中若无自动正确性检查,其质量难以评估,且即使正确的尝试也可能包含不确定的 token 预测。我们提出可靠性-不确定性自蒸馏(\texttt{Reliability--Uncertainty Self-Distillation,简称 RUSD})以从这些尝试中学习。同一模型同时充当教师和学生:教师接收问题和完整的解题尝试,学生使用仅含问题的提示匹配教师的 token 级预测。\texttt{RUSD} 将每次尝试的蒸馏损失通过结合从自评判器的分数与置信度估计的判定可靠性,以及学生预测不确定性的方式进行加权。在11个推理基准上,\texttt{RUSD} 的平均准确率优于全部5个基线,在 Qwen3-4B 上达到70.47%,在 Qwen3-8B 上达到76.69%(较基础模型分别高出8.90和2.10个百分点)。数据和代码可在 https://github.com/JP-25/RUSD 获取。

英文摘要:

Reference solutions and question-specific rubrics are not always available to guide the teacher in self-distillation. A large language model's own solution attempts offer an alternative, but their quality is difficult to assess in open-ended tasks without automatic correctness checks, and even correct attempts can involve uncertain token predictions. We propose Reliability--Uncertainty Self-Distillation (\method{}) to learn from these attempts. The same model serves as teacher and student: the teacher receives the question and completed attempt, and the student matches the teacher's token-level predictions using a question-only prompt. \method{} weights each attempt's distillation loss by additively combining judged reliability, estimated from a self-judge's score and confidence, with student predictive uncertainty. Across eleven reasoning benchmarks, \method{} outperforms all five baselines in average accuracy, reaching 70.47\% on Qwen3-4B and 76.69\% on Qwen3-8B (8.90 and 2.10 percentage points above the base models). Data and code are available at https://github.com/JP-25/RUSD.

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

↑