FastMix: Fast Data Mixture Optimization via Gradient Descent
FastMix: 通过梯度下降实现快速数据混合优化
机构 * University of Hong Kong(香港大学) ; Tencent(腾讯) ; Chinese University of Hong Kong(香港中文大学)
AI总结 提出FastMix框架,将数据混合选择重新表述为双层优化问题,通过联合优化混合系数和模型参数,实现高效、可扩展的数据混合发现,在预训练和后训练中均优于基线方法且大幅降低搜索成本。
Journal ref ICLR-2026