GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformers
Elias Frantar Thanks: Corresponding author: Affiliation: IST Austria Saleh Ashkboos Affiliation: ETH Zurich Torsten Hoefler Affiliation: ETH Zurich Dan Alistarh Affiliation: IST Austria、NeuralMagic
Abstract
Generative Pre-trained Transformer models, known as GPT or OPT, set themselves apart through breakthrough performance across complex language modelling tasks, but also by their extremely high computational and storage costs. Specifically, due to their massive size, even inference for large, highly-accurate GPT models may require multiple performant GPUs, which limits the usability of such models. While there is emerging work on relieving this pressure via model compression, the applicability and performance of existing compression techniques is limited by the scale and complexity of GPT models. In this paper, we address this challenge, and propose GPTQ, a new one-shot weight quantization method based on approximate second-order information, that is both highly-accurate and highly-efficient. Specifically, GPTQ can quantize GPT models with 175 billion parameters in approximately four GPU hours, reducing the bitwidth down to 3 or 4 bits per weight, with negligible accuracy degradation relative to the uncompressed baseline. Our method more than doubles the compression gains relative to previously-proposed one-shot quantization methods, preserving accuracy, allowing us for the first tim
原文 arXiv:2210.17323;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2210.17323v2