Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MobiLlama: Towards Accurate and Lightweight Fully Transparent GPT

About

"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.

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

Related benchmarks

TaskDatasetResultRank
Question AnsweringARC Challenge
Accuracy36.7
906
Multi-task Language UnderstandingMMLU
Accuracy25.3
876
Commonsense ReasoningPIQA
Accuracy74.8
751
Question AnsweringARC Easy
Normalized Acc56.7
389
Boolean Question AnsweringBoolQ
Accuracy60.3
323
Question AnsweringOBQA
Accuracy35.4
300
Question AnsweringSciQ
Accuracy89.1
283
Commonsense ReasoningSIQA
Accuracy44.7
106
Multi-level multi-discipline evaluationC-Eval
Accuracy26.2
28
Showing 9 of 9 rows

Other info

Follow for update