刷爆多模態任務榜單!賈佳亞團隊Mini-Gemini登熱榜,程式碼、模型、資料全部開源

机器之心發表於2024-04-16

Mini-Gemini 堪比 GPT4+DALLE3 王炸組合。


更高畫質影像的精確理解、更高質量的訓練資料、更強的影像解析推理能力,還能結合影像推理和生成,香港中文大學終身教授賈佳亞團隊提出的這款多模態模型 Mini-Gemini 堪稱絕絕子,相當於開源社群的 GPT4+DALLE3 的王炸組合!

圖片

  • Github 地址:https://github.com/dvlab-research/MiniGemini
  • Demo 地址: http://103.170.5.190:7860/
  • 論文地址:https://arxiv.org/pdf/2403.18814.pdf
  • 模型地址:https://huggingface.co/collections/YanweiLi/mini-gemini-6603c50b9b43d044171d0854
  • 資料地址:https://huggingface.co/collections/YanweiLi/mini-gemini-data-660463ea895a01d8f367624e

Mini-Gemini 還提供了 2B 小杯到 34B 的超大杯,最強模型在多個指標上相比 Google Gemini Pro 甚至 GPT-4V 都不遑多讓。目前,Mini-Gemini 從程式碼、模型、到資料已全部開源,還登上了 PaperWithCode 熱榜。

圖片

值得一提的是,Mini-Gemini 的影像理解和生成能力已經出了 Demo,可以線上跟自定義影像對話的那種。

操作也極其簡單,直接跟輸入影像或文字進行對話即可,歡迎來撩!

圖片

Mini-Gemini Demo 放出後受到廣大網友關注,一番 “品嚐” 後,他們認為 Mini-Gemini 跟商業模型差不了多少!

圖片

效果展示

目前,絕大多數多模態模型僅支援低解析度影像輸入和文字輸出,而在實際場景中,許多工都需要對高畫質影像進行解析,並用影像的形式進行展現。

圖片

如上圖所示,Mini-Gemini 不僅能夠根據圖片對做麵包的過程進行手把手教學,也能夠準確將不同電腦根據圖片中的各種引數進行對比。網友表示:媽媽再也不用擔心我的生活了。

更重要的是,Mini-Gemini 在保留超強的影像理解和推理能力的同時,還解鎖了影像的生成能力,就如同 ChatGPT 和生成模型的結合。

下面,讓我們透過幾個例子來更直觀地感受這種能力:

還記得 Google Gemini 的官方演示影片麼?當使用者給出兩個毛線團並問出能用它們做什麼時,Gemini 可以識別出圖片內容並給出相應的建議。

圖片

圖片

圖片

當我們把相似的輸入給到 Mini-Gemini,它會怎麼回答呢?

圖片

可以發現,Mini-Gemini 也可以識別出圖片中的元素,並且給出合理的建議,同時生成了一隻對應的毛線小熊。

圖片

透過一些抽象的多模態指令來讓模型給出推理,並生成合適的圖片,這個操作就很像是 ChatGPT 和 DALLE3 的聯動了!

接下來讓 Mini-Gemini 做自己最擅長的推理和圖片理解,看看它表現:

比如理解圖片中的矛盾點並舉一反三:輸入冰川中的仙人掌,它會解釋其中的矛盾並生成一張熱帶雨林中北極熊的圖片:

圖片

圖片呈現了仙人掌的典型棲息地與冰的存在之間的視覺矛盾,因為在沙漠環境中自然不會出現冰。這種矛盾的類似例子可能是一隻北極熊出現在熱帶雨林中,因為北極熊適應於寒冷、覆蓋著冰的環境,在炎熱潮溼的氣候中無法生存。這種並置創造了一個引人注目且超現實的視覺效果,挑戰觀眾的期待,並可能引發人們對氣候變化、環境適應或不同生態系統融合的思考。

同時,正如 ChatGPT+DALLE3 的夢幻結合一樣,Mini-Gemini 的推理生成功能還可以在多輪對話中透過簡單指令生成連環小故事。

圖片

圖片

比方說,讓它根據使用者輸入講一個貴族小老鼠的故事。Mini-Gemini 會根據前文的文字生成結果和使用者輸入進行推理,在保持一致性的情況下對圖片進行修改,使其更符合使用者的要求。

當然,Mini-Gemini 對於多模態模型的傳統技能圖片理解也不在話下。比如讓模型理解輸入曲線圖(上圖)的數學意義(高斯分佈),並讓它使用程式碼復現這張圖,透過執行生成的程式碼,模型可以高質量地還原曲線圖,節省了復現的時間。

圖片

圖片

又或者讓 Mini-Gemini 理解梗圖,透過其強大的 OCR 和推理能力,也可以準確指出笑點。

圖片

圖片

圖片

圖片

圖片

圖片

高畫質複雜的多圖表理解和歸納也是小菜一碟,Mini-Gemini 秒變打工人效率提升的超級外掛。

圖片

圖片

圖片

Mini-Gemini 背後的技術

大道至簡,Mini-Gemini的整體思路並不複雜。其中的Gemini(雙子座)表達的是使用視覺雙分支的資訊挖掘(Mining-Info in Gemini)解決高畫質影像理解問題。

詳細來說,Mini-Gemini 將傳統所使用的 ViT 當做低解析度的 Query,而使用卷積網路 (ConvNet) 將高解析度的影像編碼成 Key 和 Value。使用 Transformer 中常用的 Attention 機制,來挖掘每個低解析度 Query 所對應的高解析度區域。從而在保持最終視覺 Token 數目不變的情況下去提升對高畫質影像的響應,保證了在大語言模型 (LLM) 中對於高畫質影像的高效編碼。值得一提的是,由於高解析度分支卷積網路的使用,可以根據需要對影像所需的解析度自適應調整,能夠遇強則強。對於影像的生成部分,Mini-Gemini 藉助了 SDXL,使用 LLM 推理後所生成的文字連結兩個模型,類似於 DALLE3 的流程。

圖片

而對於資料這個萬金油,Mini-Gemini 進一步收集並最佳化了訓練資料的模型結合的文字資料進行訓練。在僅使用 2-3M 資料的情況下,實現了對影像理解、推理、和生成的統一流程。

實驗結果

Mini-Gemini 在各種 Zero-shot 的榜單上毫不遜色於各種大廠用大量資料訓練出來的模型,可謂是 “平、靚、正” !

量化指標

圖片

可以看出,Mini-Gemini 提供了多種普通和高畫質版本的模型,並且覆蓋了 2B 的小杯到 34B 的超大杯,各個版本都取得了相似引數量下領先的效果,在許多指標上甚至超越 Gemini Pro 和 GPT-4V。

相關文章