ReaLVR:讓潛在視覺推理真正「看」到圖像證據
Rethinking Latent Visual Reasoning: Grounding Latent Reasoning in Visual Evidence
背景
多模態大型語言模型(MLLM)處理視覺推理題時,通常會像文字模型一樣先寫一段 chain-of-thought,再給答案。這種做法有一個瓶頸:並非每個視覺推理步驟都適合用文字說出來,例如「這個區域和那個區域的相對位置」、「某個局部紋理的細節」。Latent visual reasoning(LVR)因此興起。它讓模型在輸出答案之前,先在連續的 latent token 中做中間運算,不必把每一步都翻成文字。
LVR 也帶來新的問題。文字 CoT 看得見,我們可以檢查它對不對,也可以對每一步做監督。latent token 則是連續向量,沒辦法直接閱讀,所以很難知道它到底學到了什麼。目前常見的訓練做法是用 GRPO 之類的強化學習,只用最終答案對錯當作 reward。
這篇論文先做了一輪分析,發現一個現象,並稱之為 latent evidence-credit gap。具體來說,當研究者對圖像做擾動,而這個擾動足以改變正確答案時,latent token 的反應卻很微弱。換句話說,這些 latent token 並沒有緊密綁定在能決定答案的視覺證據上。作者推測原因是 GRPO 訓練缺少明確監督:最終答案 reward 太粗,沒有告訴模型該保留哪些視覺證據,也沒有說明功勞(credit)要怎麼分配給一串 latent token。
方法
作者提出 ReaLVR。核心想法是把「視覺證據監督」直接加到模型自己自由生成(free-running)的 latent 軌跡上,而不是只依賴外部給定的軌跡。從摘要和我能讀到的頁面內容來看,它用了兩組對比:
- 正確答案 vs. 模型自己產生的錯誤答案:用來判斷哪些地方需要更強的監督。如果模型在某個推理軌跡上產生錯誤答案,表示這條 latent 軌跡可能沒有抓到關鍵證據。對比兩者可以指出「哪裡該加強」,也就是處理 credit assignment 的問題。
- 相關視覺證據 vs. 不匹配的視覺證據:用來界定「該保留什麼」。讓 latent token 對與問題相關的圖像內容有反應,對不相關或不匹配的內容則沒有,藉此把 latent 推理綁在真正有用的視覺證據上。
簡單講,第一組對比回答「監督要加在哪裡」,第二組對比回答「要保留哪些視覺資訊」。這兩個問題正好對應前面診斷出的缺口:最終答案 reward 既不知道該保留什麼,也不知道該怪哪個 token。
需要說明的是,我讀到的 arXiv 摘要頁與 Hugging Face 頁面內容很有限,沒有看到損失函數的具體公式、超參數,也沒有看到訓練資料的細節。上面對方法的描述只限於摘要所能支持的層次。想重現的讀者,請以論文全文(共 39 頁)為準。
實驗結果
摘要給出的主要數字如下:
- 三個模型家族:ReaLVR 在三個模型家族上,都穩定勝過所評估的 LVR baseline。
- Qwen2.5-VL-7B 上的五項任務平均:ReaLVR 達到 63.7%,是所有比較方法中最高的。摘要沒有列出是哪五個 benchmark,也沒有列出各 baseline 的分數,所以我無法精確換算出領先幅度。
- 擴展到 235B:作者宣稱自己是第一個把視覺推理在 latent space 中擴展到 frontier 規模的工作,模型最大到 235B 參數,而且在這個規模上仍然帶來穩健的提升。許多 latent reasoning 方法只在 7B 左右的模型上驗證,所以這個擴展性的結果是論文的一個賣點。
除了 benchmark 分數,作者也做了行為層面的分析,結果與前面診斷的問題對得上:
- latent token 的位置更具問題敏感性(question-sensitive):不同的問題會讓不同位置的 latent token 有不同反應。
- 與相關視覺區域的對齊更強:latent token 的注意力更集中在與問題相關的圖像區域。
- 對最受關注的 latent token,有更高的 fixed-context 依賴:這些 token 對結果的影響更大,表示它們確實承載了有用的資訊。
這些分析的用途是說明性能提升不只是分數變高,而是 latent token 的行為確實朝著「基於視覺證據」改變。不過摘要沒有給出這些分析的具體數值。
意義
這篇論文有三個值得注意的地方。
第一,它把問題診斷得很清楚。LVR 領域常見的假設是「只要有 latent token,模型就在做視覺推理」。作者用圖像擾動測試指出這個假設不一定成立:如果 latent token 對足以改變答案的擾動沒有反應,那它們有可能只是多出來的計算空間,並沒有真的在使用視覺證據。這提供了一個可以操作的診斷方法,其他 LVR 工作也可以拿來檢查自己的模型。
第二,它指出了稀疏 reward 在 latent 推理上的限制。這和 RLHF、RLVR 中熟悉的 credit assignment 問題是同一類。文字 CoT 至少可以靠閱讀中間步驟來除錯,latent 推理看不到內容,所以更需要額外的結構化監督。ReaLVR 的做法是在 on-policy 的自由生成軌跡上,用對比方式加上證據監督,不需要人工標註 latent 內容。
第三,規模驗證。如果結果如摘要所述,在 235B 規模上仍然有效,那 latent 視覺推理就不只是小模型上的研究玩具,而有機會進入大型模型的訓練流程。
也要保持一些保留。目前我能確認的只有摘要層級的結論;各 benchmark 的細項、相對提升幅度、訓練成本、以及 235B 實驗的具體設定,都需要讀全文才能判斷。「第一個擴展到 235B」是作者自己的說法,也需要對照後續工作來看。整體來說,這是一篇把 latent 推理的可解釋性和可監督性往前推進的論文,對做多模態 RL 訓練的團隊值得細讀。