用价值网络提升大模型后训练效果,可跨模型复用且性能接近全量微调
Transferable Post-training via Inverse Value Learning
- 构建价值网络学习后训练时的逻辑变化规律
- 在不同规模和词汇的模型间实现性能迁移,部分达全参数微调水平
- 适合资源有限但需快速适配新模型的研究者
随着后训练数据集和基础模型规模持续扩大,现有算法的计算开销与实施难度急剧上升。本文提出通过独立神经网络(即价值网络)建模后训练过程中对输出逻辑(logits)的影响。在小规模基础模型上使用示范数据训练该网络后,可无缝集成至其他预训练模型中,在推理阶段实现相似的能力提升。我们系统研究了该范式在预训练权重选择与连接方式上的最佳实践。结果表明,所获价值网络在同家族不同参数规模模型、持续预训练模型及跨家族异词汇模型间均具备广泛迁移能力;在某些情况下,其性能可媲美全参数微调。此外,我们还探索了提升价值模型迁移性并防止过拟合于训练基模型的方法。
原文摘要 · Abstract (English)
As post-training processes utilize increasingly large datasets and base models continue to grow in size, the computational demands and implementation challenges of existing algorithms are escalating significantly. In this paper, we propose modeling the changes at the logits level during post-training using a separate neural network (i.e., the value network). After training this network on a small base model using demonstrations, this network can be seamlessly integrated with other pre-trained models during inference, enables them to achieve similar capability enhancements. We systematically investigate the best practices for this paradigm in terms of pre-training weights and connection schemes. We demonstrate that the resulting value network has broad transferability across pre-trained models of different parameter sizes within the same family, models undergoing continuous pre-training within the same family, and models with different vocabularies across families. In certain cases, it can achieve performance comparable to full-parameter fine-tuning. Furthermore, we explore methods to enhance the transferability of the value model and prevent overfitting to the base model used during training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。