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]).