中文

选择性注意力联邦学习:用于临床文本分类的隐私与效率提升

计算与语言 2025-04-22 v3 人工智能

摘要

联邦学习(FL)在训练大型语言模型(LLMs)时面临通信开销和模型隐私方面的主要挑战,尤其是在医疗保健应用中。为此,我们提出了选择性注意力联邦学习(SAFL),一种新方法,通过动态微调仅识别为注意力关键层的变换器层。通过利用注意力模式来确定层重要性,SAFL显著减少了通信带宽并增强了差分隐私抗性。在临床 NLP 基准(i2b2 临床概念提取和 MIMIC-III 出院概述)上的评估表明,SAFL 在保持与集中式模型竞争性能的同时,大幅提高了通信效率和隐私保护。

关键词

引用

@article{arxiv.2504.11793,
  title  = {Selective Attention Federated Learning: Improving Privacy and Efficiency for Clinical Text Classification},
  author = {Yue Li and Lihong Zhang},
  journal= {arXiv preprint arXiv:2504.11793},
  year   = {2025}
}