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

騰訊開源AngelSpec:統一的MTP與塊並行推測解碼訓練框架

文章摘要

AngelSpec是一個基於Torch的開源訓練框架,支援六種推測解碼架構。它引入DFly塊擴散起草器,透過混合目標條件化和前驅條件自迴歸頭,提升解碼效率。在Hy3-295B-A21B模型上,DFly-8在併發4至64範圍內實現1.98–2.40倍加速。

來源MarkTechPost作者: Michal Sutter
騰訊開源AngelSpec:統一的MTP與塊並行推測解碼訓練框架
回報錯誤

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

查看更正說明
直接讀正文

騰訊近日釋出了AngelSpec,這是一個開源的、原生Torch的訓練框架,專門用於推測解碼的起草模型。該框架同時支援自迴歸多令牌預測(MTP)和塊並行DFlash系列。

大多數推測解碼工作都在尋找一個在平均基準混合上表現良好的通用起草器,但實際服務流量並不符合這種混合分佈。AngelSpec將工作負載異質性作為首要設計約束,並圍繞它專門調整結構、訓練資料和驗證深度。

為什麼通用起草器表現不佳?推測解碼是無損的:輕量級起草器提出多個未來令牌,目標模型透過拒絕取樣在一次前向傳播中驗證它們。加速取決於兩個因素:接受多少草案令牌以及完成草案-驗證回合的時間。這兩個因素在不同領域呈反向變化:在高熵的開放對話中,許多延續在語義上有效,接受率隨深度快速下降;而在程式碼和數學推理中,更長的可預測跨度更適合塊並行起草。

AngelSpec因此提供了兩種互補的起草器:MTP模型在豐富多樣的對話資料上訓練,而塊擴散模型則加強了程式碼和數學樣本。

在MTP路徑上,原始的Hy3模型僅有一個MTP層,沒有迴圈自條件展開。訓練目標從未為長自生成鏈做準備,導致深層接受率下降。AngelSpec透過共享引數的多深度方案解決了訓練與推理的不匹配:在訓練中,MTP塊被自迴歸展開D步,每個預測輸入下一次呼叫。關鍵改進包括凍結目標主幹、使用目標模型展開生成訓練資料。

在Hy3-A21B上,貪婪解碼下的平均接受率從52.8%提升至66.4%,平均接受長度從2.58增至2.99。深層位置的改進尤為顯著:GSM8K上p3從0.290升至0.706,HumanEval上從0.387升至0.757。

DFly是塊擴散架構,建立在DFlash之上,包含兩個結構變化:混合目標條件主幹和前驅條件自迴歸頭。混合主幹融合了DFlash的共享上下文和DFlare的逐層融合,而前驅條件頭將位置級邊際預測轉換為字首條件分佈。

早期效能方面,在Qwen3-8B上,DFly的平均接受長度達5.41,優於DSpark的5.32、DFlash的4.57和MTP的3.24。在Hy3-A21B上,DFly達到4.79,相對DFlash提升29.8%,相對MTP提升59.7%。

AngelSpec基於TorchSpec構建,實現了多項擴充套件:TTT展開、長上下文訓練(支援128K令牌)、文件感知序列打包、評估伺服器等。框架還提供可插拔介面:目標(vLLM)、目標函式和最佳化器。

目前已在Hugging Face和ModelScope上釋出七個檢查點,支援六種草案架構的訓練。DFly在Hy3-A21B上將平均接受長度提升至4.79,在HY3-295B-A21B上DFly-8實現1.98–2.40倍加速。D-cut技術在併發64下將吞吐量提升至981 tok/s,相比DFly提升15.7%。

展開要點與分析

文章情報

工程師進階

要點

  • AngelSpec統一訓練MTP和塊並行兩種互補起草模型。
  • DFly起草器在Hy3-A21B上平均接受長度達4.79,相比DFlash提升29.8%。
  • 框架包含訓練時測試、目標模型展開和文件感知序列打包。
  • 在HY3-295B-A21B上,DFly-8實現1.98–2.40倍加速(併發4–64)。

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