本文提出了 SAN 框架,用于開放詞匯語義分割。該框架成功地利用了凍結的 CLIP 模型的特征以及端到端的流程,并最大化地采用凍結的 CLIP 模型。
簡介本文介紹了一種名為Side Adapter Network (SAN)的新框架,用于基于預訓練的視覺語言模型進行開放式語義分割。該方法將語義分割任務建模為區域識別問題,并通過附加一個側面的可學習網絡來實現。該網絡可以重用CLIP(Contrastive Language-Image Pre-Training)模型的特征,從而使其非常輕便。整個網絡可以進行端到端的訓練,使側面網絡適應凍結的CLIP模型,從而使預測的掩碼提案具有CLIP感知能力。作者在多個語義分割基準測試上評估了該方法,并表明其速度快、準確度高,只增加了少量可訓練參數,在一系列數據集上相較于之前的SOTA模型取得了大幅的性能提升(如下表所示)最后,作者希望該方法能夠成為一個baseline,并幫助未來的開放式語義分割研究。
論文鏈接:
https://arxiv.org/abs/2211.08073? ?
Introduction
作者首先討論了語義分割的概念和現代語義分割方法的限制,以及如何將大規模視覺語言模型應用于開放式語義分割。現代語義分割方法通常依賴于大量標記數據,但數據集通常只包含數十到數百個類別,昂貴的數據收集和注釋限制了我們進一步擴展類別的可能性。最近,大規模視覺語言模型(如CLIP)的出現促進了零樣本學習的發展,這也鼓勵我們探索其在語義分割中的應用。然而,將CLIP模型應用于開放式語義分割十分困難,因為CLIP模型是通過圖像級對比學習訓練的,其學習到的表示缺乏像素級別的識別能力,而這種能力在語義分割中是必需的。解決這個問題的一個方法是在分割數據集上微調模型,但是分割數據集的數據規模遠遠小于視覺語言預訓練數據集,因此微調模型在開放式識別方面的能力通常會受到影響。
為了充分發揮視覺-語言預訓練模型在開放詞匯語義分割中的能力。作者提出了一種名為Side Adapter Network(SAN)的新框架。由于端到端訓練,SAN的掩膜預測和分類是基于CLIP輔助的。整個模型十分輕量化。SAN有兩個分支:一個用于預測掩膜,另一個用于預測應用于CLIP的注意力偏好,以進行掩膜類別識別。作者表明,這種分離的設計可以提高分割性能。此外,作者還提出了一種單向前設計,以最小化CLIP的成本:將淺層CLIP塊的特征融合到SAN中,將其他更深層次的塊與注意偏置結合以進行掩膜識別。由于訓練是端到端的,SAN可以最大程度地適應凍結的CLIP模型。作者的研究基于官方發布的ViT CLIP模型,采用Visual Transformer實現。準確的語義分割需要高分辨率圖像,但發布的ViT CLIP模型設計用于低分辨率圖像(如),直接應用于高分辨率圖像會導致性能下降。為了緩解輸入分辨率的沖突,作者在CLIP模型中使用低分辨率圖像,在SAN中使用高分辨率圖像。作者表明,這種不對稱的輸入分辨率非常有效。此外,作者還探討了僅微調ViT模型的位置嵌入,并取得了改進。作者在各種基準測試中評估了他們的方法。與之前的方法相比,作者的方法在所有基準測試中都取得了最好的性能。作者的方法只有8.4M可訓練參數和64.3 GFLOPs。 ?Method
3.1 基礎架構
SAN的詳細架構如下圖所示。輸入圖像被分成個patch。首先通過一個線性層將圖片轉化為Visual Tokens。這些Visual Tokens會與個可學習的Query Tokens拼接起來,并送到后續的Transformer Layer中。每個Transformer Layer的Visual Tokens和Query Tokens都添加了position embedding。示例圖片SAN的輸出由兩部分構成:掩膜提議(Mask Proposals)和注意力偏好(Attention Biases)。在掩膜提議中,Query Tokens和Visual Tokens首先通過兩個單獨的3層MLP,投影成256維,我們將投影的Query Tokens表示為,其中是Query Tokens的數量,投影的Visual Tokens表示為,其中和是輸入圖像的高度和寬度。然后,通過和的內積生成掩膜: 其中。生成注意力偏好的過程類似于掩膜提議。Query Tokens和Visual Tokens也通過3層MLP進行投影,表示為和,其中是CLIP模型的注意頭數。通過對和進行內積,我們得到注意力偏好: 其中。此外,如果需要,注意力偏好還將進一步調整為,其中和是CLIP中注意力映射的高度和寬度。在實踐中,和可以共享,并且注意力偏好將應用于CLIP的多個自注意層,即偏好將在不同的自注意層中使用。這樣的雙輸出設計的動機很直觀:作者認為用于在CLIP中識別掩模的感興趣區域可能與掩模區域本身不同。作者在后文的對比實驗中也證實了這個想法。3.2掩膜預測
原始的CLIP模型只能通過標記進行圖像級別的識別。作者工作在不改變CLIP模型參數的情況下,嘗試通過指導標記的注意力圖在感興趣區域上實現精確的掩膜識別。為了實現這個目標,作者創建了一組名為標記(仿照Maskclip,如下圖)。這些標記單向地通過Visual Tokens進行更新,但是Visual Tokens和標記都不受的影響。在更新標記時,預測的注意力偏差被添加到注意力矩陣中: 其中表示層編號,表示第個注意力頭,和是的Query 和Key,是Visual Tokens 的Key。,和分別是Query、Key和Value的編碼權重。通過注意力偏好,標記的特征逐漸演變以適應掩膜預測,并且可以通過比較標記和類名CLIP文本編碼之間的距離/相似性來輕松獲得掩膜的類別預測,表示為,其中是類別數。3.3分割結果生成
使用上文提到的掩膜和類別預測,我們可以計算語義分割圖: 其中。這是標準的語義分割輸出,因此與主流的語義分割評估兼容。在訓練,我們通過Dice Loss 和binary cross-entropy loss 來監督掩膜生成,通過cross-entropy loss 來監督掩膜識別。總損失為: 其中作者使用的損失權重,,分別為5.0,5.0和2.0。通過端到端的訓練,SAN可以最大程度地適應凍結的CLIP模型,并得到很好的結果。 ?討論
具體來說,作者提出了一種全新的端到端架構,以極小的參數量在多個數據集上取得了SOTA效果。SAN的主要特點如下:
-
SAN中沿用了MaskCLIP得出的結論:在下游數據集上微調會破壞CLIP優秀的特征空間。因此在SAN的設計中,無需微調(fine-tune)CLIP模型,以便最大程度的保持CLIP模型的開放詞匯能力。
-
在凍結CLIP模型的同時,引入了額外的可編碼網絡,能夠根據下游任務數據集學習分割所需要的特征,彌補了CLIP模型對于位置信息的缺失。
- 將語義分割任務分解為掩膜預測與類別預測兩個子任務。CLIP模型的開放識別能力不僅僅依賴于物體區域本身,也依賴于物體的上下文信息(Context Information)。這促使作者提出掩膜預測與類別預測解耦的雙輸出設計,下表顯示該設計可以進一步提升模型的預測精度。
-
充分復用了CLIP模型的特征,大幅度降低所需的額外參數量的同時獲得最佳性能。下表展示了復用CLIP特征帶來的性能增益。
-
物聯網
+關注
關注
2913文章
44924瀏覽量
377022
原文標題:CVPR 2023 | 華科&MSRA新作:基于CLIP的輕量級開放詞匯語義分割架構
文章出處:【微信號:tyutcsplab,微信公眾號:智能感知與物聯網技術研究所】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論