#183 · 主分类: 深度学习框架
MFTCoder
面向代码大语言模型的高精度、高效多任务微调框架。该工作已被KDD 2024接收。
项目最后更新:12/30/24
GitHub Stars
711
Forks数量
69
贡献者数量
12
许可证
Apache-2.0
收录理由
MFTCoder 处理代码模型微调的方式有些特别:它不把模型逐个任务地单独适配,而是把多个相关的代码任务放在一起训练,项目方认为这样既能提升准确率,也能提高效率。框架提供了两条训练路线,一条基于 PEFT,另一条基于 accelerate,其中 accelerate 路线支持 QLoRA 搭配 DeepSpeed ZeRO-3 或 FSDP,即便没有大型 GPU 集群,也能对超大模型进行微调。最新版本还加入了 DPO、RPO、ORPO 这类偏好对齐方法,用来调整模型的输出行为,而不只是教它新任务。两篇经过同行评审的论文分别发表在 KDD 2024 和 EMNLP 2024,详细解释了多任务训练和收敛平衡的具体做法,如果你想知道代码背后的理论依据,而不只是拿个脚本跑一遍,这些资料会很有帮助。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。