比 CNN 更強有力,港中文賈佳亞團隊提出兩類新型自注意力網絡|CVPR2020

paper：https://hszhao.github.io/papers/cvpr20_san.pdf

code：https://github.com/hszhao/SAN

該文是香港中文大學賈佳亞老師團隊投於CVPR2020的一篇關於動態濾波器卷積的文章。對動態濾波器卷積實現不明白的小夥伴可能看該文會非常費力。本文首先在不改變原文意思的基礎上進行簡單的翻譯與介紹，在後續的文章中本人將從另一個角度對該文進行解析並進行復現和轉換，歡迎持續關注。

Abstract

近期的研究表明：自注意力可以作為圖像識別模型的基礎模塊而存在。作者探索了自注意力機制的變種並對其在圖像識別中的有效性進行了評價。作者考慮了兩種形式的自注意力機制：(1) pairwise self-attention，它是標準點積注意力的擴展，本質上是一組操作；(2) patchwise self-attention一種比卷積更強有力的的操作，其對應的模型明顯優於標準卷積版本。作者還通過實驗驗證了所提表達方式的魯棒性，並證實自注意力網絡從魯棒性和泛化性能方面均優於標準卷積網絡。

看完上面這一段介紹，大家有沒有感覺看的雲裡霧裡，不知所云。這到底是個什麼鬼？沒關係，下面的原文翻譯會讓你更加的不知所云，拭目以待！Method

在卷積網絡中，網絡中的OP有兩個作用：(1) 特徵聚合(feature aggregation)，對局部近鄰特徵採用核參數進行加權得到輸出特徵;(2) 特徵變換(feature transformation)，它有後接線性映射以及非線性函數達成(即BatchNorm+ReLU)。

作者認為：前述兩個功能可以解耦。如果可以採用一種特殊的機制進行特徵聚合，那麼採用感知器(它包含線性映射與非線性函數)進行逐點特徵變換。作者主要聚焦於特徵聚合部分進行研究。其實說白了就是嘗試提出一種特殊的機制替代標準卷積。

常規卷積通過固定核進行特徵聚合，即全圖權值共享，不會隨特徵內容而自適應變化，同時參數量會隨著聚合特徵數量增加而增加。鑑於此，作者提出幾種特徵聚合機制並構建了一個高性能圖像識別框架(它交錯執行特徵聚合(自注意力)和特徵變換(逐點感知器))。

作者探索了兩種類型的自注意力機制：(1) pairwise self-attention;(2) patchwise self-attention.

Pairwise Self-attention

首先，給出Pair Self-attention的定義：

其中，

其中，而

作者探索了下面幾種形式的相關函數

Position Encoding

pair self-attention的一個突出特徵：特徵向量

將水平和垂直坐標分別進行歸一化，歸一化到

Patchwise Self-attention

作者研究的另一種自注意力機制為patchwise self-attention，其公式定義如下：

其中

在patchwise self-attention中，作者允許構建的額權值向量直接作用於

類似的，作者對

類似地，作者研究了下面幾種形式的

Self-attention Block

上圖給出了本文所提出的自注意力機制在殘差模塊中嵌入示意圖。輸入特徵圖將經由兩個處理流：(1)左邊分支用於評估輸出注意力權值；(2)右邊分支將對輸入進行線性變換，為更有效的處理進行降維。前述兩個分支的輸出將通過Hadamard乘積進行特聚合，聚合後特徵將經由規範化操作+非線性激活與最後的非線性層得到與輸入同等通道的輸出。

看到這裡，不知道各位小夥伴是否看懂了作者想要表達的意思？反正我第一次看的時候是沒看懂，也不知道到底是怎麼做的，看的雲裡霧裡一塌糊塗。沒關係，我後面會提供一種更為通用的解釋來介紹上述兩種自注意力機制。Network Architectures

前面基本上已經把本文的核心全部介紹完畢了，那麼接下來就是要說明一下如何利用前述提到的兩種模塊搭建一個完成的神經網絡了。

ResNet不愧是深度學習的裡程碑，萬能油網絡啊，太多太多的網絡都是在ResNet的基礎上進行修改發paper，造就了太多的高質量paper(比如Res2Net, ResNeXt, MobileNetV2等)。

沒有任何意外，作者還是在ResNet的基礎上進行改造。下表為改造模型參數列表，作者提供了三種深度的網絡SAN10, SAN15, SAN19，它們分別對應ResNet26, ResNet38, ResNet50。

前面給出了作者改造的網絡結構：SAN系列。我們從下面幾個部分針對該系列模型進行簡單的分析。

Backbone: SAN的骨幹部分包含5個階段，不同階段具有不同的空間解析度，最終輸出32倍下採樣的特徵。每個階段包含不同數量的自注意力模塊，相鄰的兩個階段參考DenseNet通過過渡層(過渡層的作用是降低空間解析度並提升通道數)橋接。在五個階段之外，後接一個分來Head模塊，該模塊由全局均值池化、全連接層以及softmax構成。Transition：過渡層的用於降低特徵空間解析度，因此可以降低計算複雜度提升感受野。過渡層由BatchNorm、ReLU、MaxPool以及1x1卷積(用於擴增通道維度)構成。Footprint：局部範圍

上表對比了自注意力模塊與常規卷積、標量注意力機制之間的區別所在。

Convolution：常規卷積具有固定的核權值，不會隨內容而變化，但和權值會跨通道變化(但這種變化是訓練確定的，很難說是真正意義上的通道自適應)；Scalar Attention：標量注意力的聚合權值會根據內容隨位置發生變換，但存在跨通道共享。Vector attention：作者所設計的自注意力模塊則同時具有內容自適應與通道自適應性。Experiments

作者在ImageNet數據集上進行了所提方法的實驗與驗證。相關訓練信息如下：

Learning rate：cosine learning rate with base 0.1

Standard data augmentation: random crop, random flip, normalize

SyncSGD: momentum=0.9, weight decay=0.0001

Label Smoothing: regularization coefficient=0.1

BatchSize: 256.

關於自注意力模塊的幾個參數設置為：

上圖給出了所提SAN與對標ResNet的性能差異對比。從中可以看到：(1) pairwise模型媲美甚至優於對標的ResNet模型；(2)patchwise模型以極大的優勢超越了對標的ResNet。

上表對比了自注意力模塊中不同相關函數下的性能，從中可以看到：(1)對於pairwise模型而言，summation、subtraction以及Hadamard乘積具有相似的精度，並優於其他兩種相關函數;(2)對於patchwise模型而言，concatenation取得最佳性能。

上表對比了自注意力模塊中不同映射函數的性能對比。從中可以看到：(1)對於pairwise模型而言，兩個全連接層即可取得最佳性能；(2) 對於patchwise模型而言，不同的配置具有相似的性能(考慮到內存佔用與計算量問題，作者最終選用了兩個全連接層的配置)。

上表驗證了自注意力模塊中使用三個離散變換函數的必要性。從中可以看到：使用三個獨立的變換函數具有最佳性能,使用獨立變換函數的另外一個優勢是降低計算複雜度。

上表對比了自注意力模塊中不同局部範圍(即卷積核尺寸)下的性能對比。從中可以看到：(1)在常規卷積中，提升卷積核尺寸會提升計算複雜度與內存佔用；(2)在SAN中，性能先隨核尺寸增大而提升，而後達到飽和。對於pairwise模型而言，提升核尺寸對於參數沒有任何影響。綜合考量後，作者將核尺寸設置為

最後，作者評價了位置編碼在pairwise模型中的重要性，結果見上表。從中可以看到：位置編碼具有很重要的影響：位置編碼能帶來5%的性能提升。

此外，作者從兩個方面驗證了模型的魯棒性。(1) 旋轉魯棒性(見上表),pairwise模型具有更佳的抗旋轉性魯棒性，這也與前面提到的patchwise不具有置換不變性不謀而合；(2) 對抗魯棒性(見下表)，patchwise具有更佳的對抗攻擊魯棒性;(3)相比常規卷積網絡，自注意力網絡具有更佳的魯棒性和泛化性能。

Conclusion

作者研究了自注意力在圖像識別模型中的有效性，作者提出了兩種形式的自注意力：pairwise與patchwise。作者通過實驗得到以下幾點重要發現：

基於pairwise的網絡取得了匹配甚至超越純卷積網絡的性能，這意味著計算機視覺中深度學習的成功並非與卷積網絡緊密相連，還有其他的方案具有媲美甚至更優的判別能力(比如permutation- and cardinality-invariance)；基於patchwise的網絡取得了超越純卷積網絡的性能，這意味著塊自注意力可能在其他計算機視覺中取得更好的性能增益；相比標量(scale/dot-product)自注意力機制(當前主流注意力機制)，向量(vector)自注意力更有力且具有明顯優勢。