InternVL 系列調(diào)研

MLLM模型基本范式:

image.png

1.主要調(diào)研InternVL 2.0-40B

InternVL2-40B: 總計(jì) 40.07B; Vit: 5.54B; 143.17M瞬内;LLM: 34.39B
stage1訓(xùn)練數(shù)據(jù):我們?cè)贗nternVL 1.5中使用的預(yù)訓(xùn)練數(shù)據(jù)集擴(kuò)展了從多種來(lái)源收集的數(shù)據(jù)怜庸。這些數(shù)據(jù)集覆蓋了多個(gè)任務(wù)缆巧,包括標(biāo)題生成、視覺(jué)問(wèn)答、檢測(cè)蓉坎、定位和OCR。
stage2訓(xùn)練數(shù)據(jù): 包括了諸如EgoTaskQA胡嘿、Mementos蛉艾、STAR、NTU RGB+D衷敌、VideoChat2IT和LSMDC-QA這樣的視頻數(shù)據(jù)勿侯,以及Medical-Diff-VQA、Pathology-VQA缴罗、PMC-CaseReport助琐、PMC-VQA、Slake和VQA-RAD這樣的醫(yī)療數(shù)據(jù)面氓。我們還包括了SROIE兵钮、FUNSD和POIE,以進(jìn)一步增強(qiáng)模型識(shí)別手寫(xiě)字體的能力舌界。

1.1 模型評(píng)估和數(shù)據(jù)集:
1.2 Mini-InternVL:

技術(shù)報(bào)告:https://internvl.github.io/blog/2024-05-25-Mini-InternVL-1.5/
OpenGVLab:InternVL 1.5迷你款來(lái)了呻拌!8%的參數(shù)=80%的性能葱轩,一塊1080Ti 就能跑!

1.3 InternVL-1.5:

論文標(biāo)題:How Far Are We to GPT-4V? Closing the Gap to Commercial Multimodal Models with Open-Source Suites
論文鏈接:https://arxiv.org/pdf/2404.1682
具有卓越的bilingual能力柏锄,尤其Chinese相關(guān)任務(wù)中酿箭。模型基于"Vit-MLP-LLM"架構(gòu),結(jié)合了InternVit-6B視覺(jué)編碼器和InternLM2-20B語(yǔ)言模型趾娃,并通過(guò)動(dòng)態(tài)分辨率策略和數(shù)據(jù)翻譯管道來(lái)增強(qiáng)其對(duì)不同語(yǔ)言和圖像分辨率的支持缭嫡。
模型結(jié)構(gòu): ViT-MLP-LLM架構(gòu),通過(guò)MLP投影器將預(yù)訓(xùn)練的InternVit-6B和InternLM2-20B結(jié)合抬闷,此處還采用簡(jiǎn)單的pixel shuffle方法妇蛀,將視覺(jué)tokens的數(shù)量減少到四分之一。

2024-07-20 11-39-46屏幕截圖.png

1.4 InternVL:

InternVL模型結(jié)構(gòu)笤成,訓(xùn)練方法(3個(gè)漸進(jìn)階段) 注意加圖 :包括視覺(jué)-語(yǔ)言對(duì)比訓(xùn)練评架、視覺(jué)-語(yǔ)言生成訓(xùn)練和監(jiān)督式微調(diào)。這些階段有效利用不同來(lái)源的公開(kāi)數(shù)據(jù)炕泳。中間有些trainable, frozen和share weight纵诞。
訓(xùn)練過(guò)程:
1階段:采用LLama-7B編碼文本為T_f, InternViT-6B來(lái)提取視覺(jué)特征I_f, 遵循CLIP的目標(biāo)函數(shù),在圖像-文本對(duì)的相似度分?jǐn)?shù)上最小化一個(gè)對(duì)稱的交叉熵?fù)p失培遵。數(shù)據(jù)集清洗后的49.8億圖像-文本對(duì)浙芙。登刺。所有參數(shù)完全可訓(xùn)練
2階段:視覺(jué)-語(yǔ)言生成訓(xùn)練:將InternVit-6B和QLLama連接起來(lái),采用生成訓(xùn)練策略嗡呼。具體來(lái)說(shuō)纸俭,QLLaMa繼承第一階段LLaMA-7B權(quán)重,保持InterVit-6B和QLLaMA凍結(jié)南窗,只用過(guò)濾后的高質(zhì)量數(shù)據(jù)訓(xùn)練新增加的可學(xué)習(xí)查詢和交叉注意力層揍很。進(jìn)一步過(guò)濾低質(zhì)量標(biāo)題數(shù)據(jù),從一階段49.8億減少到10億万伤。保持InternVit-6B窒悔、QLLaMa凍結(jié),只訓(xùn)練參數(shù)
3階段:一種單獨(dú)使用InternVit-6B壕翩,另一種是使用整個(gè)InternVL模型蛉迹。使用高質(zhì)量Caption/VQA/多輪對(duì)話數(shù)據(jù)(4M)進(jìn)行sft訓(xùn)練。

1.5 InternViT模型:【 InternVit-6B(https://zhuanlan.zhihu.com/p/427388113)】

Vit:切分image到固定的圖片塊放妈,將這些圖片塊拉平成序列北救,加入positionembedding喂入標(biāo)準(zhǔn)Transformer encoder。在頭部加一個(gè)類(lèi)似的[CLS]標(biāo)簽芜抒,用于處理分類(lèi)任務(wù)珍策。主要特殊部分是patchEmbedding,其余部分和transformer類(lèi)似宅倒。

ViT模型結(jié)構(gòu)

PatchEmbedding: 具體處理input x的維度是(B攘宙,C,H拐迁,W)其中B是batch size蹭劈,C通常是三通道,H和W分別是圖片的高和寬线召,而輸出則是(B铺韧,N,E)缓淹,B依然是batch size哈打,N則是每張圖被切割成了patch之后,patch的數(shù)量讯壶,E是embed_size料仗,每個(gè)patch會(huì)通過(guò)一個(gè)全連接網(wǎng)絡(luò)轉(zhuǎn)換成一個(gè)向量,E是這個(gè)向量的長(zhǎng)度伏蚊,根據(jù)卷積的原理立轧,也可以理解為每個(gè)patch的特征數(shù)量。

class PatchEmbed(nn.Module):
    """ 2D Image to Patch Embedding
    """
    def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True):
        super().__init__()
        # img_size = (img_size, img_size)
        img_size = to_2tuple(img_size)
        patch_size = to_2tuple(patch_size)
        self.img_size = img_size
        self.patch_size = patch_size
        self.grid_size = (img_size[0] // patch_size[0], img_size[1] // patch_size[1])
        self.num_patches = self.grid_size[0] * self.grid_size[1]
        self.flatten = flatten
        # 輸入通道,輸出通道氛改,卷積核大小匀借,步長(zhǎng)
        # C*H*W->embed_dim*grid_size*grid_size
        self.proj = nn.Conv2d(in_chans, embed_dim, kernel_size=patch_size, stride=patch_size)
        self.norm = norm_layer(embed_dim) if norm_layer else nn.Identity()

    def forward(self, x):
        B, C, H, W = x.shape
        assert H == self.img_size[0] and W == self.img_size[1], \
            f"Input image size ({H}*{W}) doesn't match model ({self.img_size[0]}*{self.img_size[1]})."
        x = self.proj(x)
        if self.flatten:
            x = x.flatten(2).transpose(1, 2)  # BCHW -> BNC
        x = self.norm(x)
        return x
1.6 Pixel Shuffle和Dynamic High Resolution和CrossAttention

PixelShuffle:參考torch.nn.PixelShuffle(upscale_factor), 維度 (B,C * r * r, H, w) ,reshape成 (B, C, H * r, w * r)平窘。
DynamicHighResolution: 對(duì)輸入圖片是resize成448的倍數(shù),設(shè)置比例2:3凳怨,(H=448*2, W = 448*3), 然后按照預(yù)定義尺寸比例從圖片上crop相應(yīng)區(qū)域瑰艘。這里resize比例2:3是find_closet_aspect_ratio和aspect_ratio獲取最優(yōu)比例.同時(shí)還會(huì)將整個(gè)圖片resize成448*448作為 thumbnail, 接在序列最后。
Cross Attention: 還是傳統(tǒng)QKV, 但是Query, 來(lái)自一個(gè)模態(tài)的表示(如文本)肤舞, 但是Key和Value紫新,來(lái)自另外一個(gè)模態(tài)的表示.

1.7 CLIP(Constrasive Language-Image Pre-training):

算法原理: 文本和圖像在特征域進(jìn)行對(duì)齊, 分別對(duì)image和text進(jìn)行特征提取, 提取到特征以后, 做了normalize, 直接相乘來(lái)計(jì)算余弦距離, 同一pair對(duì)的結(jié)果趨于1, 不同pair趨于0, 采用constrasive loss.

1.8 InternVL訓(xùn)練完成后,使用方法:
  1. InternVL-使用方法

PS:

  1. 書(shū)生·萬(wàn)象多模態(tài)大模型(InternVL)系列 - 知乎 (zhihu.com)
  2. 多模態(tài)理解開(kāi)源王者:InternVL 1.5->InternVL 2.0 - 知乎 (zhihu.com)
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末李剖,一起剝皮案震驚了整個(gè)濱河市芒率,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌篙顺,老刑警劉巖偶芍,帶你破解...
    沈念sama閱讀 221,331評(píng)論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異德玫,居然都是意外死亡匪蟀,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,372評(píng)論 3 398
  • 文/潘曉璐 我一進(jìn)店門(mén)宰僧,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)材彪,“玉大人,你說(shuō)我怎么就攤上這事琴儿《位” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 167,755評(píng)論 0 360
  • 文/不壞的土叔 我叫張陵造成,是天一觀的道長(zhǎng)显熏。 經(jīng)常有香客問(wèn)我,道長(zhǎng)谜疤,這世上最難降的妖魔是什么佃延? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 59,528評(píng)論 1 296
  • 正文 為了忘掉前任,我火速辦了婚禮夷磕,結(jié)果婚禮上履肃,老公的妹妹穿的比我還像新娘。我一直安慰自己坐桩,他們只是感情好尺棋,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,526評(píng)論 6 397
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著,像睡著了一般膘螟。 火紅的嫁衣襯著肌膚如雪成福。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 52,166評(píng)論 1 308
  • 那天荆残,我揣著相機(jī)與錄音奴艾,去河邊找鬼。 笑死内斯,一個(gè)胖子當(dāng)著我的面吹牛蕴潦,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播俘闯,決...
    沈念sama閱讀 40,768評(píng)論 3 421
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼潭苞,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了真朗?” 一聲冷哼從身側(cè)響起此疹,我...
    開(kāi)封第一講書(shū)人閱讀 39,664評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎遮婶,沒(méi)想到半個(gè)月后蝗碎,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,205評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡旗扑,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,290評(píng)論 3 340
  • 正文 我和宋清朗相戀三年衍菱,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片肩豁。...
    茶點(diǎn)故事閱讀 40,435評(píng)論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡脊串,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出清钥,到底是詐尸還是另有隱情琼锋,我是刑警寧澤,帶...
    沈念sama閱讀 36,126評(píng)論 5 349
  • 正文 年R本政府宣布祟昭,位于F島的核電站缕坎,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏篡悟。R本人自食惡果不足惜谜叹,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,804評(píng)論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望搬葬。 院中可真熱鬧荷腊,春花似錦、人聲如沸急凰。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,276評(píng)論 0 23
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至疾忍,卻和暖如春乔外,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背一罩。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,393評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工杨幼, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人聂渊。 一個(gè)月前我還...
    沈念sama閱讀 48,818評(píng)論 3 376
  • 正文 我出身青樓推汽,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親歧沪。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,442評(píng)論 2 359

推薦閱讀更多精彩內(nèi)容