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

PocketLLM:为个性化大语言模型实现端侧微调

PocketLLM: Enabling On-Device Fine-Tuning for Personalized LLMs

OPPO Research Institute(OPPO研究院)

Dan Peng, Zhihui Fu, Jun Wang

AI总结:

PocketLLM 提出用无梯度优化替代基于梯度的优化,以降低内存需求,使 RoBERTa-large 和 OPT-1.3B 能在 OPPO Reno 6 手机上分别以约 4GB 和 6.5GB 内存完成端侧微调,实现资源受限设备上的个性化大语言模型并保护数据隐私。

摘要:

大语言模型(LLMs)的近期进展确实展示了其令人瞩目的能力。在移动设备上,每天产生的大量有价值的非公开数据为本地微调个性化大语言模型带来了巨大前景,同时通过端侧处理保护隐私。然而,移动设备资源的限制对直接的端侧大语言模型微调构成了挑战,这主要是由于基于梯度的优化需要保存梯度和优化器状态,具有较高的内存占用。为了解决这一问题,我们提出采用无梯度优化技术,即使在内存受限的移动设备上也能实现大语言模型的端侧微调。实验结果表明,采用无梯度优化技术,RoBERTa-large 模型和 OPT-1.3B 可以分别在 OPPO Reno 6 智能手机上使用约 4GB 和 6.5GB 的内存进行本地微调。这凸显了在移动设备上进行端侧大语言模型微调的可行性,为在资源受限设备上实现个性化大语言模型铺平了道路,同时保障数据隐私。

英文摘要:

Recent advancements in large language models (LLMs) have indeed showcased their impressive capabilities. On mobile devices, the wealth of valuable, non-public data generated daily holds great promise for locally fine-tuning personalized LLMs, while maintaining privacy through on-device processing. However, the constraints of mobile device resources pose challenges to direct on-device LLM fine-tuning, mainly due to the memory-intensive nature of derivative-based optimization required for saving gradients and optimizer states. To tackle this, we propose employing derivative-free optimization techniques to enable on-device fine-tuning of LLM, even on memory-limited mobile devices. Empirical results demonstrate that the RoBERTa-large model and OPT-1.3B can be fine-tuned locally on the OPPO Reno 6 smartphone using around 4GB and 6.5GB of memory respectively, using derivative-free optimization techniques. This highlights the feasibility of on-device LLM fine-tuning on mobile devices, paving the way for personalized LLMs on resource-constrained devices while safeguarding data privacy.

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

↑