An Image Patch is a Wave: Phase-Aware Vision MLP
arXiv:2111.12294
Abstract
In the field of computer vision, recent works show that a pure MLP architecture mainly stacked by fully-connected layers can achieve competing performance with CNN and transformer. An input image of vision MLP is usually split into multiple tokens (patches), while the existing MLP models directly aggregate them with fixed weights, neglecting the varying semantic information of tokens from different images. To dynamically aggregate tokens, we propose to represent each token as a wave function with two parts, amplitude and phase. Amplitude is the original feature and the phase term is a complex value changing according to the semantic contents of input images. Introducing the phase term can dynamically modulate the relationship between tokens and fixed weights in MLP. Based on the wave-like token representation, we establish a novel Wave-MLP architecture for vision tasks. Extensive experiments demonstrate that the proposed Wave-MLP is superior to the state-of-the-art MLP architectures on various vision tasks such as image classification, object detection and semantic segmentation. The source code is available at https://github.com/huawei-noah/CV-Backbones/tree/master/wavemlp_pytorch and https://gitee.com/mindspore/models/tree/master/research/cv/wave_mlp.
arXiv commentThis paper is accepted by CVPR 2022 (oral presentation)
References in corpus (16)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Decoupled Weight Decay Regularization
- mixup: Beyond Empirical Risk Minimization
- MLP-Mixer: An all-MLP Architecture for Vision
- Transformer in Transformer
- Swin Transformer: Hierarchical Vision Transformer using Shifted Windows
- Visual Transformers: Token-based Image Representation and Processing for Computer Vision
- CycleMLP: A MLP-like Architecture for Dense Prediction
- AS-MLP: An Axial Shifted MLP Architecture for Vision
- Beyond Self-attention: External Attention using Two Linear Layers for Visual Tasks
- Rethinking Spatial Dimensions of Vision Transformers
- Pay Attention to MLPs
- S-MLP: Spatial-Shift MLP Architecture for Vision
- Vision Permutator: A Permutable MLP-Like Architecture for Visual Recognition
- Global Filter Networks for Image Classification