CCG Crucible Computing Group Home

EDGE-LLM: Enabling Efficient Large Language Model Adaptation on Edge Devices via Unified Compression and Adaptive Layer Voting

Zhongzhi Yu, Zheng Wang, Yuhan Li, Ruijie Gao, Xiaoya Zhou, Sreenidhi Reddy Bommu, Yang (Katie) Zhao, Yingyan (Celine) Lin
DAC 2024 (2024)

DOI: 10.1145/3649329.3658473

Abstract

Efficient adaption of large language models (LLMs) on edge devices is essential for applications requiring continuous and privacy-preserving adaptation and inference. However, existing tuning techniques fall short because of the high computation and memory overhead. To this end, we introduce a computation- and memory-efficient LLM tuning framework, called Edge-LLM, to facilitate affordable and effective LLM adaptation on edge devices. Specifically, Edge-LLM features three core components: (1) a layer-wise unified compression (LUC) technique to reduce the computation overhead by generating layer-wise pruning sparsity and quantization bit-width policies, (2) an adaptive layer tuning and voting scheme to reduce the memory overhead by reducing the backpropagation depth, and (3) a complementary hardware scheduling strategy to handle the irregular computation patterns introduced by LUC and adaptive layer tuning, thereby achieving improved real hardware efficiency. Extensive experiments demonstrate that Edge-LLM achieves on-device adaptation with comparable task accuracy as vanilla tuning methods with a 2.92× speed up and a 4× reduction in memory overhead. Our code is available at https://github.com/GATECH-EIC/Edge-LLM

Cite this paper

@inproceedings{yu2024edge,
  title = {EDGE-LLM: Enabling Efficient Large Language Model Adaptation on Edge Devices via Unified Compression and Adaptive Layer Voting},
  author = {Zhongzhi Yu and Zheng Wang and Yuhan Li and Ruijie Gao and Xiaoya Zhou and Sreenidhi Reddy Bommu and Yang (Katie) Zhao and Yingyan (Celine) Lin},
  year = {2024},
  booktitle = {DAC 2024},
  doi = {10.1145/3649329.3658473},
  url = {https://crucible.eecs.umich.edu/papers/edge-llm-enabling-efficient-large-language-model-adaptation-on-edge-devices-via-17}
}

← All publications