计算机科学 > 机器学习
[提交于 2024年2月1日
]
标题: EE-Tuning:一种经济且可扩展的早期退出大型语言模型调优方案
标题: EE-Tuning: An Economical yet Scalable Solution for Tuning Early-Exit Large Language Models
摘要: 这项工作介绍了EE-Tuning,这是一种轻量且经济的解决方案,用于训练/调整早期退出大型语言模型(LLMs)。 与常见的全参数预训练方法不同,EE-Tuning通过添加额外的早期退出层来增强任何预训练(可能经过微调)的标准LLM,并以参数高效的方式进行调整,这需要显著更少的计算资源和训练数据。 我们的EE-Tuning实现通过广泛的性能优化实现了出色的训练效率,并由于其与3D并行性的完全兼容性而具有可扩展性。 系统实验的结果验证了EE-Tuning的有效性,确认在有限的训练预算下可以实现有效的早期退出LLM推理。 希望使早期退出LLMs对社区可用,我们将在https://github.com/pan-x-c/EE-LLM发布我们的EE-Tuning实现的源代码。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.