Proceedings

Faster and Lighter LLMs: A Survey on Current Challenges and Way Forward

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Survey Track. Pages 7980-7988.

https://doi.org/10.24963/ijcai.2024/883

Abstract

Despite the impressive performance of LLMs, their widespread adoption faces challenges due to substantial computational and memory requirements during inference. Recent advancements in model compression and system-level optimization methods aim to enhance LLM inference. This survey offers an overview of these methods, emphasizing recent developments. Through experiments on LLaMA(/2)-7B, we evaluate various compression techniques, providing practical insights for efficient LLM deployment in a unified setting. The empirical analysis on LLaMA(/2)-7B highlights the effectiveness of these methods. Drawing from survey insights, we identify current limitations and discuss potential future directions to improve LLM inference efficiency. We release the codebase to reproduce the results presented in this paper at https://github.com/nyunAI/Faster-LLM-Survey

  1. [Agarwal et al., 2023] R. Agarwal, N. Vieillard, Y. Zhou, P. Stanczyk, S. Ramos, M. Geist, and O. Bachem. Gener- alized knowledge distillation for auto-regressive language models. arXiv, 2023.
  2. [An et al., 2023] Y. An, X. Zhao, T. Yu, M. Tang, and J. Wang. Fluctuation-based adaptive structured pruning for large language models. arXiv, 2023.
  3. [Banner et al., 2019] R. Banner, Y. Nahshan, E. Hoffer, and D. Soudry. Post-training 4-bit quantization of convolution networks for rapid-deployment. arXiv, 2019.
  4. [Brown et al., 2020] T Brown, B Mann, N Ryder, M Sub- biah, J D Kaplan, P Dhariwal, A Neelakantan, P Shyam, G Sastry, A Askell, et al. Language models are few-shot learners. NeurIPS, 33:1877-1901, 2020.
  5. [Chavan et al., 2022] A Chavan, Z Shen, Z Liu, Z Liu, K T Cheng, and E P Xing. Vision transformer slim- ming: Multi-dimension searching in continuous optimiza- tion space. In CVPR, pages 4931-4941, 2022.
  6. [Chavan et al., 2023] A. Chavan, N. Lele, and D. Gupta. Re- thinking compression: Reduced order modelling of latent features in large language models. arXiv, 2023.
  7. [Chen et al., 2023a] T. Chen, T. Ding, B. Yadav, I. Zharkov, and L. Liang. Lorashear: Efficient large language model structured pruning and knowledge recovery. arXiv, 2023.
  8. [Chen et al., 2023b] Z. Chen, Q. Gao, A. Bosselut, A. Sab- harwal, and K. Richardson. Disco: Distilling counterfac- tuals with large language models. arXiv, 2023.
  9. [Choukroun et al., 2019] Y. Choukroun, E. Kravchik, F. Yang, and P. Kisilev. Low-bit quantization of neural networks for efficient inference. arXiv, 2019.
  10. [Dao et al., 2022] T. Dao, D. Fu, S. Ermon, A. Rudra, and Christopher Ré. Flashattention: Fast and memory-efficient exact attention with io-awareness. NeurIPS, 35:16344- 16359, 2022.
  11. [Dao, 2023] Tri Dao. Flashattention-2: Faster attention with better parallelism and work partitioning. arXiv, 2023.
  12. [Denton et al., 2014] E. Denton, W. Zaremba, J. Bruna, Y. LeCun, and R. Fergus. Exploiting linear structure within convolutional networks for efficient evaluation. arXiv, 2014.
  13. [Dettmers et al., 2022] T. Dettmers, M. Lewis, Y. Belkada, and L. Zettlemoyer. Llm.int8(): 8-bit matrix multiplication for transformers at scale. arXiv, 2022.
  14. [Dettmers et al., 2023] T. Dettmers, A. Pagnoni, A. Holtz- man, and L. Zettlemoyer. Qlora: Efficient finetuning of quantized llms. arXiv, 2023.
  15. [Esser et al., 2019] S. K Esser, J. L McKinstry, D. Bablani, R. Appuswamy, and D. S Modha. Learned step size quan- tization. arXiv, 2019.
  16. [Frankle and Carbin, 2019] J. Frankle and M. Carbin. The lottery ticket hypothesis: Finding sparse, trainable neural networks. arXiv, 2019. [Frantar and Alistarh, 2023]
  17. E. Frantar and D. Alistarh. Sparsegpt: Massive language models can be accurately pruned in one-shot. arXiv, 2023.
  18. [Frantar et al., 2023] E. Frantar, S. Ashkboos, T. Hoefler, and D. Alistarh. Gptq: Accurate post-training quantiza- tion for generative pre-trained transformers. arXiv, 2023.
  19. [Hassibi et al., 1993] B. Hassibi, D. G Stork, and G. J Wolff. Optimal brain surgeon and general network pruning. In ICNN, pages 293-299. IEEE, 1993.
  20. [Hinton et al., 2015] G. Hinton, O. Vinyals, and J. Dean. Distilling the knowledge in a neural network. arXiv, 2015.
  21. [Hu et al., 2021] E. J Hu, P. Wallis, Z. Allen-Zhu, Y. Li, S. Wang, L. Wang, W. Chen, et al. Lora: Low-rank adap- tation of large language models. In ICLR, 2021.
  22. [Jaderberg et al., 2014] M. Jaderberg, A. Vedaldi, and A. Zisserman. Speeding up convolutional neural networks with low rank expansions. arXiv, 2014.
  23. [Janowsky, 1989] S. A. Janowsky. Pruning versus clipping in neural networks. Phys. Rev. A, 39:6600-6603, 1989.
  24. [Jha et al., 2023] A. H. Jha, T. S., E. P. Walsh, D. Groen- eveld, E. Strubell, and I. Beltagy. How to train your (com- pressed) large language model. arXiv, 2023.
  25. [Jiang et al., 2023a] T. Jiang, D. Wang, F. Zhuang, R. Xie, and F. Xia. Pruning pre-trained language models without fine-tuning. arXiv, 2023.
  26. [Jiang et al., 2023b] Y. Jiang, C. Chan, M. Chen, and W. Wang. Lion: Adversarial distillation of proprietary large language models. arXiv, 2023.
  27. [Kaushal et al., 2023] A. Kaushal, T. Vaidhya, and I. Rish. Lord: Low rank decomposition of monolingual code llms for one-shot compression. arXiv, 2023.
  28. [Kenton and Toutanova, 2019] Jacob Devlin Ming- Wei Chang Kenton and L. Kristina Toutanova. Bert: Pre-training of deep bidirectional transformers for lan- guage understanding. In Proceedings of NAACL-HLT, pages 4171-4186, 2019.
  29. [Kim et al., 2023] S. Kim, C. Hooper, A. Gholami, Z. Dong, X. Li, S. Shen, M. W. Mahoney, and K. Keutzer. Squeezellm: Dense-and-sparse quantization. arXiv, 2023.
  30. [Kwon et al., 2023] W. Kwon, Z. Li, S. Zhuang, Y. Sheng, L. Zheng, C. H. Yu, J. Gonzalez, H. Zhang, and I. Stoica. Efficient memory management for large language model serving with pagedattention. In Proceedings of the 29th Symposium on Operating Systems Principles, pages 611- 626, 2023.
  31. [Lagunas et al., 2021] F. Lagunas, E. Charlaix, V. Sanh, and A. M. Rush. Block pruning for faster transformers. arXiv, 2021.
  32. [Lebedev et al., 2015] V. Lebedev, Y. Ganin, M. Rakhuba, I. Oseledets, and V. Lempitsky. Speeding-up convolu- tional neural networks using fine-tuned cp-decomposition. arXiv, 2015.
  33. [LeCun et al., 1989] Y. LeCun, J. Denker, and S. Solla. Op- timal brain damage. NeurIPS, 2, 1989.
  34. [Leviathan et al., 2023] Y. Leviathan, M. Kalman, and Y. Matias. Fast inference from transformers via specu- lative decoding. In ICML, pages 19274-19286. PMLR, 2023.
  35. [Li et al., 2016] H. Li, A. Kadav, I. Durdanovic, H. Samet, and H. P. Graf. Pruning filters for efficient convnets. arXiv, 2016.
  36. [Li et al., 2023] Y. Li, Y. Yu, Q. Zhang, C. Liang, P. He, W. Chen, and T. Zhao. Losparse: Structured compression of large language models based on low-rank and sparse approximation. arXiv, 2023.
  37. [Liang et al., 2023] C. Liang, S. Zuo, Q. Zhang, P. He, W. Chen, and T. Zhao. Less is more: Task-aware layer- wise distillation for language model compression. arXiv, 2023.
  38. [Lin et al., 2023] J. Lin, J. Tang, H. Tang, S. Yang, X. Dang, C. Gan, and S. Han. Awq: Activation-aware weight quan- tization for llm compression and acceleration. arXiv, 2023.
  39. [Liu et al., 2017] Z. Liu, J. Li, Z. Shen, G. Huang, S. Yan, and C. Zhang. Learning efficient convolutional networks through network slimming. arXiv, 2017.
  40. [Liu et al., 2023] Z. Liu, B. Oguz, C. Zhao, E. Chang, P. Stock, Y. Mehdad, Y. Shi, R. Krishnamoorthi, and V. Chandra. Llm-qat: Data-free quantization aware train- ing for large language models. arXiv, 2023.
  41. [Ma et al., 2023] X. Ma, G. Fang, and X. Wang. Llm-pruner: On the structural pruning of large language models. arXiv, 2023.
  42. [Molchanov et al., 2016] P. Molchanov, S. Tyree, T. Karras, T. Aila, and J. Kautz. Pruning convolutional neural net- works for resource efficient inference. In ICLR, 2016.
  43. [Ni et al., 2020] R. Ni, H. M. Chu, O. Castañeda, P. Y. Chi- ang, C. Studer, and T. Goldstein. Wrapnet: Neural net in- ference with ultra-low-resolution arithmetic. arXiv, 2020.
  44. [S. et al., 2023] Hang S., Bei L., and Yanmin Q. One-Shot Sensitivity-Aware Mixed Sparsity Pruning for Large Lan- guage Models. arXiv, 2023.
  45. [Sainath et al., 2013] T. N. Sainath, B. Kingsbury, V. Sind- hwani, E. Arisoy, and B. Ramabhadran. Low-rank matrix factorization for deep neural network training with high- dimensional output targets. In ICASSP, pages 6655-6659, 2013.
  46. [Sanh et al., 2020] V. Sanh, T. Wolf, and A. M. Rush. Move- ment pruning: Adaptive sparsity by fine-tuning. arXiv, 2020.
  47. [Shao et al., 2023] W. Shao, M. Chen, Z. Zhang, P. Xu, L. Zhao, Z. Li, K. Zhang, P. Gao, Y. Qiao, and P. Luo. Omniquant: Omnidirectionally calibrated quantization for large language models. arXiv, 2023.
  48. [Sharma et al., 2023] P. Sharma, J. T. Ash, and D. Misra. The truth is in there: Improving reasoning in language models with layer-selective rank reduction. arXiv, 2023.
  49. [Song et al., 2023] Y. Song, Z. Mi, H. Xie, and H. Chen. Powerinfer: Fast large language model serving with a consumer-grade gpu. arXiv, 2023.
  50. [Sun et al., 2023] M. Sun, Z. Liu, A. Bair, and J. Z. Kolter. A simple and effective pruning approach for large language models. arXiv, 2023.
  51. [Tai et al., 2016] C. Tai, T. Xiao, Y. Zhang, X. Wang, and Weinan E. Convolutional neural networks with low-rank regularization. arXiv, 2016.
  52. [Tailor et al., 2021] S. A. Tailor, J. F. Marques, and N. D. Lane. Degree-quant: Quantization-aware training for graph neural networks. arXiv, 2021.
  53. [Tao et al., 2022] C. Tao, L. Hou, W. Zhang, L. Shang, X. Jiang, Q. Liu, P. Luo, and N. Wong. Compression of generative pre-trained language models via quantization. arXiv, 2022.
  54. [Tian et al., 2022] Yo. Tian, D. Krishnan, and P. Isola. Con- trastive representation distillation. arXiv, 2022.
  55. [Touvron et al., 2023a] H. Touvron, T. Lavril, G. Izacard, X. Martinet, M. A. Lachaux, T. Lacroix, B. Rozière, N. Goyal, E. Hambro, F. Azhar, et al. Llama: Open and efficient foundation language models. arXiv, 2023.
  56. [Touvron et al., 2023b] H. Touvron, L. Martin, K. Stone, P. Albert, A. Almahairi, Y. Babaei, N. Bashlykov, S. Batra, P. Bhargava, S. Bhosale, et al. Llama 2: Open foundation and fine-tuned chat models. arXiv, 2023.
  57. [Wang et al., 2023a] H. Wang, S. Ma, L. Dong, S. Huang, H. Wang, L. Ma, F. Yang, R. Wang, Y. Wu, and F. Wei. Bitnet: Scaling 1-bit transformers for large language mod- els. arXiv, 2023.
  58. [Wang et al., 2023b] P. Wang, Z. Wang, Z. Li, Y. Gao, B. Yin, and X. Ren. Scott: Self-consistent chain-of- thought distillation. arXiv, 2023.
  59. [Wu et al., 2023] X. Wu, Z. Yao, and Y. He. Zeroquant-fp: A leap forward in llms post-training w4a8 quantization using floating-point formats. arXiv, 2023.
  60. [Xia et al., 2023] M. Xia, T. Gao, Z. Zeng, and D. Chen. Sheared llama: Accelerating language model pre-training via structured pruning. arXiv, 2023.
  61. [Xiao et al., 2023] G. Xiao, J. Lin, M. Seznec, H. Wu, J. De- mouth, and S. Han. Smoothquant: Accurate and effi- cient post-training quantization for large language models. arXiv, 2023.
  62. [Xu et al., 2023] M. Xu, Y. Lei Xu, and D. P. Mandic. Ten- sorgpt: Efficient compression of the embedding layer in llms based on the tensor-train decomposition. arXiv, 2023.
  63. [Zhang et al., 2023] M. Zhang, H. Chen, C. Shen, Z. Yang, L. Ou, X. Yu, and B. Zhuang. Loraprune: Pruning meets low-rank parameter-efficient fine-tuning. arXiv, 2023.
  64. [Zhu et al., 2023] X. Zhu, J. Li, Y. Liu, C. Ma, and W. Wang. A survey on model compression for large language mod- els. arXiv, 2023.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards