Bitnet.cpp: Efficient Edge Inference for Ternary LLMs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Jinheng, Zhou, Hansong, Song, Ting, Cao, Shijie, Xia, Yan, Cao, Ting, Wei, Jianyu, Ma, Shuming, Wang, Hongyu, Wei, Furu
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912234005856256
author Wang, Jinheng
Zhou, Hansong
Song, Ting
Cao, Shijie
Xia, Yan
Cao, Ting
Wei, Jianyu
Ma, Shuming
Wang, Hongyu
Wei, Furu
author_facet Wang, Jinheng
Zhou, Hansong
Song, Ting
Cao, Shijie
Xia, Yan
Cao, Ting
Wei, Jianyu
Ma, Shuming
Wang, Hongyu
Wei, Furu
contents The advent of 1-bit large language models (LLMs), led by BitNet b1.58, has spurred interest in ternary LLMs. Despite this, research and practical applications focusing on efficient edge inference for ternary LLMs remain scarce. To bridge this gap, we introduce Bitnet.cpp, an inference system optimized for BitNet b1.58 and ternary LLMs. Given that mixed-precision matrix multiplication (mpGEMM) constitutes the bulk of inference time in ternary LLMs, Bitnet.cpp incorporates a novel mpGEMM library to facilitate sub-2-bits-per-weight, efficient and lossless inference. The library features two core solutions: Ternary Lookup Table (TL), which addresses spatial inefficiencies of previous bit-wise methods, and Int2 with a Scale (I2_S), which ensures lossless edge inference, both enabling high-speed inference. Our experiments show that Bitnet.cpp achieves up to a 6.25x increase in speed over full-precision baselines and up to 2.32x over low-bit baselines, setting new benchmarks in the field. Additionally, we expand TL to element-wise lookup table (ELUT) for low-bit LLMs in the appendix, presenting both theoretical and empirical evidence of its considerable potential. Bitnet.cpp is publicly available at https://github.com/microsoft/BitNet/tree/paper , offering a sophisticated solution for the efficient and practical deployment of edge LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2502_11880
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Bitnet.cpp: Efficient Edge Inference for Ternary LLMs
Wang, Jinheng
Zhou, Hansong
Song, Ting
Cao, Shijie
Xia, Yan
Cao, Ting
Wei, Jianyu
Ma, Shuming
Wang, Hongyu
Wei, Furu
Machine Learning
Artificial Intelligence
Computation and Language
Distributed, Parallel, and Cluster Computing
The advent of 1-bit large language models (LLMs), led by BitNet b1.58, has spurred interest in ternary LLMs. Despite this, research and practical applications focusing on efficient edge inference for ternary LLMs remain scarce. To bridge this gap, we introduce Bitnet.cpp, an inference system optimized for BitNet b1.58 and ternary LLMs. Given that mixed-precision matrix multiplication (mpGEMM) constitutes the bulk of inference time in ternary LLMs, Bitnet.cpp incorporates a novel mpGEMM library to facilitate sub-2-bits-per-weight, efficient and lossless inference. The library features two core solutions: Ternary Lookup Table (TL), which addresses spatial inefficiencies of previous bit-wise methods, and Int2 with a Scale (I2_S), which ensures lossless edge inference, both enabling high-speed inference. Our experiments show that Bitnet.cpp achieves up to a 6.25x increase in speed over full-precision baselines and up to 2.32x over low-bit baselines, setting new benchmarks in the field. Additionally, we expand TL to element-wise lookup table (ELUT) for low-bit LLMs in the appendix, presenting both theoretical and empirical evidence of its considerable potential. Bitnet.cpp is publicly available at https://github.com/microsoft/BitNet/tree/paper , offering a sophisticated solution for the efficient and practical deployment of edge LLMs.
title Bitnet.cpp: Efficient Edge Inference for Ternary LLMs
topic Machine Learning
Artificial Intelligence
Computation and Language
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2502.11880