AI News HubLIVE
站內改寫2 分鐘閱讀

面向低比特整數的有符號對稱量化

本文提出有符號對稱量化方法,解決標準對稱量化器因整數範圍不平衡導致的正異常值截斷問題,同時避免非對稱量化的運行時開銷。理論分析表明該方法在ℓ2誤差上條件最優,且88-99%的LLM權重組滿足條件。實驗在Qwen3、Llama3等模型上驗證了困惑度和少樣本準確率的提升。

來源arXiv Machine Learning作者: Ian Colbert, Eashan Dash, Pablo Monteagudo-Lago, Juan Amboage, Srinidhi N, Giuseppe Franco, Nicholas J. Fraser, Arun Ramachandran

量化是深度學習推理加速的重要手段,尤其在大型語言模型(LLM)部署中,通過減少權重和激活值的位寬來降低內存佔用和計算延遲。標準對稱整數量化(如INT8)使用零點為0的縮放因子,但有符號整數比正表示多一個負值,導致默認將額外精度賦予負異常值,而正異常值可能因超出範圍被截斷。在4比特等低精度場景下,這種截斷會顯著增加量化誤差,成為非平凡的錯誤源。非對稱量化通過引入零點偏移解決了此問題,但帶來了運行時開銷——例如在AMD EPYC 'Turin' CPU上運行llama.cpp時,4比特對稱格式比非對稱節省9%內存,吞吐量高2.45倍。

本文提出的有符號對稱量化方案,通過調整縮放因子的符號,將額外表示值分配給主要異常值所在尾端,既保持了對稱量化的高效計算,又避免了截斷誤差。具體而言,作者設計了一個輕量級的符號選擇規則,確定縮放因子的正負,使得額外的負表示值被用於覆蓋絕對值較大的異常值尾端,而零點仍保持為0。這一方法無需額外的運行時計算,與標準對稱量化具有相同的計算效率。

理論貢獻方面,作者首先證明了有符號absmax網格在ℓ2量化誤差下是條件最優的,即在一定條件下,該網格能夠達到最低的均方誤差;其次,他們揭示了一個有趣的等價關係:標準對稱量化器取反縮放因子的操作,在數學上等價於在有符號整數字母表上做一個單位的零點偏移。進一步的分析表明,在低比特寬度下,預訓練LLM中88-99%的權重分組滿足該條件最優性條件,這意味着有符號對稱量化在大多數情況下能夠達到理論上的最佳誤差。

實驗部分基於Qwen3、Qwen3.5和Llama3系列模型,全面評估了有符號對稱量化相對於標準無符號對稱量化的性能。結果顯示,在無任何額外推理成本的前提下,該方法在困惑度和少樣本學習準確率上均取得一致改進,尤其在4比特設置下提升顯著。例如,在Llama3-8B模型上,4比特有符號對稱量化的困惑度比無符號版本降低了約0.3,同時少樣本準確率提高了1-2個百分點。這些結果表明,有符號對稱量化是一種高效且實用的低比特LLM推理方案,有望成為未來硬件和軟件部署的默認選擇。