Organizations: School of Computer and Information Engineering, Xiamen University of Technology, Xiamen, 361000, China · School of Artificial Intelligence, Hebei University of Technology · Department of Electrical and Electronic Engineering, The University of Hong Kong, Hong Kong · Center of Research for Cyber Security and Network (CSNET), Faculty of Computer Science and Information Technology, Universiti Malaya, 50603 Kuala Lumpur, Malaysia · Department of Electrical Engineering, Chalmers University of Technology, 41296 Gothenburg, Sweden · Institute for Imaging, Data and Communications, School of Engineering, The University of Edinburgh, EH9 3JL Edinburgh, U.K. · Department of Computer Science, City University of Hong Kong, Kowloon, Hong Kong SAR, China · School of Engineering, Edith Cowan University, Perth, WA 6027, Australia
The growing complexity of neural networks hinders the deployment of distributed machine learning on resource-constrained devices. Split learning (SL) offers a promising solution by partitioning the large model and offloading the primary training workload from edge devices to an edge server. However, the increasing number of participating devices and model complexity leads to significant communication overhead from the transmission of smashed data (e.g., activations and gradients), which constitutes a critical bottleneck for SL. To tackle this challenge, we propose SL-FAC, a communication-efficient SL framework comprising two key components: adaptive frequency decomposition (AFD) and frequency-based quantization compression (FQC). AFD first transforms the smashed data into the frequency domain and decomposes it into spectral components with distinct information. FQC then applies customized quantization bit widths to each component based on its spectral energy distribution. This collaborative approach enables SL-FAC to achieve significant communication reduction while strategically preserving the information most crucial for model convergence. Extensive experiments confirm the superior performance of SL-FAC for improving the training efficiency.
Figures & tables
Fig. 1: The architecture of the proposed SL-FAC framework. (a) AFD for transforming the smashed data into the frequency domain and decoupling it into components corresponding to distinct information types; (b) FQC for customizing quantization bit widths for each component. Here, Lf- and Hf-features (gradients) denote low and high frequency features (gradients), while Le- and He-feature represent low and high energy features, respectively.
Fig. 2: The training performance of SL-FAC on the MNIST and HAM10000 datasets (from top to bottom), under IID (left) and non-IID (right) settings.
Fig. 3: The impact of energy threshold θ on model performance using the MNIST dataset under IID and non-IID settings.
Fig. 4: The ablation experiments for AFD and FQC (from top to bottom) on the MNIST dataset under IID (left) and non-IID (right) settings.
Split learning (SL) transfers most of the training workload to the server, which alleviates computational burden on client devices. However, the transmission of intermediate feature representations, referred to as smashed data, incurs significant communication overhead, particularly when a large number of client devices are involved. To address this challenge, we propose an adaptive channel pruning-aided SL (ACP-SL) scheme. In ACP-SL, a label-aware channel importance scoring (LCIS) module is designed to generate channel importance scores, distinguishing important channels from less important ones. Based on these scores, an adaptive channel pruning (ACP) module is developed to prune less important channels, thereby compressing the corresponding smashed data and reducing the communication overhead. Experimental results show that ACP-SL consistently outperforms benchmark schemes in test accuracy. Furthermore, it reaches a target test accuracy in fewer training rounds, thereby reducing communication overhead.
Jialei Tan, Zheng Lin, Xiangming Cai +4
School of Physics and Information Engineering, Fuzhou University, Fuzhou 350108, China · Department of Electrical and Electronic Engineering, The University of Hong Kong, Hong Kong · School of Electrical Engineering and Telecommunications, University of New South Wales Sydney, Sydney, NSW 2052, Australia +3
Federated fine-tuning of on-device large language models (LLMs) faces a significant computing burden. To overcome this limitation, split learning (SL) has emerged as a promising solution, which offloads the primary training workload to a powerful server. However, SL requires exchanging high-dimensional activations and gradients between clients and the server, resulting in prohibitive communication costs. To overcome this challenge, we propose SplitLite, a communication-efficient split federated LoRA fine-tuning method that exploits the low effective rank structure of consecutive-epoch activation and gradient residuals. Our key finding is that, when LoRA uses rank r updates in parameter space, the activation and gradient residuals of the same data sample between adjacent epochs also exhibit effective rank-2r and rank-4r structures, respectively. By revealing this property, SplitLite transmits only quantized truncated singular value decomposition (SVD) residual factors, thereby significantly reducing both activation uplink and gradient downlink traffic. Extensive experiments on the GLUE benchmark across a series of advanced on-device LLMs demonstrate that our method reduces activation uplink communication costs by up to 93.5% and total communication costs by up to 83.7%, without performance degradation.
Tao Li, Yulin Tang, Qi Guo +1
Department of Electrical and Computer Engineering, The University of Hong Kong, Hong Kong SAR, China · Department of Electrical Engineering, California Institute of Technology, Pasadena, CA, USA
Distributed Fine-Tuning (DFT) of large-scale Foundation Models (FMs) on resource-constrained edge devices is limited by local compute constraints and communication overhead. Parallel Split Learning (PSL) reduces client-side computation by keeping few model layers on each client and offloading the remaining computation to the server; however, clients must exchange intermediate activations and gradients with the server at every training step. Existing SL communication-compression methods mainly rely on task-agnostic heuristics, such as sparsification and quantization. While learnable SL compressors can better adapt to intermediate representations, they require co-training with the target model. Therefore, directly inserting them into off-the-shelf FMs introduces feature-distribution misalignment and degrades DFT performance. To address this, we propose AE-PSL, a communication-efficient PSL framework that compresses intermediate activations and gradients using a lightweight AutoEncoder (AE) placed at the split layer. To ensure compatibility of AE compression with pre-trained FMs, AE-PSL introduces a novel two-stage alignment mechanism, which adapts the AE to the pre-trained model's feature manifold and client-specific feature distributions before DFT.
Bas Meuwissen, Vasileios Tsouvalas, Nirvana Meratnia
Eindhoven University of Technology, Eindhoven, The Netherlands