FedTreeLoRA: Reconciling Statistical and Functional Heterogeneity in Federated LoRA Fine-Tuning
Jieming Bian, Lei Wang, Letian Zhang, Jie Xu
Abstract
Federated Learning (FL) with Low-Rank Adaptation (LoRA) has become a standard for privacy-preserving LLM fine-tuning. However, existing personalized methods predominantly operated under a restrictive Flat-Model Assumption: they addressed client-side statistical heterogeneity but treated the model as a monolithic block, ignoring the functional heterogeneity across LLM layers. We argue that these two statistical (horizontal) and functional (vertical) dimensions, are orthogonal in source yet coupled in interaction , implying that the optimal depth of parameter sharing is functionally dependent on client similarity. To address this, we propose FedTreeLoRA , a framework employing tree-structured aggregation for fine-grained, layer-wise alignment. By dynamically constructing an aggregation hierarchy, FedTreeLoRA allows clients to share broad consensus on shallow ’trunks‘ while progressively specializing on deep ‘branches'. Experiments on NLU and NLG benchmarks demonstrate that FedTreeLoRA significantly outperforms state-of-the-art methods by effectively reconciling generalization and personalization.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 2d0b03db-c598-42f3-8012-45626a153a9eBuilds on9
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Improving LoRA in Privacy-preserving Federated LearningYoubang Sun, Zitao Li, Yaliang Li, Bolin DingICLR 2024 · 173 citations
- Dual-Personalizing Adapter for Federated Foundation ModelsYiyuan Yang, Guodong Long, Tao Shen, Jing Jiang et al.NeurIPS 2024 · 84 citations
- LoRA-FAIR: Federated LoRA Fine-Tuning with Aggregation and Initialization RefinementJieming Bian, Lei Wang, Letian Zhang, Jie XuICCV 2025 · 56 citations
- Taming Cross-Domain Representation Variance in Federated Prototype Learning with Heterogeneous Data DomainsLei Wang, Jieming Bian, Letian Zhang, Chen Chen et al.NeurIPS 2024 · 36 citations
Related papers
- pFedGPT: Hierarchically Optimizing LoRA Aggregation Weights for Personalized Federated GPT ModelsZhanming Shen, Tianqi Xu, Hao Wang, Jian Li et al.EMNLP 2025 · 1 citation
- Heterogeneous Federated Fine-Tuning with Parallel One-Rank AdaptationZikai Zhang, Rui Hu, Jiahao XuICLR 2026 · 6 citations
- Tensor-Aggregated LoRA in Federated Fine-TuningZhixuan Li, Binqian Xu, Xiangbo Shu, Jiachao Zhang et al.ICCV 2025 · 2 citations
- FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank AdaptationsZiyao Wang, Zheyu Shen, Yexiao He, Guoheng Sun et al.NeurIPS 2024 · 227 citations
- HeteroFL-LoRA: Federated LoRA Fine-Tuning Across Heterogeneous LFMs via Singular Value CollaborationZhuojia Wu, Qi Zhang, Xuerong Zhao, Duoqian Miao et al.KDD 2026
