作者丨Xi Chen等

來源丨機器之心

編輯丨張倩

語言和視覺任務的建模中，更大的神經(jīng)網(wǎng)絡模型能獲得更好的結果，幾乎已經(jīng)是共識。在語言方面，T5、GPT-3、Megatron-Turing、GLAM、Chinchilla 和 PaLM 等模型顯示出了在大文本數(shù)據(jù)上訓練大型 transformer 的明顯優(yōu)勢。視覺方面，CNN、視覺 transformer 和其他模型都從大模型中取得了很好的結果。language-and-vision 建模也是類似的情況，如 SimVLM、Florence、CoCa、GIT、BEiT 和 Flamingo。

在這篇論文中，來自谷歌的研究者通過一個名為 PaLI （Pathways Language and Image）的模型來延續(xù)這一方向的研究。

PaLI 使用單獨 “Image-and-text to text” 接口執(zhí)行很多圖像、語言以及 “圖像 + 語言” 任務。PaLI 的關鍵結構之一是重復使用大型單模態(tài) backbone 進行語言和視覺建模，以遷移現(xiàn)有能力并降低訓練成本。

在語言方面，作者復用有 13B 參數(shù)的 mT5-XXL。mT5-XXL 已經(jīng)把語言理解和泛化能力一體打包。作者通過實驗證明這些功能可以維護并擴展到多模態(tài)情況。

在視覺方面，除復用 2B 參數(shù) ViT-G 模型外，作者還訓練了擁有 4B 參數(shù)的模型 ViT-e（“enormous”）。ViT-e 在圖像任務上表現(xiàn)出很好的性能（ImageNet 上準確率達到 90.9％；ObjectNet 準確率達到 84.9%）。

作者發(fā)現(xiàn)了聯(lián)合 scaling 視覺和語言組件的好處，視覺提供了更好的投入回報（每個參數(shù) / FLOP 帶來的準確度提升）。實驗結果表明，最大的 PaLI 模型——PaLI-17B 在兩種任務模式下表現(xiàn)相對平衡，ViT-e 模型約占總參數(shù)的 25％。而先前的大規(guī)模視覺和語言建模工作，情況并非總是如此（Wang 等人，2022a；Alayrac 等人，2022），因為視覺和語言 backbone 之間的先驗量表并不匹配。

作者通過將多個圖像和 （或） 語言任務轉換為廣義的類似 VQA 的任務，實現(xiàn)它們之間的知識共享。使用 “image+query to answer” 來構建所有任務，其中檢索和回答都表示為文本標記。這使得 PaLI 能夠使用跨任務的遷移學習，并在廣泛的視覺和語言問題中增強 language-and-image 理解能力：圖像描述、視覺問答、場景文本理解等（如圖 1 所示）。

為了訓練 PaLI-17B，作者構建了全新的大容量 image-and-language 數(shù)據(jù)集 WebLI，包含 10B 的圖文對數(shù)據(jù)，WebLI 數(shù)據(jù)集包含 100 多種語言的文本。通過訓練模型用多種語言執(zhí)行多模態(tài)任務，這大大增加了任務的多樣性，并測試了模型在跨任務和跨語言之間有效擴展的能力。作者也提供了數(shù)據(jù)卡來介紹有關 WebLI 及其構造的信息。

PaLI-17B 在多個 benchmark 上都達到了 SOTA，表現(xiàn)優(yōu)于某些強大的模型（見表 1）。

具體來說，PaLI 在 COCO 數(shù)據(jù)集 benchmark 上的表現(xiàn)優(yōu)于多數(shù)新舊模型，在 Karpaty 分割上的得分為 149.1。PaLI 在 VQAv2 上使用類似 Flamingo 的開放詞匯文本生成的設置達到 84.3% 的最新 SOTA，該結果甚至優(yōu)于在固定詞匯分類環(huán)境中評估的模型，例如 CoCa、SimVLM、BEiT-3。作者的工作為未來的多模態(tài)模型提供了 scaling 路線圖。Model scaling 對于多語言環(huán)境中的語言圖像理解特別重要。作者的結果支持這樣一個結論：與其他替代方案相比，scaling 每個模式的組件會產(chǎn)生更好的性能。

這篇文章在知乎上引發(fā)了一些討論。有人感嘆說，「剛要匯報 beit3，隨便一刷知乎，又被超了」（引自知乎用戶 @走遍山水路）。還有人認為，論震撼程度，PaLI 比不上 BEiT-3，「畢竟 model scaling 這事大家已經(jīng)比較麻了」。但「谷歌把這個大家伙做出來了，還達到了一系列新 SOTA，并且零樣本都已經(jīng)做得很突出，還是非常令人敬佩」（引自知乎用戶 @霜清老人）。

以下是論文細節(jié)。

模型架構

作者使用 PaLI 的目的是執(zhí)行單模態(tài)（語言、視覺）和多模態(tài)（語言和視覺）任務。這些任務中的許多任務最好由不同的模型處理。如圖像分類及許多 VQA 需要從固定集合中預測元素，而 language-only 任務和圖像描述需要開放詞匯文本生成。作者通過使用所有任務所需的通用接口來解決該問題：模型接受圖像和文本字符串作為輸入，并生成文本作為輸出。在預訓練和微調(diào)時使用相同的接口。由于所有任務都使用相同的模型執(zhí)行，即沒有任務特定的參數(shù)，因此使用基于文本的提示指導模型需要執(zhí)行的任務。

圖 2 展示了模型架構的高階示意圖。其核心是一個文本 encoder-decoder transformer。為了將視覺作為輸入，向文本編碼器提供視覺“tokens”：視覺 transformer 將圖像作為輸入，并輸出相關特征。通過交叉注意力將視覺 token 傳遞到 encoder-decoder 模型之前，不會將池化應用于視覺 transformer 的輸出。

作者重復使用之前訓練過的單模態(tài)模型。對于文本 encoder-decoder，重復使用預訓練的 mT5（Xue 等，2021）模型，而對于圖像編碼，則重復使用大型 vanilla ViT 模型（Dosovitskiy 等，2021; Zhai 等，20222a）。

實驗結果

作者在三個純英文圖像的 benchmark 上評估了 PaLI 模型的變體，結果如表 4 所示。

作者對四個僅英文視覺問答（VQA）benchmark 進行評估，結果見表 6。

作者將 mT5-XXL 和 PaLI-17B 在一系列語言理解任務 benchmark 進行比較，對比結果如表 8 所示。

作者使用 224x224 分辨率（在高分辨率預微調(diào)之前）對 PaLI 模型在 Imagenet 和 Imagenet OOD 數(shù)據(jù)集上進行評估，評估結果如表 9 所示。

審核編輯：郭婷