Towards Coarse-to-Fine Evaluation of Inference Efficiency for Large Language Models

Yushuo Chen,Tianyi Tang,Erge Xiang,Linjiang Li,Wayne Xin Zhao,Jing Wang,Yunpeng Chai,Ji-Rong Wen
2024-04-17
Abstract:In real world, large language models (LLMs) can serve as the assistant to help users accomplish their jobs, and also support the development of advanced applications. For the wide application of LLMs, the inference efficiency is an essential concern, which has been widely studied in existing work, and numerous optimization algorithms and code libraries have been proposed to improve it. Nonetheless, users still find it challenging to compare the effectiveness of all the above methods and understand the underlying mechanisms. In this work, we perform a detailed coarse-to-fine analysis of the inference performance of various code libraries. To evaluate the overall effectiveness, we examine four usage scenarios within two practical applications. We further provide both theoretical and empirical fine-grained analyses of each module in the Transformer architecture. Our experiments yield comprehensive results that are invaluable for researchers to evaluate code libraries and improve inference strategies.
Computation and Language,Artificial Intelligence
What problem does this paper attempt to address?
This paper mainly discusses the evaluation problem of the inference efficiency of large-scale language models (LLMs). Although there are currently various optimization algorithms and code libraries to improve the inference speed, there is a lack of standard benchmarks for comprehensive comparison of these methods. The paper designs experiments for different application scenarios to evaluate the inference performance of multiple code libraries through coarse-grained and fine-grained analysis. It also conducts theoretical and empirical analysis on each module in the Transformer architecture to identify efficiency bottlenecks. The study also proposes a comprehensive benchmark to help evaluate code libraries and improve inference strategies.