在自然語言處理領域,擴散語言模型透過逐步去噪掩碼令牌來生成文本,其並行生成能力正逐漸成為自迴歸解碼的有力替代方案。然而,現有解碼方法在最佳化生成質量與計算效率的平衡上仍面臨挑戰。日前,一篇發表在arXiv上的論文提出了AdaLook——一種新穎的自適應多步前瞻解碼框架,旨在解決擴散語言模型在長距離依賴性上的不足。
當前常見的單步前瞻解碼方法僅關注下一步的即時資訊增益,這可能導致在長序列生成中做出次優決策。儘管直覺上擴充套件前瞻步驟可能改善效果,但研究人員發現,固定深度的前瞻策略會引入大量額外計算,且難以適應解碼過程中不同狀態的需求。針對這一問題,AdaLook透過動態機制確定是否繼續前瞻:當候選分數的方差較低時,系統認為當前狀態足夠穩定,便提前終止前瞻;反之,若需要進一步探索,則會觸發分支擴充套件,從中間狀態重新啟動前瞻。這種自適應設計既避免了不必要的深度計算,又確保解碼器能從資訊豐富的中間狀態獲益。
實驗部分,研究團隊在多個基準資料集和不同規模的擴散語言模型上測試了AdaLook的效能。結果顯示,與現有單步前瞻方法相比,AdaLook在保持相當或更高生成精度的前提下,顯著減少了所需的解碼步驟。例如,在機器翻譯和文本摘要任務中,AdaLook生成的文本在BLEU和ROUGE評分上均有所提升,同時解碼速度提高了約20%-30%。此外,論文還透過消融實驗驗證了候選分數方差閾值和分支擴充套件策略的有效性。
總體而言,AdaLook為擴散語言模型的解碼提供了一種靈活且高效的新方案。其核心思想——即根據即時解碼狀態自適應調整探索深度——不僅適用於當前的掩碼擴散模型,也可能啟發其他並行生成正規化的最佳化。未來,研究團隊計劃將這一框架擴充套件到更多型別的擴散模型,並探索與其他解碼加速技術的結合。