Hyper-Parallel Decoding(HPD):為 AVE 解碼帶來十倍級加速的新方法

背景:某些生成任務需從同一文件解出多個獨立序列。核心:HPD透過批次共享記憶體與計算並操控位置ID,實現亂序與並行生成,且可在單一提示堆疊多份文件以放大並行度。主要結果:在不降低品質下,推理成本與時間最多減少13.8倍,並可在單提示並行解碼至多96個token。

超平行解碼加速屬性數值

速報:Hyper-Parallel Decoding(HPD)帶來 AVE 解碼顯著加速

研究團隊提出一種稱為 Hyper-Parallel Decoding(HPD)的解碼演算法,針對需要從同一文件中生成多個獨立序列的任務(例如屬性-數值提取,AVE)大幅提升離線推理效率。

HPD 的關鍵在於同時利用批次間共享的記憶體與計算工作,並透過操控位置 ID 允許亂序生成 token。這種做法把原本序列化的解碼過程改為可在提示內並行產生多組輸出。實驗發現屬性-數值對在條件上具有獨立性,因而可以在同一 prompt 內平行產生多個值;進一步把多份文件堆疊於單一提示,單次可並行解碼至多 96 個 token。

在多項 AVE 測試中,HPD 在不犧牲輸出品質的前提下,將推理成本與總推理時間縮短最多 13.8 倍。該方法與所有大型語言模型相容,且設計並不依賴 AVE 特有假設,理論上可推廣至任何輸出結構彼此獨立的場景。

對產業應用而言,HPD 可在離線批次推理中節省大量成本並加快處理速度,尤其適合處理大量文件、需同時抽取多組屬性值的工作負載。

延伸閱讀

原始來源:ArXiv AI


系統聲明:本文的深度點評與首圖視覺,皆為 AI 代理人獨立運算生成。機器視角偶有偏差,請輔以人類智慧進行交叉驗證。

Read more