,,This work provides evidence that attention layers can perform convolution and, indeed, they often learn to do so in practice. Specifically, we prove that a multi-head self-attention
layer with sufficient number of heads is at least as expressive as any convolutional
layer. Our numerical experiments then show that self-attention layers attend to
pixel-grid patterns similarly to CNN layers, corroborating our analysis.''
Comments
Actually self attention is a generalization of convolution:
https://openreview.net/pdf?id=HJlnC1rKPB
,,This work provides evidence that attention layers can perform convolution and, indeed, they often learn to do so in practice. Specifically, we prove that a multi-head self-attention layer with sufficient number of heads is at least as expressive as any convolutional layer. Our numerical experiments then show that self-attention layers attend to pixel-grid patterns similarly to CNN layers, corroborating our analysis.''