用聚类中心替代模型参数,让大模型联邦学习更省带宽、更安全。
Trustformer: A Trusted Federated Transformer
- 每层用k-means聚类生成中心点,只传中心点而非完整参数
- 翻译任务上性能接近顶尖方法,通信量降低超70%
- 结合Intel SGX保障传输安全,适合隐私敏感的大型模型训练
Transformer作为序列数据深度学习的核心架构,在自然语言处理等任务中表现卓越,推动了大语言模型(LLM)的发展。然而,训练过程中的数据隐私保护仍是关键挑战。联邦学习(FL)虽能提供解决方案,但存在两大瓶颈:一是如FedAvg或FedSGD等聚合方式可能导致敏感信息泄露;二是Transformer模型体量庞大,通信开销高。本文提出一种新型联邦学习方法,通过在本地模拟全局模型,避免传输完整模型权重。具体地,对每一层Transformer进行k-means聚类,本地计算聚类中心并仅上传这些中心点,而非全量参数或梯度。为增强安全性,采用Intel SGX实现中心点的安全传输。在机器翻译任务上的实验表明,该方法在保持与现有最优基准相当性能的同时,显著降低了通信成本,为Transformer模型的高效且隐私保护的联邦训练提供了新路径。
原文摘要 · Abstract (English)
Transformers, a cornerstone of deep-learning architectures for sequential data, have achieved state-of-the-art results in tasks like Natural Language Processing (NLP). Models such as BERT and GPT-3 exemplify their success and have driven the rise of large language models (LLMs). However, a critical challenge persists: safeguarding the privacy of data used in LLM training. Privacy-preserving techniques like Federated Learning (FL) offer potential solutions, but practical limitations hinder their effectiveness for Transformer training. Two primary issues are (I) the risk of sensitive information leakage due to aggregation methods like FedAvg or FedSGD, and (II) the high communication overhead caused by the large size of Transformer models. This paper introduces a novel FL method that reduces communication overhead while maintaining competitive utility. Our approach avoids sharing full model weights by simulating a global model locally. We apply k-means clustering to each Transformer layer, compute centroids locally, and transmit only these centroids to the server instead of full weights or gradients. To enhance security, we leverage Intel SGX for secure transmission of centroids. Evaluated on a translation task, our method achieves utility comparable to state-of-the-art baselines while significantly reducing communication costs. This provides a more efficient and privacy-preserving FL solution for Transformer models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。