Organizations: University of Houston, USA · Pengcheng Laboratory, China · Beijing University of Posts and Telecommunications, China · Waseda University, Japan · Keio University, Japan
There is a huge gap between numerous intriguing applications fostered by on-device large language model (LLM) fine-tuning (FT) from fresh mobile data and the limited resources of a mobile device. While existing server-assisted methods (e.g., split learning or side-tuning) may enable LLM FT on the local mobile device, they suffer from heavy communication burdens of activation transmissions. In addition, their multiple on-device forward passes over the same samples across training epochs lead to substantial iterative computation on the mobile device. To address those issues, we develop Eco-MobiLLM, an efficient LLM FT method which can be deployed on the mobile device via server-assisted side-tuning. To reduce communication cost, Eco-MobiLLM develops an activation shortcut that transmits only the tokens involved in the loss calculation instead of full activation matrices to guide the side network tuning. Besides, to improve computing efficiency and further accelerate FT convergence, Eco-MobiLLM introduces a novel two-stage training paradigm, which allows the server to reuse historical activations and saves the mobile device from repeatedly computing forward passes for the recurring data samples. Extensive experimental results demonstrate Eco-MobiLLM's superiority.
Figures & tables
Figure 1: Device-server collaborative LLM FT.
Method
Per-Batch Comm. (MB)
Per-Epoch Comp. (PFLOPS)
Total Comp. (PFLOPS)
BitFit-local
-
18.4
208.3
BitFit-SL
33.4
1.5
17.4
MobiLLM
200.4
9.5
124.5
Table 1: Communication and computational costs (On Xavier, task: [email protected], batch size=8, seq len=256).
Figure 2: An overview of the Eco-MobiLLM system.
Figure 3: Sketches of the adapter side-tuning.
Figure 4: Token-level selection for different LLM FT tasks.
Figure 5: Convergence performance comparisons over various models, tasks, and devices.
Model
OPT-1.3B
RoBERTa-Large
Data (Metrics)
GLUE (Avg.)
Dialogsum (R1/R2/RL)
GLUE (Avg.)
Eco-MobiLLM
81.1
21.1 / 7.9 / 17.1
86.8
w/o. TS
81.2
21.1 / 7.7 / 17.0
86.7
w/o. TSTP
81.1
21.0 / 7.7 / 17.1
86.8
w/o. TS/TSTP
81.4
21.3 / 7.6 / 17.5
86.6
Table 4: Ablation study on LLM fine-tuning performance (TS: T oken S elector, TSTP: T wo S tage T raining P aradigm, w/o. : Without).
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
Dataset
Description
# Samples (train/eval)
CoLA
Linguistic Acceptability
8551 / 1043
SST2
Sentiment Analysis
67349 / 872
MRPC
Sentence Equivalence
3668 / 408
STSB
Sentence Similarity
5749 / 1500
QQP
Paraphrase Recognition
363846 / 40430
RTE
Textual Entailment
2490 / 277
Appendix
Table 5: Dataset statistics.
Method
Server cache size (GB)
RTE
SST2
QNLI
Eco-MobiLLM
0.23
6.21
36.77
w/o. TS
58.88
1589.76
3637.07
Appendix
Table 6: Server cache size across tasks (OPT-1.3B).
Figure 6: Impact of network speed on execution time per batch (On Xavier, OPT-1.3B).
Model
OPT-1.3B
RoBERTa-Large
Target Acc.
82.5%
87.1%
Method
On-device Comp. (PFLOPs)
Per Epoch Comp.
Total Comp.
Per Epoch Comp.
Total Comp.
Eco-MobiLLM
9.5
9.5
2.4
2.4
w/o. TS
9.5
9.5
2.4
2.4
w/o. TSTP
9.5
124.5
2.4
32.5
Appendix
Table 7: Ablation studies on the proposed Eco-MobiLLM designs regarding on-device computational cost. (Xavier, task: MRPC; TS: T oken S elector, TSTP: T wo S tage T raining P aradigm, w/o. : Without).
Table 8: Ablation studies on the proposed Eco-MobiLLM designs regarding communication cost.
Method
Hidden Size = 16 (R1 / R2 / RL)
Hidden Size = 64 (R1 / R2 / RL)
Hidden Size = 128 (R1 / R2 / RL)
Hidden Size = 256 (R1 / R2 / RL)
Eco-MobiLLM
19.8 / 7.0 / 15.5
21.1 / 7.9 / 17.1
21.9 / 8.4 / 17.9
21.6 / 8.3 / 18.1
w/o. TS
19.8 / 7.1 / 15.5
21.1 / 7.7 / 17.0
22.1 / 8.3 / 18.1
21.5 / 8.1 / 18.0
w/o. TSTP
19.6 / 7.4 / 15.6
21.0 / 7.7 / 17.1
21.9 / 8.3 / 17.9
21.7 / 8.4 / 18.0
w/o. TS/TSTP
19.6 / 7.2 / 15.8
21.3 / 7.6 / 17.5
21.8 / 8.5 / 18.0
21.7 / 8.4 / 18.1
Appendix
Table 9: Ablation study on fine-tuning performance with different internal low-rank hidden sizes of the trainable side-network modules. (task: [email protected]).
Collaboratively fine-tuning (FT) large language models (LLMs) over heterogeneous mobile devices fosters immense potential applications of personalized intelligence. However, such a vision faces critical system challenges. Existing federated LLM fine-tuning approaches remain limited by prohibitive on-device resource overhead, straggler-prone synchronous aggregation, and the assumption of a unified pretrained backbone across heterogeneous devices. In this paper, we propose EC-MobiLLM, a novel design for efficient collaborative LLM fine-tuning across heterogeneous mobile devices and heterogeneous backbones. EC-MobiLLM adopts a server-assisted side-tuning paradigm to minimize on-device overhead and pioneers non-blocking asynchronous collaboration to accelerate training. Moreover, EC-MobiLLM introduces adaptive feature alignment to support collaboration across heterogeneous backbones, aligning with practical needs in real-world deployments. Extensive experimental results demonstrate that EC-MobiLLM can maintain robust fine-tuning performance while achieving extremely low on-device memory, with at least 95.2% reduction in computation overhead, 93.2% reduction in communication costs and 5.1× faster convergence compared to existing methods, validating its efficacy for practical LLM adaptation over heterogeneous mobile devices.
Xingke Yang, Liang Li, Sicong Li +6
University of Houston, USA · Pengcheng Laboratory, China · Beijing University of Posts and Telecommunications, China +2
Mixture-of-Experts (MoE) has become the de facto architecture for hundred-billion-parameter language models, yet its advantages at sub-billion scales for on-device deployment remain largely unexplored. To close this gap, we present MobileMoE, a family of on-device MoE language models with sub-billion active parameters (0.3-0.9B active and 1.3-5.3B total) that establish a new Pareto frontier for on-device LLMs. We first formulate an on-device MoE scaling law that jointly optimizes MoE architecture under mobile memory and compute constraints, identifying an on-device sweet spot - moderate sparsity with fine-grained and shared experts - that is simultaneously memory and compute-optimal. Building on the derived architectures, we train MobileMoE with a four-stage recipe covering pre-training, mid-training, instruction fine-tuning, and quantization-aware training, all on open-source datasets. Across 14 benchmarks, MobileMoE matches or exceeds leading on-device dense LLMs with 2-4× fewer inference FLOPs, and matches or surpasses the state-of-the-art MoE OLMoE-1B-7B with up to 60% fewer parameters. To bridge the last mile to mobile deployment, we provide the first efficient MoE inference on commodity smartphones with comprehensive on-device profiling. At comparable INT4 weight memory, MobileMoE-S delivers 1.8-3.8× faster prefill and 2.2-3.4× faster decode than the dense baseline MobileLLM-Pro.
Real-time AI experiences call for on-device large language models (OD-LLMs) optimized for efficient deployment on resource-constrained hardware. The most useful OD-LLMs produce near-real-time responses and exhibit broad hardware compatibility, maximizing user reach. We present a methodology for designing such models using hardware-in-the-loop architecture search under mobile latency constraints. This system is amenable to industry-scale deployment: it generates models deployable without custom kernels and compatible with standard mobile runtimes like Executorch. Our methodology avoids specialized attention mechanisms and instead uses attention skipping for long-context acceleration. Our approach jointly optimizes model architecture (layers, dimensions) and attention pattern. To efficiently evaluate candidates, we treat each as a pruned version of a pretrained backbone with inherited weights, thereby achieving high accuracy with minimal continued pretraining. We leverage the low cost of latency evaluation in a staged process: learning an accurate latency model first, then searching for the Pareto-frontier across latency and quality. This yields MobileLLM-Flash, a family of foundation models (350M, 650M, 1.4B) for efficient on-device use with strong capabilities, supporting up to 8k context length. MobileLLM-Flash delivers up to 1.8x and 1.6x faster prefill and decode on mobile CPUs with comparable or superior quality. Our analysis of Pareto-frontier design choices offers actionable principles for OD-LLM design.