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