A Federated Fine-Tuning Paradigm of Foundation Models in Heterogenous Wireless Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Jingyi, Zhao, Zhongyuan, Wang, Qingtian, Li, Zexu, Wang, Yue, Quek, Tony Q. S.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912601717342208
author Wang, Jingyi
Zhao, Zhongyuan
Wang, Qingtian
Li, Zexu
Wang, Yue
Quek, Tony Q. S.
author_facet Wang, Jingyi
Zhao, Zhongyuan
Wang, Qingtian
Li, Zexu
Wang, Yue
Quek, Tony Q. S.
contents Edge intelligence has emerged as a promising strategy to deliver low-latency and ubiquitous services for mobile devices. Recent advances in fine-tuning mechanisms of foundation models have enabled edge intelligence by integrating low-rank adaptation (LoRA) with federated learning. However, in wireless networks, the device heterogeneity and resource constraints on edge devices pose great threats to the performance of federated fine-tuning. To tackle these issues, we propose to optimize federated fine-tuning in heterogenous wireless networks via online learning. First, the framework of switching-based federated fine-tuning in wireless networks is provided. The edge devices switches to LoRA modules dynamically for federated fine-tuning with base station to jointly mitigate the impact of device heterogeneity and transmission unreliability. Second, a tractable upper bound on the inference risk gap is derived based on theoretical analysis. To improve the generalization capability, we formulate a non-convex mixed-integer programming problem with long-term constraints, and decouple it into model switching, transmit power control, and bandwidth allocation subproblems. An online optimization algorithm is developed to solve the problems with polynomial computational complexity. Finally, the simulation results on the SST-2 and QNLI data sets demonstrate the performance gains in test accuracy and energy efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2509_19306
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Federated Fine-Tuning Paradigm of Foundation Models in Heterogenous Wireless Networks
Wang, Jingyi
Zhao, Zhongyuan
Wang, Qingtian
Li, Zexu
Wang, Yue
Quek, Tony Q. S.
Signal Processing
Artificial Intelligence
Information Theory
Networking and Internet Architecture
Edge intelligence has emerged as a promising strategy to deliver low-latency and ubiquitous services for mobile devices. Recent advances in fine-tuning mechanisms of foundation models have enabled edge intelligence by integrating low-rank adaptation (LoRA) with federated learning. However, in wireless networks, the device heterogeneity and resource constraints on edge devices pose great threats to the performance of federated fine-tuning. To tackle these issues, we propose to optimize federated fine-tuning in heterogenous wireless networks via online learning. First, the framework of switching-based federated fine-tuning in wireless networks is provided. The edge devices switches to LoRA modules dynamically for federated fine-tuning with base station to jointly mitigate the impact of device heterogeneity and transmission unreliability. Second, a tractable upper bound on the inference risk gap is derived based on theoretical analysis. To improve the generalization capability, we formulate a non-convex mixed-integer programming problem with long-term constraints, and decouple it into model switching, transmit power control, and bandwidth allocation subproblems. An online optimization algorithm is developed to solve the problems with polynomial computational complexity. Finally, the simulation results on the SST-2 and QNLI data sets demonstrate the performance gains in test accuracy and energy efficiency.
title A Federated Fine-Tuning Paradigm of Foundation Models in Heterogenous Wireless Networks
topic Signal Processing
Artificial Intelligence
Information Theory
Networking and Internet Architecture
url https://arxiv.org/abs/2509.19306