Augmenting Convolutional networks with attention-based aggregation

Hugo Touvron, Matthieu Cord, Alaaeldin El-Nouby +4

We show how to augment any convolutional network with an attention-based global map to achieve non-local reasoning. We replace the final average pooling by an attention-based aggre…