跳到主要內容
AI News HubLIVE
站內改寫2 分鐘閱讀

LEAP:透過前視早收斂令牌檢測解鎖擴散語言模型並行性

文章摘要

擴散語言模型(dLLM)因其高度並行處理的潛力而備受關注。然而,現有方法依賴高置信度閾值來保證準確性,嚴重限制了並行擴充套件性。本文透過令牌級統計分析發現,大量令牌在去噪早期就已收斂到正確預測,但未達到標準置信度閾值,表明現有標準過於保守。為此,提出LEAP(前視早收斂令牌檢測加速並行解碼),一種無需訓練、即插即用的方法,透過未來上下文過濾和多序列疊加檢測早收斂令牌,實現可靠早期解碼。實驗表明,LEAP顯著降低推理延遲和解碼步數,較置信度解碼平均減少約30%去噪步數,在GSM8K資料集上與dParallel結合達到每秒7.2個令牌的處理速度,同時保持模型精度。

來源arXiv Machine Learning作者: Haohui Zhang, Zhiye Wang, Xiaoying Gan, Xinbing Wang, Bo Jiang
LEAP:透過前視早收斂令牌檢測解鎖擴散語言模型並行性
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

擴散語言模型(Diffusion Language Models, dLLMs)因其在高度並行處理方面的潛力而受到廣泛關注。與自迴歸模型逐步生成不同,dLLMs透過迭代去噪過程同時生成多個令牌,從而有望大幅提升解碼速度。然而,現有dLLMs的並行能力依賴於條件獨立假設,該假設要求模型在解碼時對每個令牌具有高置信度,以確保邊際分佈與聯合分佈之間的差異可忽略不計。這種嚴格的置信度閾值雖然保證了生成質量,但嚴重限制了並行解碼的擴充套件性——因為只有少數令牌能在早期滿足高置信度條件,大部分令牌仍需逐步去噪,導致並行優勢無法充分發揮。

為了解決這一問題,Haohui Zhang等研究者對令牌級動態進行了系統的統計分析。他們發現,在去噪過程中,相當大比例的令牌在早期迭代中就已經收斂到了正確的預測結果,但它們的置信度分數卻遠低於標準閾值。這表明現有的置信度標準過於保守,錯失了提前解碼的機會。基於這一洞察,團隊提出了LEAP(Lookahead Early-Convergence Token Detection for Accelerated Parallel Decoding)方法。

LEAP是一種無需訓練、即插即用的技術,核心思想是識別那些儘管置信度不高但實際上已經正確收斂的令牌。它透過兩種機制實現:一是利用未來上下文過濾,即參考後續去噪步驟中的資訊來驗證當前令牌的收斂性;二是採用多序列疊加,透過比較多個獨立去噪序列中令牌的一致性來增強檢測可靠性。LEAP首先檢測出早收斂令牌,然後允許它們提前解碼,而無需等待所有令牌達到高置信度。實驗驗證表明,早收斂令牌與最終正確令牌之間具有高度一致性,保證了早期解碼的可靠性。

在多個領域的基準測試中,LEAP展示了顯著的效能提升。與傳統的置信度基準解相比,使用LEAP後平均去噪步數減少了約30%。在數學推理資料集GSM8K上,將LEAP與現有的dParallel方法結合,解碼速度提升至每秒7.2個令牌,同時保持了模型精度。這一成果不僅直接降低了推理延遲,還挑戰了“高置信度是保證準確性的必要條件”這一傳統觀念,為並行解碼開闢了新正規化。LEAP的研究表明,透過更精細的令牌級分析和上下文感知的檢測策略,可以更早地釋放並行潛力,而無需犧牲生成質量。未來,該工作有望在即時應用和大規模部署中發揮重要作用。

展開要點與分析

文章情報

工程師進階

要點

  • 發現大量令牌在去噪早期正確收斂但未達高置信度閾值,現有置信度標準過於保守
  • 提出LEAP方法,無需訓練,透過未來上下文過濾和多序列疊加檢測早收斂令牌
  • LEAP減少約30%去噪步數,與dParallel結合在GSM8K上達到7.2令牌/步
  • 打破對高置信度先驗的依賴,為並行解碼提供新正規化

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。