AutoMixAlign: 多任务偏好优化中的自适应数据配比
多任务偏好对齐(helpfulness、harmlessness、coding 等)面临一个工程痛点:数据配比怎么选?均匀混合容易被大数据集主导,按任务等权又可能在简单任务上浪费容量。常见做法是跑大量消融实验或凭经验设定比例,成本高且不一定最优。
AutoMixAlign(AutoMixAlign: Adaptive Data Mixing for Multi-Task Preference Optimization in LLMs,ACL 2025)的做法是:先为每个任务单独训练 specialist 模型作为参照基线,再在训练 generalist 模型时基于 minimax 优化动态调整各任务的权重或采样比例,优先关注那些 generalist 与 specialist 之间 loss 差距最大的任务。在 Zephyr-7B 上的实验中,多任务平均性能最高提升约 9.42%,且避免了标准 DPO 混合训练中常见的"某个任务退化"问题。