用团队对抗机制提升大模型对齐效果,无需人工标注。
Team-Based Self-Play With Dual Adaptive Weighting for Fine-Tuning LLMs

- 让当前模型与历史版本既合作又竞争,稳定训练过程。
- 通过自适应加权调整响应和成员贡献,提升优化效率。
- 适合追求高质量对齐且无标注数据的模型迭代场景。
现有自训练方法虽减少对人工标注数据的依赖,但仍面临两大问题:(i) 对合成数据质量敏感,导致迭代训练中不稳定与偏差放大;(ii) 随训练迭代,正负回应差距缩小,优化效果下降。本文提出团队自博弈双自适应加权算法(TPAW),在全自监督设置下改进对齐效果。TPAW采用团队框架,使当前策略模型与历史检查点既协作又竞争,促进更稳定高效的优化。设计两种自适应加权机制:(i) 响应重加权,动态调整目标响应权重;(ii) 玩家加权,实时调节各成员贡献。从SFT模型初始化,TPAW可无需额外人工监督持续优化对齐。实验表明,其在多种基础模型与LLM基准上均优于现有基线。代码已公开于https://github.com/lab-klc/TPAW。
原文摘要 · Abstract (English)
While recent self-training approaches have reduced reliance on human-labeled data for aligning LLMs, they still face critical limitations: (i) sensitivity to synthetic data quality, leading to instability and bias amplification in iterative training; (ii) ineffective optimization due to a diminishing gap between positive and negative responses over successive training iterations. In this paper, we propose Team-based self-Play with dual Adaptive Weighting (TPAW), a novel self-play algorithm designed to improve alignment in a fully self-supervised setting. TPAW adopts a team-based framework in which the current policy model both collaborates with and competes against historical checkpoints, promoting more stable and efficient optimization. To further enhance learning, we design two adaptive weighting mechanisms: (i) a response reweighting scheme that adjusts the importance of target responses, and (ii) a player weighting strategy that dynamically modulates each team member's contribution during training. Initialized from a SFT model, TPAW iteratively refines alignment without requiring additional human supervision. Experimental results demonstrate that TPAW consistently outperforms existing baselines across various base models and LLM benchmarks. Our code is publicly available at https://github.com/lab-klc/TPAW.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。