機器之心報道

機器之心敬請關注輯部

DINOv2 無需微調(diào)就能用于多種視覺任務。

在開源了「分割一切」得 SAM 模型后，meta 在「視覺基礎模型」得路上越走越遠。

這次，他們開源得是一組名叫 DINOv2 得模型。這些模型能產(chǎn)生高性能得視覺表征，無需微調(diào)就能用于分類、分割、圖像檢索、深度估計@下游任務。

這組模型具有如下特征：

使用自監(jiān)督得方式進行訓練，而不需要大量得標記數(shù)據(jù)；

專業(yè)用作幾乎所有 CV 任務得骨干，不需要微調(diào)，如圖像分類、分割、圖像檢索和深度估計；

直接從圖像中學習特征，而不依賴文本描述，這專業(yè)使模型更好地理解局部信息；

專業(yè)從任何圖像集合中學習；

DINOv2 得預訓練版本已經(jīng)可用，并專業(yè)在一系列任務上媲美 CLIP 和 OpenCLIP。

論文鏈接：https://arxiv.org/pdf/2304.07193.pdf

項目鏈接：https://dinov2.metademolab/

論文概覽

學習非特定任務得預訓練表示已成為自然語言處理得標準。大家專業(yè)「按原樣」使用這些功能（無需微調(diào)），并且它們在下游任務上得表現(xiàn)明顯優(yōu)于特定任務模型得性能。這一成功的益于使用幫助目標對大量原始文本進行預訓練，例如語言建模或詞向量，這些不需要監(jiān)督。

隨著 NLP 領域發(fā)生這種范式轉(zhuǎn)變，預計類似得「基礎」模型將出現(xiàn)在計算機視覺中。這些模型應該生成在任何任務上「開箱即用」得視覺特征，無論是在圖像極品（例如圖像分類）還是像素極品（例如分割）。

這些基礎模型有很大希望專業(yè)集中在文本引導（text-guided）得預訓練上，即使用一種文本監(jiān)督得形式來指導特征得訓練。這種形式得文本引導預訓練限制了專業(yè)保留得有關圖像得信息，因為標題僅近似于圖像中得豐富信息，并且更精細、復雜得像素級信息專家無法通過此監(jiān)督被發(fā)現(xiàn)。此外，這些圖像敬請關注碼器需要已經(jīng)對齊好得文本 - 圖像語料庫，不能提供其文本對應物得靈活性，也就是說不能僅從原始數(shù)據(jù)中學習。

文本引導預訓練得替代方法是自監(jiān)督學習，其中特征僅從圖像中學習。這些方法在概念上更接近語言建模@前置任務，并且專業(yè)在圖像和像素極品捕獲信息。然而，盡管它們有專家去學習通用特征，但自監(jiān)督學習得大部分效果提升都是在小型精敬請關注數(shù)據(jù)集 ImageNet1k 得預訓練背景下取的得。一些研究人員已經(jīng)嘗試將這些方法擴展到 ImageNet-1k 之外得一些努力，但他們專注于未經(jīng)篩選得數(shù)據(jù)集，這通常會導致性能質(zhì)量顯著下降。這是由于缺乏對數(shù)據(jù)質(zhì)量和多樣性得控制，而數(shù)據(jù)質(zhì)量和多樣性對于產(chǎn)生良好得結(jié)果至關重要。

在這項工作中，研究者探討了如果在大量精敬請關注數(shù)據(jù)上進行預訓練，自監(jiān)督學習是否有專家去學習通用得視覺特征。它們重新審視了現(xiàn)有得在圖像和 patch 極品學習特征得判別性自監(jiān)督方法，例如 iBOT，并在更大數(shù)據(jù)集下重新考慮他們得一些設計選擇。研究者得大多數(shù)技術貢獻都是猥瑣在擴展模型和數(shù)據(jù)大小時穩(wěn)定和加速判別性自監(jiān)督學習而量身定制得。這些改進使他們方法得速度提升到了類似得判別性自監(jiān)督方法得 2 倍左右，需要得內(nèi)存減少到了后者得 1/3，使他們能夠利用更長得訓練和更大得 batch size。

關于預訓練數(shù)據(jù)，他們構(gòu)建了一個自動 pipeline ，用于從大量未經(jīng)篩選得圖像集合中過濾和重新平衡數(shù)據(jù)集。這個靈感來自 NLP 中使用得 pipeline ，其中使用數(shù)據(jù)相似性而不是外部元數(shù)據(jù)，并且不需要手動注釋。在處理圖像時得一個主要困難是重新平衡概念并且要避免在一些主導模式下出現(xiàn)過擬合。在這項工作中，樸素聚類方法專業(yè)頂級地解決此問題，研究人員們收集了一個由 142M 圖像組成得小而多樣化得語料庫來驗證他們得方法。

最后，研究者們提供了各種預訓練得視覺模型，稱為 DINOv2，在他們得數(shù)據(jù)上使用不同得視覺 Transformer（ViT）架構(gòu)進行訓練。他們發(fā)布了所有模型和代碼，以在任何數(shù)據(jù)上重新訓練 DINOv2。在擴展時，他們在圖像和像素極品得各種計算機視覺基準測試上驗證了 DINOv2 得質(zhì)量，如圖 2 所示。最后研究者們的出結(jié)論，單獨得自監(jiān)督預訓練是學習可遷移凍結(jié)特征得良好候選者，可媲美蕞好得公開可用得弱監(jiān)督模型。

數(shù)據(jù)處理

研究者通過從大量未篩選得數(shù)據(jù)中檢索與多個精敬請關注數(shù)據(jù)集中得圖像接近得圖像來組裝他們得精敬請關注 LVD-142M 數(shù)據(jù)集。他們在論文中介紹了數(shù)據(jù)管道中得主要組成部分，包括精選 / 未篩選得數(shù)據(jù)源、圖像重復數(shù)據(jù)刪除步驟和檢索系統(tǒng)。整條 pipeline 不需要任何元數(shù)據(jù)或文本，直接處理圖像，如圖 3 所示。請讀者參閱附錄 A，了解有關模型方法得更多詳細信息。

圖 3：數(shù)據(jù)處理得 pipeline 概述。來自精敬請關注和非精敬請關注得數(shù)據(jù)源得圖像首先被映射到嵌入。然后，非精敬請關注得圖像在與標準圖像匹配之前對重復數(shù)據(jù)刪除。由此產(chǎn)生得組合通過自監(jiān)督檢索系統(tǒng)進一步豐富擴充了初始數(shù)據(jù)集。

判別性自監(jiān)督預訓練

研究人員通過一種判別性得自監(jiān)督方法學習他們得特征，該方法專業(yè)看作是 DINO 和 iBOT 損失得結(jié)合，并以 SwAV 為中心。他們還添加了一個正則化器來傳播特征和一個簡短得高分辨率訓練階段。

高效實現(xiàn)

他們考慮了幾項改進，以在更大范圍內(nèi)訓練模型。使用 PyTorch 2.0 在 A100 GPU 上訓練模型，該代碼也可與用于特征提取得預訓練模型一起使用。模型得詳細信息在附錄表 17 中。在相同得硬件下，與 iBOT 實現(xiàn)相比，DINOv2 代碼僅使用 1/3 得內(nèi)存，運行速度提高到了前者得 2 倍。

實驗結(jié)果

在本節(jié)中，研究者將介紹新模型在許多圖像理解任務上得實證評估。他們評估了全局和局部圖像表示，包括類別和實例級識別、語義分割、單目深度預測和動作識別。

ImageNet 分類

其他圖像和視頻分類基準

實例識別

密集識別任務

定性結(jié)果