QeRL: Beyond Efficiency -- Quantization-enhanced Reinforcement Learning for LLMs

[論文] QeRL:超越效率 — 量化增強的大型語言模型強化學習

NVIDIA 和 MIT 的研究人員開發了 QeRL,這是一個透過量化技術增強大型語言模型強化學習(RL)的突破性框架。結合 NVFP4 量化和低秩適應(LoRA),QeRL 實現了更快速的 RL 訓練並降低記憶體開銷。關鍵創新是自適應量化雜訊機制,將量化雜訊轉化為改善訓練探索的工具。QeRL 在數學推理任務上的速度和準確度都超越了標準技術。值得注意的是,它能夠在單張 H100 GPU 上訓練 320 億參數模型,讓大規模 RL 訓練變得更加民主化。這種方法挑戰了將量化視為妥協的傳統觀點,證明了其在 RL 環境中提升模型性能的潛力。

來自 NVIDIA 和 MIT 的研究人員推出了一個突破性框架,不僅解決了大型語言模型強化學習(RL)的運算挑戰,更展示了量化技術如何實際提升模型性能。這篇論文「QeRL: Beyond Efficiency — Quantization-enhanced Reinforcement Learning for LLMs」挑戰了量化必然會降低模型品質的傳統觀念,反而證明了精心控制的量化雜訊能夠顯著改善 RL 訓練過程中的探索能力。

QeRL 結合了 NVFP4 量化和低秩適應(LoRA)的協同效應,實現更快速的 RL 訓練同時降低記憶體開銷。最令人矚目的是,這個框架能夠在單張 H100 80GB GPU 上訓練 320 億參數的語言模型——這在過去被認為是不可能的壯舉——同時在數學基準測試中達到與全參數微調相當的性能。研究人員證明,QeRL 在訓練速度和最終準確度方面都超越了標準的 16 位元 LoRA 和 QLoRA 技術,特別是在具挑戰性的推理任務上。

驅動 QeRL 成功的關鍵洞察在於,量化雜訊在監督式微調中通常被視為有害,但在強化學習環境中實際上會增加策略熵。這種增加的熵提升了模型在訓練過程中探索不同解決方案的能力,最終帶來更好的性能。基於這個觀察,研究人員引入了自適應量化雜訊(AQN)機制,能夠在整個訓練過程中動態調整雜訊水準,以最佳化探索與利用之間的平衡。

傳統的大型語言模型 RL 訓練面臨重大的運算挑戰。標準的全參數微調方法需要大量的 GPU 記憶體,因為必須同時運行多個模型(策略模型和參考模型)。此外,這個過程涉及運算成本高昂的多階段操作,包括推演、獎勵計算、邏輯評估和梯度更新。這些挑戰對於專注於推理的大型語言模型特別嚴峻,因為它們需要處理複雜任務的長序列。

像 LoRA 這樣的參數高效微調方法透過減少可訓練參數的數量部分解決了這些問題。然而,它們無法解決推演速度緩慢這個根本瓶頸。同時,QLoRA 結合了 NormalFloat 4 位元(NF4)量化與 LoRA,但由於 NF4 查找表操作的運算開銷,實際上使推演速度減慢了 1.5-2 倍。

QeRL 透過三個關鍵創新來解決這些限制。首先,它用 NVFP4 取代較慢的 NF4 格式,這是一種在 NVIDIA 最新 GPU 上具有原生硬體支援的 4 位元浮點格式。其次,它在推演和預填充階段都採用基於 Marlin 的方法,顯著加速這些操作。最後,它引入了自適應量化雜訊機制,將靜態量化雜訊轉化為動態探索工具。

AQN 技術的運作方式是在訓練過程中注入通道級隨機雜訊,並使用指數排程動態調整探索雜訊。這使得模型能夠在早期訓練中受益於更高的探索性,同時隨著訓練進展逐漸專注於利用。為了在不增加參數開銷的情況下實現這一點,研究人員開發了雜訊共享策略,將雜訊向量合併到模型的層正規化組件中。

實驗結果令人印象深刻。在 GSM8K 數學推理基準測試中,QeRL 使用 Qwen2.5-7B-Instruct 模型達到了 90.8% 的分數,超越了 16 位元 LoRA(88.1%)和 QLoRA(85.0%)。同樣地,在 MATH 500 基準上,QeRL 達到了與全微調相當的準確度。這些性能提升伴隨著顯著的效率改善,QeRL 相較於 QLoRA 在端到端訓練中提供了約 1.8 倍的加速。

這些結果特別值得注意的是,它們與該領域的傳統理解相矛盾。雖然量化傳統上被視為效率與準確度之間的必要妥協,但 QeRL 證明了在 RL 環境中,量化實際上可以透過增強探索來改善模型性能。這個發現為量化與強化學習交叉領域的研究開闢了新途徑。

該框架能夠在單張 H100 GPU 上使用 GRPO(群組相對策略最佳化)訓練 320 億模型,代表了大規模 RL 訓練存取權的重大民主化。過去,這樣的訓練需要多 GPU 設置或專用硬體,將研究限制在資源豐富的機構。QeRL 使這種能力對更廣泛的研究人員和實務工作者變得可及。

QeRL 的影響超越了單純的效率提升。透過實現大型語言模型更快速且更易取得的 RL 訓練,這個框架可能加速需要強大推理能力領域的進展,如數學問題解決、邏輯推演和複雜決策制定。研究人員將量化雜訊視為有益特性而非需要最小化的限制,這種方法代表了我們思考模型量化方式的重要典範轉移。

在技術實作方面,QeRL 建立在大型語言模型的主流策略最佳化演算法之上,如 GRPO 和 DAPO(動態採樣策略最佳化)。該框架在指數衰減雜訊排程方面表現特別出色,在早期訓練中提供較大的雜訊值以促進探索,在後期提供較小的值以實現穩定收斂。

隨著大型語言模型持續演進且推理能力變得日益重要,像 QeRL 這樣使先進訓練技術更加可及的框架,將在民主化最先進 AI 技術的存取權方面發揮關鍵作用。這篇論文對該領域做出了重大貢獻,挑戰了傳統智慧並為大型語言模型中高效且有效的強化學習開啟了新的可能性。

🚀 以每年 5 美元解鎖無廣告體驗

✓ 14 天免費試用 ✓ 隨時取消

在〈[論文] QeRL:超越效率 — 量化增強的大型語言模型強化學習〉中有 1 則留言

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *