ELSA让边缘设备高效安全地协作训练大模型。
ELSA: Efficient LLM-Centric Split Aggregation for Privacy-Aware Hierarchical Federated Learning over the Network Edge
- 分层聚合+模型拆分,适应边缘设备算力。
- 跨设备通信量降低60%以上,隐私泄露风险显著下降。
- 适合资源受限、数据分散的边缘AI场景。
在资源受限的网络边缘训练大语言模型(LLM)面临设备算力不足、数据异构性强和隐私风险高等挑战。为此,我们提出ELSA(高效面向大模型的分层聚合框架),系统融合分片学习(SL)与分层联邦学习(HFL),实现分布式LLM微调。ELSA引入三项创新:首先,采用无任务依赖、行为感知的客户端聚类机制,利用公开探针输入构建语义指纹,并结合对称KL散度、预测一致性信任评分与延迟感知分配,协同缓解数据异构、设备不可靠与通信约束;其次,设计资源感知的动态模型拆分策略,将LLM自适应划分为三段,分别部署于客户端与边缘服务器,仅云端参与适配器聚合,有效平衡本地计算开销与全局收敛稳定性;第三,提出基于计算草图与语义子空间正交扰动(SS-OP)的轻量级通信方案,大幅降低通信开销并抑制模型交换中的隐私泄露。在多种NLP任务上的实验表明,ELSA在适应性、收敛性与鲁棒性方面均优于现有先进基线,为资源受限环境下的边缘侧大模型微调提供了可扩展、隐私友好的解决方案。
原文摘要 · Abstract (English)
Training large language models (LLMs) at the network edge faces fundamental challenges arising from device resource constraints, severe data heterogeneity, and heightened privacy risks. To address these challenges, we propose ELSA (Efficient LLM-centric Split Aggregation), a novel framework that systematically integrates split learning (SL) and hierarchical federated learning (HFL) for distributed LLM fine-tuning over resource-constrained edge networks. ELSA introduces three key innovations. First, it employs a task-agnostic, behavior-aware client clustering mechanism that constructs semantic fingerprints using public probe inputs and symmetric Kullback-Leibler (KL) divergence, augmented by prediction-consistency trust scoring and latency-aware edge assignment to jointly mitigate data heterogeneity, device unreliability, and communication constraints. Second, it employs a resource-aware dynamic model splitting strategy to adaptively partition the LLM into three segments across clients and edge servers, with the cloud used only for adapter aggregation, enabling an effective balance between on-device computation cost and global convergence stability. Third, it incorporates a lightweight communication scheme based on computational sketches combined with semantic subspace orthogonal perturbation (SS-OP) to reduce communication overhead while mitigating privacy leakage during model exchanges across the network. Extensive experiments across diverse NLP tasks demonstrate that ELSA consistently outperforms state-of-the-art baselines in terms of adaptability, convergence behavior, and robustness, establishing a scalable and privacy-aware solution for edge-side LLM fine-tuning under resource constraints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。