MixPE: Quantization and Hardware Co-design for Efficient LLM Inference

TL;DR

MixPE introduces a specialized hardware design that enhances low-bit quantization efficiency for large language model inference by reducing dequantization overhead and utilizing shift extbackslash&add operations, achieving significant speed and energy improvements.

Contribution

The paper presents MixPE, a novel mixed-precision processing element that optimizes LLM inference by reducing dequantization overhead and replacing multipliers with shift extbackslash&add operations.

Findings

2.6x speedup over state-of-the-art accelerators

1.4x energy reduction

Effective low-bit quantization for LLM inference

Abstract

Transformer-based large language models (LLMs) have achieved remarkable success as model sizes continue to grow, yet their deployment remains challenging due to significant computational and memory demands. Quantization has emerged as a promising solution, and state-of-the-art quantization algorithms for LLMs introduce the need for mixed-precision matrix multiplication (mpGEMM), where lower-precision weights are multiplied with higher-precision activations. Despite its benefits, current hardware accelerators such as GPUs and TPUs lack native support for efficient mpGEMM, leading to inefficient dequantization operations in the main sequential loop. To address this limitation, we introduce MixPE, a specialized mixed-precision processing element designed for efficient low-bit quantization in LLM inference. MixPE leverages two key innovations to minimize dequantization overhead and unlock the full potential of low-bit quantization. First, recognizing that scale and zero point are shared within each quantization group, we propose performing dequantization after per-group mpGEMM, significantly reducing dequantization overhead. Second, instead of relying on conventional multipliers, MixPE utilizes efficient shift\&add operations for multiplication, optimizing both computation and energy efficiency. Our experimental results demonstrate that MixPE surpasses the state-of-the-art quantization accelerators by $2.6\times$ speedup and $1.4\times$ energy reduction.