arXiv:2507.22633cs.LGcs.AI2025-07被引 1

解决跨模型、跨任务的联邦微调难题,提升多类客户端协同效果。

H2Tune: Federated Foundation Model Fine-Tuning with Hybrid Heterogeneity

  • 通过稀疏三重分解对齐不同模型的隐藏维度,适应资源差异。
  • 在多个数据集上比现有方法最高提升15.4%准确率。
  • 适合异构设备上联合训练大模型,如边缘智能场景。

与以往针对基础模型的联邦微调方法不同,混合异构联邦微调(HHFFT)是一种客户端在模型架构和下游任务上均存在双重异质性的未充分探索场景。该场景带来两大挑战:一是异构矩阵聚合问题,即客户端根据任务需求和资源限制采用不同规模的基础模型,导致LoRA参数聚合时出现维度不匹配;二是多任务知识干扰,本地共享参数同时包含任务共性和特异性知识,难以确保仅共享知识在客户端间传递。为此,我们提出H2Tune框架,包含三个核心组件:(i) 稀疏三重矩阵分解,通过构建秩一致中间矩阵对齐客户端隐藏维度,并基于客户端资源自适应稀疏化;(ii) 关系引导的矩阵层对齐机制,处理异构层结构与表征能力差异;(iii) 交替任务-知识解耦机制,通过交替优化分离本地模型参数中的共享与特定知识。理论分析证明其收敛速度为O(1/√T)。大量实验表明,本方法相比最先进基线最高提升15.4%准确率。代码已开源于https://anonymous.4open.science/r/H2Tune-1407。

原文摘要 · Abstract (English)

Different from existing federated fine-tuning (FFT) methods for foundation models, hybrid heterogeneous federated fine-tuning (HHFFT) is an under-explored scenario where clients exhibit double heterogeneity in model architectures and downstream tasks. This hybrid heterogeneity introduces two significant challenges: 1) heterogeneous matrix aggregation, where clients adopt different large-scale foundation models based on their task requirements and resource limitations, leading to dimensional mismatches during LoRA parameter aggregation; and 2) multi-task knowledge interference, where local shared parameters, trained with both task-shared and task-specific knowledge, cannot ensure only task-shared knowledge is transferred between clients. To address these challenges, we propose H2Tune, a federated foundation model fine-tuning with hybrid heterogeneity. Our framework H2Tune consists of three key components: (i) sparsified triple matrix decomposition to align hidden dimensions across clients through constructing rank-consistent middle matrices, with adaptive sparsification based on client resources; (ii) relation-guided matrix layer alignment to handle heterogeneous layer structures and representation capabilities; and (iii) alternating task-knowledge disentanglement mechanism to decouple shared and specific knowledge of local model parameters through alternating optimization. Theoretical analysis proves a convergence rate of O(1/\sqrt{T}). Extensive experiments show our method achieves up to 15.4% accuracy improvement compared to state-of-the-art baselines. Our code is available at https://anonymous.4open.science/r/H2Tune-1407.

联邦学习大模型微调异构协同知识解耦

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。