OneBit: Towards Extremely Low-bit Large Language Models
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2024
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866929608673198080 |
|---|---|
| author | Xu, Yuzhuang Han, Xu Yang, Zonghan Wang, Shuo Zhu, Qingfu Liu, Zhiyuan Liu, Weidong Che, Wanxiang |
| author_facet | Xu, Yuzhuang Han, Xu Yang, Zonghan Wang, Shuo Zhu, Qingfu Liu, Zhiyuan Liu, Weidong Che, Wanxiang |
| contents | Model quantification uses low bit-width values to represent the weight matrices of existing models to be quantized, which is a promising approach to reduce both storage and computational overheads of deploying highly anticipated LLMs. However, current quantization methods suffer severe performance degradation when the bit-width is extremely reduced, and thus focus on utilizing 4-bit or 8-bit values to quantize models. This paper boldly quantizes the weight matrices of LLMs to 1-bit, paving the way for the extremely low bit-width deployment of LLMs. For this target, we introduce a 1-bit model compressing framework named OneBit, including a novel 1-bit parameter representation method to better quantize LLMs as well as an effective parameter initialization method based on matrix decomposition to improve the convergence speed of the quantization framework. Sufficient experimental results indicate that OneBit achieves good performance (at least 81% of the non-quantized performance on LLaMA models) with robust training processes when only using 1-bit weight matrices. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_11295 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | OneBit: Towards Extremely Low-bit Large Language Models Xu, Yuzhuang Han, Xu Yang, Zonghan Wang, Shuo Zhu, Qingfu Liu, Zhiyuan Liu, Weidong Che, Wanxiang Computation and Language Model quantification uses low bit-width values to represent the weight matrices of existing models to be quantized, which is a promising approach to reduce both storage and computational overheads of deploying highly anticipated LLMs. However, current quantization methods suffer severe performance degradation when the bit-width is extremely reduced, and thus focus on utilizing 4-bit or 8-bit values to quantize models. This paper boldly quantizes the weight matrices of LLMs to 1-bit, paving the way for the extremely low bit-width deployment of LLMs. For this target, we introduce a 1-bit model compressing framework named OneBit, including a novel 1-bit parameter representation method to better quantize LLMs as well as an effective parameter initialization method based on matrix decomposition to improve the convergence speed of the quantization framework. Sufficient experimental results indicate that OneBit achieves good performance (at least 81% of the non-quantized performance on LLaMA models) with robust training processes when only using 1-bit weight matrices. |
| title | OneBit: Towards Extremely Low-bit Large Language Models |
| topic | Computation and Language |
| url | https://arxiv.org/abs/2402.11295 |