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

AIPO:透過主動互動學習推理

文章摘要

AIPO是一個增強的強化學習框架,透過引入驗證、知識和推理三個協作智慧體,在訓練過程中主動引導策略模型突破推理能力邊界,從而提升大語言模型的推理效能。實驗表明該方法在多個基準測試中一致優於現有方法。

來源arXiv Computational Linguistics作者: Junnan Liu, Linhao Luo, Thuy-Trang Vu, Gholamreza Haffari
AIPO:透過主動互動學習推理
回報錯誤

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

查看更正說明
直接讀正文

近年來,大語言模型(LLM)在推理能力上取得了顯著進展,這一進展很大程度上得益於基於可驗證獎勵的強化學習(RLVR)。然而,現有的強化學習演算法存在一個根本性侷限:其探索範圍受限於策略模型固有的能力邊界。雖然已有方法嘗試引入外部專家演示來擴充套件這一邊界,但它們通常依賴於完整的軌跡級指導,導致樣本效率低下、資訊稀疏,並可能將探索限制在靜態的指導空間中。

受多智慧體系統潛力的啟發,研究人員提出了AIPO框架,這是一種增強的強化學習框架,透過在探索過程中進行主動的多智慧體互動來提升LLM推理能力。具體而言,AIPO使策略模型在遇到推理瓶頸時能夠主動諮詢三個功能協作智慧體:驗證智慧體(Verify Agent)、知識智慧體(Knowledge Agent)和推理智慧體(Reasoning Agent),從而獲得細粒度、有針對性的指導,在訓練過程中主動擴充套件能力邊界。驗證智慧體負責檢查中間步驟的正確性,知識智慧體提供相關事實和背景知識,推理智慧體則引導更有效的推理路徑。這種協作使得策略模型能夠超越自身能力限制,學習更復雜的推理策略。

為了克服從智慧體反饋學習時出現的離策略偏差和梯度消失問題,AIPO引入了一個定製的重要性取樣係數以及裁剪策略。重要性取樣係數調整不同反饋樣本的權重,減少偏差;裁剪策略則限制更新幅度,防止梯度消失。訓練完成後,策略模型可以獨立進行推理,無需依賴協作智慧體,因此部署時不會增加額外開銷。

在包括AIME、MATH500、GPQA-Diamond和LiveCodeBench在內的多種推理基準上的大量實驗表明,AIPO consistently提升了推理效能,在不同策略模型和RLVR演算法上具有魯棒泛化能力,並有效擴充套件了策略模型的推理能力邊界。例如,在AIME(美國數學邀請賽)題目上,AIPO顯著提高了準確率;在MATH500上,它超越了現有最佳方法。此外,該方法在程式碼生成基準LiveCodeBench上也表現優異,顯示了其跨領域的泛化能力。這些結果證明,透過主動多智慧體互動,可以顯著突破LLM的推理能力上限。該方法的程式碼和資料已公開發布,便於後續研究和應用。

展開要點與分析

文章情報

工程師進階

要點

  • AIPO框架允許LLM在探索過程中主動諮詢三個專業智慧體以突破能力邊界。
  • 引入重要性取樣係數和裁剪策略解決從智慧體反饋學習時的偏差和梯度消失問題。
  • 訓練後策略模型可獨立推理,無需依賴協作智慧體。
  • 在AIME、MATH500等基準上顯著提升推理效能,且泛化性強。

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