arXiv 研究:DPO 超参数随模型规模变化,且不可跨数据集迁移
The Shape of the DPO Hyperparameter Landscape: What to Measure, How It Moves with Scale, and When Not to Transfer It
训练 DPO 的朋友可以看看:学习率要按模型大小重新调,验证损失也不靠谱,文章还给了一套省算力的调参方法。
一项 arXiv 研究用密集超参数网格实验,分析 DPO 训练中学习率与参考策略正则化参数 β 如何随模型规模和更新步数变化。实验发现最优学习率随模型大小改变,超参数在不同训练数据集之间不可迁移。验证损失与下游基准上的最优超参数仅弱相关,不是好的选择代理。研究表明对中间 checkpoint 的短期候选训练能大幅降低超参数搜索成本,作者已公开全部训练配置和评估结果。
The Shape of the DPO Hyperparameter Landscape: What to Measure, How It Moves with Scale, and When Not to Transfer It
Direct preference optimisation (DPO) is widely used for language model post-training, but its performance depends strongly on the choice of its hyperparameters, such as the learning rate and the reference-policy regularisation parameter $β$. To better understand the DPO hyperparameter landscape, we conduct extensive experiments using a dense hyperparameter grid across different model scales from the same family. This allows us to analyse how the optimal DPO hyperparameters evolve with model size and the number of update steps, with intermediate checkpoints providing an inexpensive signal to accelerate the hyperparameter optimisation process. Furthermore, by evaluating our models on both static and dynamic benchmarks, we identify which benchmarks provide the strongest signal for hyperparameter selection. Finally, we investigate how different state-of-the-art hyperparameter optimisation strategies perform in this setting. Our experiments show that the optimal learning rate varies with model size and that hyperparameters are not transferable across training datasets. We also find that validation loss is a poor proxy for hyperparameter optimisation, as hyperparameters that minimize validation loss only weakly correlate with optimal hyperparameters for downstream benchmarks. Therefore, DPO hyperparameters need to be tuned specifically for the target model, dataset, and evaluation objective. We also show that short candidate training runs can substantially reduce the cost of this selection. We release all our training configurations and evaluation results to support reproducible analysis of selection quality and training cost.