让大模型一次训练就学会多目标权衡,灵活调整不同任务的优先级。
COS-DPO: Conditioned One-Shot Multi-Objective Fine-Tuning Framework
- 通过条件控制辅助目标权重,实现单次训练覆盖多目标优化
- 可在后训练阶段生成完整的帕累托前沿解,支持灵活调优
- 适合需要快速适配多任务场景的大模型部署应用
在大模型对齐及其他机器学习应用中,常面临多目标微调(MOFT)问题,即同时使用针对不同目标标注的数据集对已有模型进行微调。为解决该挑战,我们提出一种条件式单次多目标微调框架(COS-DPO),扩展了原本用于高效大模型对齐的直接偏好优化(DPO)技术,以适应多目标设定。通过直接对辅助目标权重进行条件控制,我们的Weight-COS-DPO方法实现了高效的单次训练过程,能够生成帕累托前沿的完整轮廓,并在后训练阶段实现全面的权衡解决方案。基于损失函数线性变换性质的理论分析,我们进一步提出Temperature-COS-DPO方法,将温度参数注入模型输入,增强对主目标与辅助目标间权衡的后训练控制灵活性。我们在多种任务中验证了COS-DPO框架的有效性和效率,包括学习排序(LTR)和大模型对齐任务,凸显其在大规模机器学习部署中的可行性。
原文摘要 · Abstract (English)
In LLM alignment and many other ML applications, one often faces the Multi-Objective Fine-Tuning (MOFT) problem, i.e., fine-tuning an existing model with datasets labeled w.r.t. different objectives simultaneously. To address the challenge, we propose a Conditioned One-Shot fine-tuning framework (COS-DPO) that extends the Direct Preference Optimization technique, originally developed for efficient LLM alignment with preference data, to accommodate the MOFT settings. By direct conditioning on the weight across auxiliary objectives, our Weight-COS-DPO method enjoys an efficient one-shot training process for profiling the Pareto front and is capable of achieving comprehensive trade-off solutions even in the post-training stage. Based on our theoretical findings on the linear transformation properties of the loss function, we further propose the Temperature-COS-DPO method that augments the temperature parameter to the model input, enhancing the flexibility of post-training control over the trade-offs between the main and auxiliary objectives. We demonstrate the effectiveness and efficiency of the COS-DPO framework through its applications to various tasks, including the Learning-to-Rank (LTR) and LLM alignment tasks, highlighting its viability for large-scale ML deployments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。