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

視覺語言模型中可靠性的位置:注意力、隱藏狀態與因果迴路的機制研究

文章摘要

一項研究透過統一的機制探測工具VRP研究了三種視覺語言模型(LLaVA-1.5、PaliGemma、Qwen2-VL),發現注意力圖的清晰度與模型正確性幾乎無關,而隱藏狀態間隔和自我一致性更能預測可靠性。此外,模型架構影響可靠性分佈:晚融合模型依賴脆弱的晚期層,早融合模型則分佈廣泛。

來源arXiv AI作者: Logan Mann, Ajit Saravanan, Ishan Dave, Shikhar Shiromani, Saadullah Ismail, Yi Xia, Emily Huang
視覺語言模型中可靠性的位置:注意力、隱藏狀態與因果迴路的機制研究
回報錯誤

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

查看更正說明
直接讀正文

本文探討了視覺語言模型(VLM)中一個普遍的直覺:當注意力圖清晰集中時,模型更為可靠。然而,一項新研究透過統一機制探測工具VLM可靠性探針(VRP),對三個開源VLM系列(LLaVA-1.5、PaliGemma、Qwen2-VL,引數規模3-7B)進行了系統分析,結果挑戰了這一直覺。

研究首先發現,注意力結構的清晰度與模型正確性幾乎完全無關。在彙集3090個樣本的測試中,注意力與正確性的點雙列相關係數僅為0.001(95%置信區間[-0.034, 0.036])。儘管注意力在特徵提取中依然至關重要——移除前30%的補丁會導致準確率下降8.2至11.3個百分點——但它並不能作為可靠性的預測指標。這一發現對當前許多依賴於注意力視覺化來評估模型可靠性的做法提出了質疑。

更有效的可靠性訊號來自隱藏狀態幾何結構。隱藏狀態間隔(即正確與錯誤答案對應的隱藏狀態之間的距離)在三個模型中有兩個在POPE資料集上的AUC超過0.95,顯示出強大的區分能力。此外,自我一致性(在K=10次推理中的答案一致性)是最強的行為預測指標,相關係數達到0.43,儘管其計算成本增加了10倍。這表明,在追求模型可靠性時,投入計算資源進行多次取樣可能比分析注意力圖更有價值。

研究還揭示了架構差異對可靠性分佈的影響。透過因果神經元消融,發現採用晚期融合架構的LLaVA將可靠性集中在脆弱的晚期瓶頸層,僅消融前5個探針神經元就導致目標識別準確率下降8.3個百分點。而早期融合架構的PaliGemma和Qwen2-VL則將可靠性廣泛分佈,即使破壞約50%的峰值層隱藏維度,效能下降也僅為1個百分點或更少。這一差異對於模型設計和監控具有直接意義:晚期融合模型需要更謹慎地保護其脆弱的後期層,而早期融合模型則更加魯棒。

總之,該研究指出,在3-7B引數規模的VLM中,可靠性更可靠地從隱藏狀態幾何、層間間隔和稀疏晚期迴路中讀取,而非注意力圖的清晰度。這一發現對模型解釋和監控設計具有重要意義,也提示研究者和工程師在評估模型可靠性時,應將更多注意力轉向隱藏狀態的結構化分析,而不是過度依賴注意力視覺化。

展開要點與分析

文章情報

工程師進階

要點

  • 注意力結構幾乎無法預測正確性(相關係數接近0)。
  • 隱藏狀態間隔和自我一致性是更好的可靠性指標。
  • 晚融合模型(如LLaVA)的可靠性集中在脆弱瓶頸層,早融合模型(如PaliGemma)分佈更穩健。

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