arXiv:2506.04051cs.CLcs.AI2025-06

让大模型只在有把握时才回答,否则部分放弃,提升可靠性。

High Accuracy, Less Talk (HALT): Reliable LLMs through Capability-Aligned Finetuning

  • 通过拆分预训练模型的回答并标记错误片段,生成能力对齐的微调数据。
  • 平均正确率提升15%,F1分数相比基线提高4%,兼顾完整性和准确性。
  • 可调节阈值,适合对可靠性要求高的医疗、数学等专业场景使用。

当前大语言模型对所有提示都会回应,但在缺乏知识或能力时会产生幻觉。本文提出HALT方法,使模型仅在有把握时生成内容,否则部分放弃。具体做法是将预训练模型的回答拆分为事实片段,利用真实信息识别错误片段,并根据可调阈值移除或替换为“此处不确定”。在生物写作、数学、编程和医学四个领域对四个开源模型进行微调,测试三种权衡阈值。结果表明,HALT在保持响应完整性的同时,平均正确率提升15%,F1分数(完整性和正确性均值)较基线提升4%。通过优化最高正确率,单个Llama3-70B模型在所有领域正确率从51%提升至87%,同时保留标准微调下53%的响应完整性。

原文摘要 · Abstract (English)

Large Language Models (LLMs) currently respond to every prompt. However, they can produce incorrect answers when they lack knowledge or capability -- a problem known as hallucination. We instead propose post-training an LLM to generate content only when confident in its correctness and to otherwise (partially) abstain. Specifically, our method, HALT, produces capability-aligned post-training data that encodes what the model can and cannot reliably generate. We generate this data by splitting responses of the pretrained LLM into factual fragments (atomic statements or reasoning steps), and use ground truth information to identify incorrect fragments. We achieve capability-aligned finetuning responses by either removing incorrect fragments or replacing them with "Unsure from Here" -- according to a tunable threshold that allows practitioners to trade off response completeness and mean correctness of the response's fragments. We finetune four open-source models for biography writing, mathematics, coding, and medicine with HALT for three different trade-off thresholds. HALT effectively trades off response completeness for correctness, increasing the mean correctness of response fragments by 15% on average, while resulting in a 4% improvement in the F1 score (mean of completeness and correctness of the response) compared to the relevant baselines. By tuning HALT for highest correctness, we train a single reliable Llama3-70B model with correctness increased from 51% to 87% across all four domains while maintaining 53% of the response completeness achieved with standard finetuning.

大模型可靠幻觉抑制微调方法

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