Skip to main content
CenXiv.org
此网站处于试运行阶段,支持我们!
我们衷心感谢所有贡献者的支持。
贡献
赞助
cenxiv logo > cs > arXiv:2402.00518v1

帮助 | 高级搜索

计算机科学 > 机器学习

arXiv:2402.00518v1 (cs)
[提交于 2024年2月1日 ]

标题: EE-Tuning:一种经济且可扩展的早期退出大型语言模型调优方案

标题: EE-Tuning: An Economical yet Scalable Solution for Tuning Early-Exit Large Language Models

Authors:Xuchen Pan, Yanxi Chen, Yaliang Li, Bolin Ding, Jingren Zhou
摘要: 这项工作介绍了EE-Tuning,这是一种轻量且经济的解决方案,用于训练/调整早期退出大型语言模型(LLMs)。 与常见的全参数预训练方法不同,EE-Tuning通过添加额外的早期退出层来增强任何预训练(可能经过微调)的标准LLM,并以参数高效的方式进行调整,这需要显著更少的计算资源和训练数据。 我们的EE-Tuning实现通过广泛的性能优化实现了出色的训练效率,并由于其与3D并行性的完全兼容性而具有可扩展性。 系统实验的结果验证了EE-Tuning的有效性,确认在有限的训练预算下可以实现有效的早期退出LLM推理。 希望使早期退出LLMs对社区可用,我们将在https://github.com/pan-x-c/EE-LLM发布我们的EE-Tuning实现的源代码。
摘要: This work introduces EE-Tuning, a lightweight and economical solution to training/tuning early-exit large language models (LLMs). In contrast to the common approach of full-parameter pre-training, EE-Tuning augments any pre-trained (and possibly fine-tuned) standard LLM with additional early-exit layers that are tuned in a parameter-efficient manner, which requires significantly less computational resources and training data. Our implementation of EE-Tuning achieves outstanding training efficiency via extensive performance optimizations, as well as scalability due to its full compatibility with 3D parallelism. Results of systematic experiments validate the efficacy of EE-Tuning, confirming that effective early-exit LLM inference can be achieved with a limited training budget. In hope of making early-exit LLMs accessible to the community, we release the source code of our implementation of EE-Tuning at https://github.com/pan-x-c/EE-LLM.
主题: 机器学习 (cs.LG) ; 人工智能 (cs.AI); 计算与语言 (cs.CL)
引用方式: arXiv:2402.00518 [cs.LG]
  (或者 arXiv:2402.00518v1 [cs.LG] 对于此版本)
  https://doi.org/10.48550/arXiv.2402.00518
通过 DataCite 发表的 arXiv DOI

提交历史

来自: Yanxi Chen [查看电子邮件]
[v1] 星期四, 2024 年 2 月 1 日 11:39:04 UTC (1,413 KB)
全文链接:

获取论文:

    查看标题为《》的 PDF
  • 查看中文 PDF
  • 查看 PDF
  • TeX 源代码
查看许可
当前浏览上下文:
cs.LG
< 上一篇   |   下一篇 >
新的 | 最近的 | 2024-02
切换浏览方式为:
cs
cs.AI
cs.CL

参考文献与引用

  • NASA ADS
  • 谷歌学术搜索
  • 语义学者
a 导出 BibTeX 引用 加载中...

BibTeX 格式的引用

×
数据由提供:

收藏

BibSonomy logo Reddit logo

文献和引用工具

文献资源探索 (什么是资源探索?)
连接的论文 (什么是连接的论文?)
Litmaps (什么是 Litmaps?)
scite 智能引用 (什么是智能引用?)

与本文相关的代码,数据和媒体

alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)

演示

复制 (什么是复制?)
Hugging Face Spaces (什么是 Spaces?)
TXYZ.AI (什么是 TXYZ.AI?)

推荐器和搜索工具

影响之花 (什么是影响之花?)
核心推荐器 (什么是核心?)
IArxiv 推荐器 (什么是 IArxiv?)
  • 作者
  • 地点
  • 机构
  • 主题

arXivLabs:与社区合作伙伴的实验项目

arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。

与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。

有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.

这篇论文的哪些作者是支持者? | 禁用 MathJax (什么是 MathJax?)
  • 关于
  • 帮助
  • contact arXivClick here to contact arXiv 联系
  • 订阅 arXiv 邮件列表点击这里订阅 订阅
  • 版权
  • 隐私政策
  • 网络无障碍帮助
  • arXiv 运营状态
    通过...获取状态通知 email 或者 slack

京ICP备2025123034号