圖中體現(xiàn)了兩種幻覺,紅色部分錯誤地描述了狗的顏色(屬性幻覺),藍(lán)色部分描述了圖中實(shí)際不存在的事物(目標(biāo)幻覺)?;糜X對模型的可靠性產(chǎn)生了顯著的負(fù)面影響,因此引起了許多研究者的重視。
以往的方法主要集中在 MLLM 本身,通過在訓(xùn)練數(shù)據(jù)以及架構(gòu)上進(jìn)行改進(jìn),以重新微調(diào)的方式訓(xùn)練一個新的 MLLM。這種方式會造成較大的數(shù)據(jù)構(gòu)建和訓(xùn)練開銷,且較難推廣到各種已有的 MLLMs。
近日,來自中科大等機(jī)構(gòu)的研究者們提出了一種免訓(xùn)練的即插即用的通用架構(gòu)“啄木鳥(Woodpecker)”,通過修正的方式解決 MLLM 輸出幻覺的問題。
論文鏈接:
https://arxiv.org/pdf/2310.16045.pdf
代碼鏈接:
https://github.com/BradyFU/Woodpecker
Woodpecker 可以修正各種場景下模型輸出的幻覺,并輸出檢測框作為引證,表明相應(yīng)的目標(biāo)確實(shí)存在。例如,面對描述任務(wù),Woodpecker 可以修正其中帶有幻覺的部分。
對于 MLLM 難以檢測到的小對象,Woodpecker 也可以精準(zhǔn)修正:
面對 MLLM 難以解決的復(fù)雜的計(jì)數(shù)場景,Woodpecker 同樣可以進(jìn)行解決:
對于目標(biāo)屬性類的幻覺問題,Woopecker 處理地也很好:
我們還提供了 Demo 供讀者測試使用,如下圖所示,上傳圖片并輸入請求,就可以得到修正前以及修正后的模型答復(fù),以及供參考驗(yàn)證的新圖片。
方法
Woodpecker 的架構(gòu)如下,它包括五個主要步驟:關(guān)鍵概念提取、問題構(gòu)造、視覺知識檢驗(yàn)、視覺斷言生成以及幻覺修正。
關(guān)鍵概念提?。?/span>關(guān)鍵概念指的是 MLLM 的輸出中最可能存在幻覺的存在性目標(biāo),例如上圖描述中的“自行車;垃圾桶;人”。我們可以 Prompt 大語言模型來提取出這些關(guān)鍵概念,這些關(guān)鍵概念是后續(xù)步驟進(jìn)行的基礎(chǔ);
問題構(gòu)造:圍繞著前一步提取出的關(guān)鍵概念,Prompt 大語言模型來提出一些有助于檢驗(yàn)圖片描述真?zhèn)蔚膯栴},如“圖中有幾輛自行車?”、“垃圾桶邊上的是什么?”等等;
視覺知識檢驗(yàn):使用視覺基礎(chǔ)模型對提出的問題進(jìn)行檢驗(yàn),獲得與圖片以及描述文本相關(guān)的信息。例如,我們可以利用 GroundingDINO 來進(jìn)行目標(biāo)檢測,確定關(guān)鍵目標(biāo)是否存在以及關(guān)鍵目標(biāo)的數(shù)量。這里我們認(rèn)為像 GroundingDINO 這類視覺基礎(chǔ)模型對圖片的感知能力比 MLLM 本身的感知能力更強(qiáng)。對于目標(biāo)顏色等這類屬性問題,我們可以利用 BLIP-2 來進(jìn)行回答。BLIP-2這類傳統(tǒng) VQA 模型輸出答案的長度有限,幻覺問題也更少;
視覺斷言生成:基于前兩步中獲得的問題以及對應(yīng)的視覺信息,合成結(jié)構(gòu)化的“視覺斷言”。這些視覺斷言可以看做與原有 MLLM 的回答以及輸入圖片相關(guān)的視覺知識庫;
幻覺修正:基于前面得到的,使用大語言模型對 MLLM 的文本輸出進(jìn)行逐一修正,并提供目標(biāo)對應(yīng)的檢測框信息作為視覺檢驗(yàn)的參照。
?
實(shí)驗(yàn)效果
實(shí)驗(yàn)選取了幾個典型的 MLLM 作為基線,包括:LLaVA,mPLUG-Owl,Otter,MiniGPT-4 論文中首先測試了 Woodpecker 在面對目標(biāo)幻覺時的修正能力,在 POPE 驗(yàn)證集的實(shí)驗(yàn)結(jié)果如下表所示:結(jié)果表明在不同的 MLLM 上應(yīng)用 Woodpecker 修正后,均有不同程度的提升。在隨機(jī)設(shè)定下,Woodpecker 給 MiniGPT-4和 mPLUG-Owl 和在準(zhǔn)確率指標(biāo)上分別帶來了 30.66% 和 24.33% 的提升。
此外,研究者還應(yīng)用更全面的驗(yàn)證集 MME,進(jìn)一步測試 Woodpecker 在面對屬性幻覺時的修正能力,結(jié)果如下表所示:
從表中可見 Woodpecker 不僅在應(yīng)對目標(biāo)幻覺時有效,在修正顏色等屬性幻覺時也具有出色的表現(xiàn)。LLaVA 的顏色得分從 78.33 分大幅提升到 155 分!經(jīng)過 Woodpecker 修正后,四個基線模型在四個測試子集上的總分均超過 500 分,在總體感知能力上獲得了顯著提升。
為了更直接地衡量修正表現(xiàn),更直接的方式是使用開放評測。不同于以往將圖片轉(zhuǎn)譯后送入純文本 GPT-4 的做法,文章利用 OpenAI 最近開放的視覺接口,提出使用 GPT-4 (Vision) 對修正前后的圖片描述直接對下列兩個維度進(jìn)行打分:- 準(zhǔn)確度:模型的答復(fù)相對于圖片內(nèi)容是否準(zhǔn)確
-
詳細(xì)程度:模型答復(fù)的細(xì)節(jié)豐富度
結(jié)果表明經(jīng)過 Woodpecker 修正后圖片描述的準(zhǔn)確性有一定的提升,這說明該框架可以有效修正描述中幻視的部分。另一方面,Woodpecker 修正后引入的定位信息豐富了文本描述,提供了進(jìn)一步的位置信息,從而提升了細(xì)節(jié)豐富度。GPT-4V 輔助的評測樣例如下圖所示:
·
原文標(biāo)題:幻覺降低30%!首個多模態(tài)大模型幻覺修正工作Woodpecker
文章出處:【微信公眾號:智能感知與物聯(lián)網(wǎng)技術(shù)研究所】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
-
物聯(lián)網(wǎng)
+關(guān)注
關(guān)注
2900文章
44065瀏覽量
370253
原文標(biāo)題:幻覺降低30%!首個多模態(tài)大模型幻覺修正工作Woodpecker
文章出處:【微信號:tyutcsplab,微信公眾號:智能感知與物聯(lián)網(wǎng)技術(shù)研究所】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
發(fā)布評論請先 登錄
相關(guān)推薦
評論