中文

通过自监督数据选择与合成实现设备端大语言模型个性化

计算与语言 2024-04-18 v4

摘要

在大语言模型(LLM)部署到边缘设备后,期望这些设备能够从用户生成的对话数据中学习,以实时生成用户特定且个性化的回复。然而,用户生成的数据通常包含敏感和隐私信息,若非被禁止,也不倾向于将此类数据上传至云端进行标注。虽然可以通过直接请求用户提供偏好回复的方式在本地获得标注,但此类标注必须保持稀疏以免影晌用户体验。此外,边缘设备的存储通常过于有限,无法支持基于全部用户生成数据的大规模微调。考虑到稀疏标注与有限的设备端存储,如何启用设备端 LLM 个性化仍是一个开放问题。本文提出一种新颖的框架,以自监督方式在线选择并存储最具代表性的数据。此类数据占用内存极小,并允许不频繁地请求用户标注以进行进一步微调。为提升微调质量,利用 LLM 生成多个语义相似的问句文本与期望回复对。我们的实验表明,相较于朴素基线,所提框架在用户特定内容生成能力(准确率)与微调速度(性能)上均达到最佳。据我们所知,这是首个设备端 LLM 个性化框架。

关键词

引用

@article{arxiv.2311.12275,
  title  = {Enabling On-Device Large Language Model Personalization with Self-Supervised Data Selection and Synthesis},
  author = {Ruiyang Qin and Jun Xia and Zhenge Jia and Meng Jiang and Ahmed Abbasi and Peipei Zhou and Jingtong Hu and Yiyu Shi},
  journal= {arXiv preprint arXiv:2311.12275},
  year   = {2024}
}

备注

Accepted by 2024 61th ACM/IEEE Design Automation Conference (DAC)