今天小編分享的互聯網經驗:推特熱帖:k1.5 很牛,因為借鑑了 UCLA 與 CMU 合作團隊的這項技術,歡迎閱讀。
2025 年 1 月 20 日 Kimi k1.5 正式發布,伴随着技術報告的公布,有網友表示:" 這應該是全球範圍内,除 OpenAI 之外的公司首次實現 o1 正式版的多模态推理性能了吧!"
一時間,Kimi k1.5 成了話題王者。
但在一個月後的 2 月 24 日,X 上出現了一篇關于 Kimi k1.5 的技術爆料帖,博主直言 k1.5 所用到的強化學習算法,其實是借鑑了自己在 24 年 5 月提出的一種名為 SPPO 的技術。
消息一出,瞬間吸引了數萬人關注。
Kimi k1.5 背後的 SPPO 技術
在這則爆料中,博主 Yue Wu 先是對 SPPO 進行了簡單解釋,并且附上了相關論文(https://arxiv.org/abs/2405.00675),簡單來說,SPPO 是一種自博弈算法,最初的動機來源于刻畫廣泛意義上的人類偏好,并且使用了如下圖所示的平方損失函數:
值得一提的是,點開論文鏈接,你會發現原來 Yue Wu 和 Zhiqing Sun 同為這篇文章的第一作者。
緊接着,他開始對 SPPO 技術進行解析:
通過迭代求解上式中的 theta_t,我們可以得到一個與人類偏好對齊良好的語言模型。SPPO 使用勝率(紅色部分)作為獎勵,并用常數近似基線(藍色部分)。
讓我們感興趣的是,我們發現它與 RLHF 目标的策略梯度有着深層的聯系:如果我們直接用普通的策略梯度優化 RLHF (人類反饋強化學習)目标會怎樣?根據策略梯度定理,策略梯度實際上也具有平方損失形式(藍色項是策略梯度中的基線):
從數學上,我們證明了 SPPO 的平方損失等價于普通策略梯度的一種半在線變體:
SPPO 中的勝率充當獎勵函數(紅色部分)。
分區函數項自然地成為(軟)值函數(藍色部分)。
那麼這到底意味着什麼呢?
标準策略梯度(PPO、GRPO、REINFORCE)在每一步都收集遵循當前策略的樣本。
SPPO 在每次迭代開始時只采樣一次,然後通過平方損失進行優化。
這使得 SPPO 成為一種輕量級的 RLHF 方法——無需即時生成!
上述分析揭示了大型語言模型(LLM)後訓練階段一個有趣的發展趨勢:
離線 DPO(IPO、KTO 等)取代 RLHF(獎勵模型 + 強化學習)
迭代 DPO、SPPO 等方法将離線方法轉化為在線對齊方法
更加精細的迭代 → 回歸到在線強化學習
鑑于 GRPO(Deepseek-R1)和平方損失(Kimi k1.5)的成功,端到端強化學習的強大作用愈發凸顯,或許在大型語言模型(LLM)後訓練階段無需額外技巧——價值函數、廣義優勢估計(GAE),甚至梯度裁剪都無需使用。
另一個簡單但有趣的發現是,他們發現 SPPO 暗中在詞匯級别優化最優最大熵策略。其平方損失隐含地最小化了學習到的策略與最優詞匯級别策略之間的 KL 散度。
在我們後續的研究 GPO 中,我們直接最小化相對獎勵與對數比率之間的平方損失。這兩項工作中的平方損失等價于策略梯度,但它是以迭代的方式進行的。
SPPO 技術背後的科研大牛
除了提出助力 Kimi k1.5 大獲成功的 SPPO 技術外,Wu Yue 也是一個學術背景很強的科研大牛。他本科期間師從北京大學的王立威教授,博士期間師從加利福尼亞大學洛杉矶分校的顧全全教授,目前以博士後研究員的身份在普林斯頓大學人工智能實驗室繼續着自己的科研之路。
除此之外,2023 年至今他一共參與發布了 9 篇 Paper,其中 3 篇均為第一作者。
強大的學術背景之外,Wu Yue 的實習經歷也非常加分。2022 年至 2024 年,他分别在 NEC 美研院、字節美國 AI lab 和 Meta 工作實習。在 NEC 美研院期間,Wu Yue 從事個性化聯邦學習研究,并開發了一種基于混合模型的方法,該方法被 ICML 2023 接受發表;在字節美國 AI lab 時,他專注于藥物發現領網域的多構象生成,将分子動力學的物理先驗納入基于擴散的生成模型,相關成果被 ICML 2024 接受;來到 Meta 後,Wu Yue 又致力于詞匯級别獎勵建模和新架構設計,用于一般人類偏好和一般偏好優化,為生成式人工智能的發展做出了貢獻。
雷峰網還了解到,與他同為第一作者的 Zhiqing Sun ,目前已經從 CMU 畢業,并在今年 2 月加入 OpenAI。