arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.16840cs.CL
收藏

MobiLlama:迈向准确且轻量的全透明GPT

MobiLlama: Towards Accurate and Lightweight Fully Transparent GPT

Mohamed bin Zayed University of AI(穆罕默德·本·扎耶德人工智能大学);Australian National University(澳大利亚国立大学);Aalto University(阿尔托大学);The University of Melbourne(墨尔本大学);Linköping University(林雪平大学)

Omkar Thawakar, Ashmal Vayani, Salman Khan, Hisham Cholakal, Rao M. Anwer, Michael Felsberg, Tim Baldwin, Eric P. Xing, Fahad Shahbaz Khan

AI总结:

本文提出0.5B参数的全透明开源小语言模型MobiLlama,通过从大模型出发并结合参数共享方案,在资源受限设备上实现准确且高效的部署。

摘要:

“越大越好”一直是近期大语言模型(LLMs)发展的主导趋势。然而,LLMs并不适合需要设备端处理、能效、低内存占用和响应效率的场景。这些要求对于隐私、安全和可持续部署至关重要。本文通过应对为资源受限设备设计准确且高效的小语言模型(SLMs)这一挑战,探索了“少即是多”的范式。我们的主要贡献是引入了一个准确且完全透明的开源5亿(0.5B)参数SLM,名为MobiLlama,旨在满足资源受限计算的特定需求,强调在降低资源需求的同时提升性能。MobiLlama是一种SLM设计,它从一个更大的模型出发,并应用谨慎的参数共享方案来降低预训练和部署成本。我们的工作不仅致力于弥合开源SLM的差距,还确保完全透明,完整的训练数据流水线、训练代码、模型权重以及超过300个检查点连同评估代码均可在 https://github.com/mbzuai-oryx/MobiLlama 获取。

英文摘要:

"Bigger the better" has been the predominant trend in recent Large Language Models (LLMs) development. However, LLMs do not suit well for scenarios that require on-device processing, energy efficiency, low memory footprint, and response efficiency. These requisites are crucial for privacy, security, and sustainable deployment. This paper explores the "less is more" paradigm by addressing the challenge of designing accurate yet efficient Small Language Models (SLMs) for resource constrained devices. Our primary contribution is the introduction of an accurate and fully transparent open-source 0.5 billion (0.5B) parameter SLM, named MobiLlama, catering to the specific needs of resource-constrained computing with an emphasis on enhanced performance with reduced resource demands. MobiLlama is a SLM design that initiates from a larger model and applies a careful parameter sharing scheme to reduce both the pre-training and the deployment cost. Our work strives to not only bridge the gap in open-source SLMs but also ensures full transparency, where complete training data pipeline, training code, model weights, and over 300 checkpoints along with evaluation codes is available at : https://github.com/mbzuai-oryx/MobiLlama.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑