用模型生成替代不靠谱的人类标注,能更好提升大模型性能
Iterative Label Refinement Matters More than Preference Optimization under Weak Supervision
- 通过对比反馈判断人类标注是否需替换为模型生成内容,迭代优化训练数据
- 在数学、编程等任务上,新方法比传统强化学习方法性能提升23%以上
- 适合人类标注不可靠的复杂任务场景,尤其对高阶能力训练有优势
大语言模型后训练依赖两个阶段的人类监督:示范数据用于有监督微调(SFT),偏好比较用于从人类反馈中强化学习(RLHF)。随着模型能力增强,任务难度上升,人类监督可靠性下降。本文通过小模型和限时人类模拟不可靠的示范与比较反馈,发现当监督不可靠时,SFT仍有一定效果,但常见算法DPO无法超越SFT。为此提出迭代标签精炼(ILR):利用比较反馈判断人类示范是否应由模型生成的替代方案取代,再基于更新后的数据重新进行SFT。SFT+ILR在数学、编程及安全指令遵循任务上优于SFT+DPO。结果表明,当人类监督不可靠时,应将偏好反馈用于改进训练数据而非持续训练模型,这可能更有效。代码与数据已公开于https://github.com/helloelwin/iterative-label-refinement。
原文摘要 · Abstract (English)
Language model (LM) post-training relies on two stages of human supervision: task demonstrations for supervised finetuning (SFT), followed by preference comparisons for reinforcement learning from human feedback (RLHF). As LMs become more capable, the tasks they are given become harder to supervise. Will post-training remain effective under unreliable supervision? To test this, we simulate unreliable demonstrations and comparison feedback using small LMs and time-constrained humans. We find that in the presence of unreliable supervision, SFT still retains some effectiveness, but DPO (a common RLHF algorithm) fails to improve the model beyond SFT. To address this, we propose iterative label refinement (ILR) as an alternative to RLHF. ILR improves the SFT data by using comparison feedback to decide whether human demonstrations should be replaced by model-generated alternatives, then retrains the model via SFT on the updated data. SFT+ILR outperforms SFT+DPO on several tasks with unreliable supervision (math, coding, and safe instruction-following). Our findings suggest that as LMs are used for complex tasks where human supervision is unreliable, RLHF may no longer be the best use of human comparison feedback; instead, it is better to direct feedback towards improving the training data rather than continually training the model. Our code and data are available at https://github.com/helloelwin/iterative-label-refinement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。