為離線元強化學習提供統一理論基礎和算法設計準則。

原標題：NeurIPS Spotlight | 基于信息論，決策模型有了全新預訓練范式統一框架
文章來源：機器之心
內容字數：7426字

離線元強化學習的創新算法UNICORN

近年來，人工智能技術的快速發展，尤其是大語言模型（如GPT）的應用，對各行各業產生了深遠影響。然而，在處理復雜的專業問題時，AI依然面臨諸多挑戰。針對藥物發現、自動駕駛等領域的決策需求，研究者們提出了離線元強化學習（Offline Meta-RL）這一新范式。

1. 研究背景

傳統強化學習通過與環境的實時交互來訓練智能體，但在如自動駕駛和疾病治療等高風險場景中，這種方法的試錯成本過高。因此，研究者們開始重視利用歷史數據進行離線學習。同時，復雜的任務環境要求智能體具備多任務處理能力，這催生了元強化學習（Meta-RL）的發展。

2. UNICORN算法的提出

由之江實驗室、香港中文大學和同濟大學的研究團隊提出的UNICORN算法，基于信息論，系統性地構建了一套關于任務表征學習的理論框架。該算法通過定義和解構離線元強化學習中的任務表示，統一了現有主流方法的優化目標，為未來的研究提供了新的方向。

3. 核心創新

UNICORN的創新在于從數學定義、因果關系分解和中心定理三個層面，提出了任務表示學習的統一理論框架。該框架不僅為任務表示的優化提供了理論依據，還引導了新的算法設計。

4. 實驗驗證

研究團隊在多種機器人控制任務中對UNICORN算法進行了廣泛測試，結果表明，UNICORN在同分布和分布外測試集上的表現均優于現有方法，顯示出其廣泛適用性和魯棒性。此外，UNICORN在不同數據質量和模型架構上的表現也展現了良好的遷移性。

5. 未來展望

UNICORN為離線元強化學習提供了理論基礎，未來有望在藥物設計、精準醫療等領域解決AI模型的泛化性和樣本利用率問題。同時，團隊也在探索將該框架擴展到在線強化學習等更多應用場景。

通過此研究，離線元強化學習領域的理論和實踐將得到進一步發展，為決策大模型的能力拓展奠定基礎。

聯系作者

文章來源：機器之心
作者微信：
作者簡介：專業的人工智能媒體和產業服務平臺