arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.09796cs.LGcs.CR
收藏

一种快速、高性能、安全的大语言模型分布式训练框架

A Fast, Performant, Secure Distributed Training Framework For Large Language Model

Ant Group(蚂蚁集团)

Wei Huang, Yinggui Wang, Anda Cheng, Aihui Zhou, Chaofan Yu, Lei Wang

AI总结:

针对分布式LLM训练中的参数与数据窃取风险,提出基于模型切片、TEE、轻量加密、拆分微调及SPF与LoRA结合的安全框架,在保障安全的同时维持下游任务准确率。

摘要:

分布式(联邦)LLM是利用数据孤岛协同训练特定领域LLM的重要方法。然而,从服务器或客户端恶意窃取模型参数和数据已成为一个亟待解决的问题。本文提出一种基于模型切片的安全分布式LLM。在该方案中,我们在客户端和服务器端均部署可信执行环境(TEE),并将微调结构(LoRA或P-tuning v2的embedding)放入TEE中。随后,通过轻量级加密在TEE与通用环境中执行安全通信。为进一步降低设备成本并提升模型性能和准确率,我们提出一种拆分微调方案。具体而言,我们按层拆分LLM,并将后面的层放置在服务器端TEE中(客户端无需TEE)。然后,我们将所提出的稀疏化参数微调(SPF)与LoRA部分相结合,以提高下游任务的准确率。大量实验表明,我们的方法在保持安全性的同时保证了准确率。

英文摘要:

The distributed (federated) LLM is an important method for co-training the domain-specific LLM using siloed data. However, maliciously stealing model parameters and data from the server or client side has become an urgent problem to be solved. In this paper, we propose a secure distributed LLM based on model slicing. In this case, we deploy the Trusted Execution Environment (TEE) on both the client and server side, and put the fine-tuned structure (LoRA or embedding of P-tuning v2) into the TEE. Then, secure communication is executed in the TEE and general environments through lightweight encryption. In order to further reduce the equipment cost as well as increase the model performance and accuracy, we propose a split fine-tuning scheme. In particular, we split the LLM by layers and place the latter layers in a server-side TEE (the client does not need a TEE). We then combine the proposed Sparsification Parameter Fine-tuning (SPF) with the LoRA part to improve the accuracy of the downstream task. Numerous experiments have shown that our method guarantees accuracy while maintaining security.

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

↑