Centrala begrepp
OneBit 是一種針對大型語言模型 (LLM) 的新型 1 位元模型壓縮框架,旨在在保持模型效能的同時,透過將模型權重矩陣量化為 1 位元,來顯著降低模型部署所需的儲存和計算成本。
Sammanfattning
研究目標
本研究旨在解決將大型語言模型 (LLM) 量化至極低位元(1 位元)時所面臨的效能嚴重下降問題,並開發一種高效且穩定的 1 位元模型壓縮框架。
方法
- 提出一種新穎的 1 位元線性層架構,將原始高位元權重矩陣表示為一個符號矩陣(±1)和兩個值向量,以在線性投影中提供必要的浮點精度。
- 提出符號值獨立分解 (SVID) 方法,將高位元矩陣分解為低位元矩陣,作為 1 位元架構的有效參數初始化方法。
- 採用量化感知知識蒸餾技術,將原始模型的能力轉移到所提出的 1 位元模型中。
主要發現
- OneBit 在 W1A16(1 位元權重和 16 位元激活)量化級別上實現了良好的效能,在 LLaMA 模型上至少達到了未量化效能的 81%。
- 與現有方法相比,OneBit 在量化 LLM 至 1 位元時表現出更穩定的訓練過程和更高的知識轉移效率。
- 隨著模型規模的增加,OneBit 的效能損失相對較小,這使其特別適用於壓縮大型 LLM。
主要結論
OneBit 為部署極低位元 LLM 提供了一種有效且實用的解決方案,顯著降低了儲存和計算成本,同時保持了可接受的效能水準。
意義
這項研究對於資源受限的環境(如移動設備)上的 LLM 部署具有重要意義,並為進一步探索極低位元模型壓縮技術開闢了新的方向。
局限性和未來研究方向
- 未來工作可以進一步探索更先進的量化和初始化技術,以進一步提高 1 位元模型的效能。
- 研究如何在硬體上有效地實現和部署 OneBit 也是一個重要的方向。
Statistik
在 LLaMA-13B 模型上,OneBit 在 WikiText2 數據集上的困惑度為 9.18,而 FP16 基線為 5.09。
OneBit 在 LLaMA-7B 模型上實現了平均約 4% 的零樣本準確率提升,優於最強的基線 OmniQuant。
OneBit-7B 模型在常識推理任務中優於參數更多、訓練更充分的其他模型。
OneBit 可以將 LLaMA-7B 模型的記憶體佔用從 13.5GB 減少到 1.3GB,壓縮率高達 90.4%。