通过用户反馈优化模型训练数据,让AI在未知任务上表现更好。
DUET: Optimizing Training Data Mixtures via Feedback from Unseen Evaluation Tasks
- 用影响函数+贝叶斯优化动态筛选和混合训练数据
- 在无任务数据的情况下仍能逼近最优数据组合
- 适合无法预知评估任务的现实场景,如加密对话
大语言模型性能高度依赖训练数据与下游任务的相关性。然而实际中,未见评估任务的数据往往未知(如用户与模型的对话端到端加密)。因此,难以判断哪些数据对微调模型以提升特定未见任务表现最有效。此时只能将模型部署到该任务,通过多轮反馈(如用户评分)来评估表现。本文提出DUET,一种全局到局部的算法,将影响函数作为数据选择方法,结合贝叶斯优化,基于特定未见任务的反馈优化训练数据混合比例。理论分析表明,尽管无任务数据信息,DUET的累积遗憾仍可收敛至最优数据混合。多个语言任务实验显示,相比现有方法,DUET在未见任务设置下表现更优。
原文摘要 · Abstract (English)
The performance of an LLM depends heavily on the relevance of its training data to the downstream evaluation task. However, in practice, the data involved in an unseen evaluation task is often unknown (e.g., conversations between an LLM and a user are end-to-end encrypted). Hence, it is unclear what data are relevant for fine-tuning the LLM to maximize its performance on the specific unseen evaluation task. Instead, one can only deploy the LLM on the unseen task to gather multiple rounds of feedback on how well the model performs (e.g., user ratings). This novel setting offers a refreshing perspective towards optimizing training data mixtures via feedback from an unseen evaluation task, which prior data mixing and selection works do not consider. Our paper presents DUET, a novel global-to-local algorithm that interleaves influence function as a data selection method with Bayesian optimization to optimize data mixture via feedback from a specific unseen evaluation task. By analyzing DUET's cumulative regret, we theoretically show that DUET converges to the optimal training data mixture for an unseen task even without any data knowledge of the task. Finally, our experiments across a variety of language tasks demonstrate that DUET outperforms existing data selection and mixing methods in the unseen-task setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。