Selective Attention Federated Learning: Improving Privacy and Efficiency for Clinical Text Classification

  • 2025-04-18 21:58:03
  • Yue Li, Lihong Zhang
  • 0

Abstract

Federated Learning (FL) faces major challenges regarding communicationoverhead and model privacy when training large language models (LLMs),especially in healthcare applications. To address these, we introduce SelectiveAttention Federated Learning (SAFL), a novel approach that dynamicallyfine-tunes only those transformer layers identified as attention-critical. Byemploying attention patterns to determine layer importance, SAFL significantlyreduces communication bandwidth and enhances differential privacy resilience.Evaluations on clinical NLP benchmarks (i2b2 Clinical Concept Extraction andMIMIC-III discharge summaries) demonstrate that SAFL achieves competitiveperformance with centralized models while substantially improving communicationefficiency and privacy preservation.

 

Quick Read (beta)

loading the full paper ...