12
$\begingroup$

I was hoping someone could explain to me why in the transformer model from the "Attention is all you need" paper there is no activation applied after both the multihead attention layer and to the residual connections. It seems to me that there are multiple linear layers in a row, and I have always been under the impression that you should have an activation between linear layers.

For instance when I look at the different flavors of resnet they always apply some sort of non linearity following a linear layer. For instance a residual block might look something like...

Input -> Conv -> BN -> Relu -> Conv -> (+ Input) -> BN -> Relu

or in the case of pre-activation...

Input -> BN -> Relu -> Conv -> BN -> Relu -> Conv -> (+ Input)

In all the resnet flavors I have seen, they never allow two linear layers to be connected without a relu in-between.

However in the the transformer...

Input -> Multihead-Attn -> Add/Norm -> Feed Forward(Dense Layer -> Relu -> Dense Layer) -> Add/Norm

In the multihead attention layer it performs the attention mechanism and then applies a fully connected layer to project back to the dimension of its input. However, there is no non linearity between that and feed forward network (except for maybe the softmax used in part of the attention.) A model like this would make more sense to me...

Input -> Multihead-Attn -> Add/Norm -> Relu -> Feed Forward(Dense Layer -> Relu -> Dense Layer) -> Add/Norm -> Relu

or something like the pre-activated resnet...

Input -> Relu -> Multihead-Attn -> Add/Norm -> Input2 -> Relu -> Feed Forward(Dense Layer -> Relu -> Dense Layer) -> Add/Norm(Input2)

Can anyone explain why the transformer is the way it is?

I have asked a similar question when I was looking at the architecture of wavenet on another forum but I never really got a clear answer. In that case it did not make sense to me again why there was no activation applied to the residual connections. (https://www.reddit.com/r/MachineLearning/comments/njbjfb/d_is_there_a_point_to_having_layers_with_just_a/)

$\endgroup$
1
  • $\begingroup$ FWIW, the final operation of each attention head is a weighted sum of values where the weights are computed as a softmax. Softmax is non-linear, and its shape is sometimes thought of as a multidimensional sigmoid. In some sense, the softmax-output weights serve as a sort of activation function. So maybe it's not bizarre to follow it up with the 1x1 convolutions of the position-wise feed forward layer (with layer normalization and residual connections in between). $\endgroup$ Sep 6, 2022 at 17:18

1 Answer 1

9
+50
$\begingroup$

This goes back to the purpose of self-attention.

Measure between word-vectors is generally computed through cosine-similarity because in the dimensions word tokens exist, it's highly unlikely for two words to be colinear even if they are trained to be closer in value if they are similar. However, two trained tokens will have higher cosine-similarity if they are semantically closer to each other than two completely unrelated words.

This fact is exploited by the self-attention mechanism; After several of these matrix multiplications, the dissimilar words will zero out or become negative due to the dot product between them, and the similar words will stand out in the resulting matrix.

So, as Tom points out in the comments below, self attention can be viewed as a weighted average, where less similar words become averaged out faster (toward the zero vector, on average), thereby achieving groupings of important and unimportant words (i.e. attention). The weighting happens through the dot product. If input vectors were normalized, the weights would be exactly the cosine similarities.

The important thing to take into consideration is that within the self-attention mechanism, there are no parameters; Those linear operations are just there to capture the relationship between the different vectors by using the properties of the vectors used to represent them.

Read this blog post by Peter Bloem for a more in-depth explanation of self-attention.


Edit

I should add that this explanation is less satisfactory considering how Transformers also seem to work for tasks without learned embeddings, like time-series forecasting. I have no idea why that is. However, the model was originally used for NLP, and they did use learned embeddings. So, I bet that's why that particular architecture looks the way it does.

Bloem, in the blog post above, does discuss the mathematical properties of self-attention without bringing up the fact that the original architecture does have learned embeddings.

All this shows is that having learned embeddings does not matter that much; The layers following the multi-headed attention will learn the relationships between the vectors. The general point about the properties of the dot-product being exploited does stand.

$\endgroup$
2
  • $\begingroup$ It seems like people are applying transformers to things without these embeddings, as you say. Without embeddings that have a priori meaningful relationships, all the attention stuff does not seem applicable. Since there is no activation, the same transformation can be done with a single layer if I'm not mistaken. $\endgroup$
    – Mastiff
    Feb 17 at 22:28
  • $\begingroup$ @Mastiff Well, if you want a more general answer, this is the best alternative I can come up with: There are a number of papers, starting with Metaformer, that suggest that any "token mixer" may work. As a specific example see FNet which uses Fourier transforms instead. Per these papers, attention, FFTs and even convolutions and poolings are examples of token mixers. So, my explanation still holds but for a different, possibly deeper, reason. I never added this since it's still pretty much in early research stage. Do you feel like I should? $\endgroup$
    – Avatrin
    May 21 at 16:49

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .