cs.LGFeb 27, 2026

FedTreeLoRA: Reconciling Statistical and Functional Heterogeneity in Federated LoRA Fine-Tuning

Authors: Jieming Bian, Lei Wang, Letian Zhang, Jie Xu

Organizations: University of Florida, Gainesville, FL 32611 · Middle Tennessee State University Murfreesboro, TN 37132

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.

Figures & tables

Appendix figures & tables20 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. FedLAFP: Low-Rank Aggregation Meets Full-Rank Personalization in Federated Fine-Tuning

    Sep 29, 2026Mengjun Yi, Huaian Gu, Yinghao Ai +2Low-Rank AdaptationFederated Fine-Tuning

  2. Breaking the Structural Identity: Personalized Federated LoRA Fine-tuning under Rank Heterogeneity

    Sep 1, 2026Lei Wang, Jieming Bian, Letian Zhang +1Non-IID Federated LearningLow-Rank Adaptation

  3. FedPA-LoRA: Product-Aligned Framework for Mitigating Aggregation and Initialization Errors in Heterogeneous Federated LoRA

    Aug 15, 2026Juseok Jeon, Ramy E. Ali, Doyun Kwon +3Federated Fine-TuningFederated Learning