让大模型学会自我判断是否该回答,提升指令遵循的可靠性。
Self-Judge: Selective Instruction Following with Alignment Self-Evaluation
- 用自训练框架让模型从自身生成中学习判断回答质量
- 在5个开源模型上与GPT-4评分相关性显著高于基线
- 可作奖励模型,提升对话模型得分至92.48(AlpacaEval v1)
预训练大语言模型可通过指令微调适应人类指令,但测试数据分布偏移可能导致其无法准确执行指令,产生事实错误或内容偏差。为提高模型指令遵循的可靠性,本文提出选择性指令遵循:当预期响应质量低时,系统主动拒绝执行。我们训练判别模型以预测响应的质量分数。针对标注数据稀缺问题,提出Self-J自训练框架,无需人工标注即可构建判别模型。该方法利用模型自身自评估能力,从有标注的指令微调数据中提取响应质量信息,并引入标准答案辅助自评估,通过语义相似度重新校准。训练阶段采用自蒸馏作为正则化手段,增强无参考估计能力。为验证通用指令遵循任务上的对齐评估效果,我们在Hugging Face收集大规模高质量指令数据用于训练与评估。大量实验表明,本方法在五个开源模型上与GPT-4评分相关性远超强基线,如从GPT-4和GPT-3.5-turbo蒸馏的监督模型。分析显示模型具备强大跨领域泛化能力。此外,该判别模型作为奖励模型表现优异,使用最佳32样本采样策略,使WizardLM-13B-V1.2在AlpacaEval v1和v2版本上的得分分别从89.17提升至92.48,从12.03提升至15.90。
原文摘要 · Abstract (English)
Pre-trained large language models (LLMs) can be tailored to adhere to human instructions through instruction tuning. However, due to shifts in the distribution of test-time data, they may not always execute instructions accurately, potentially generating factual errors or misaligned content when acting as chat assistants. To enhance the reliability of LLMs in following instructions, we propose the study of selective instruction following, whereby the system declines to execute instructions if the anticipated response quality is low. We train judge models that can predict numerical quality scores for model responses. To address data scarcity, we introduce Self-J, a novel self-training framework for developing judge models without needing human-annotated quality scores. Our method leverages the model's inherent self-evaluation capability to extract information about response quality from labeled instruction-tuning data. It incorporates a gold reference answer to facilitate self-evaluation and recalibrates by assessing the semantic similarity between the response sample and the gold reference. During the training phase, we implement self-distillation as a regularization technique to enhance the capability of reference-free estimation. To validate alignment evaluation on general instruction-following tasks, we collect large-scale high-quality instructions from Hugging Face for model training and evaluation. Extensive experiments on five open-source models show that our method correlates much more with GPT-4 than strong baselines, e.g., supervised models distilled from GPT-4 and GPT-3.5-turbo. Our analysis shows our model's strong generalization across domains. Additionally, our judge models serve as good reward models, e.g., boosting WizardLM-13B-V1.2 from 89.17 to 92.48 and from 12.03 to 15.90 in version v1 and v2 of AlpacaEval respectively using best-of-32 sampling with our judge models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。