arXiv reaDer
ConvMLP:ビジョンのための階層型畳み込みMLP
ConvMLP: Hierarchical Convolutional MLPs for Vision
一連の連続する多層パーセプトロンブロックで構成されるMLPベースのアーキテクチャは、最近、畳み込みおよびトランスベースの方法と同等の結果に達することがわかっています。ただし、ほとんどの場合、固定次元の入力を受け取る空間MLPを採用しているため、オブジェクト検出やセマンティックセグメンテーションなどのダウンストリームタスクに適用することは困難です。さらに、シングルステージ設計は他のコンピュータビジョンタスクのパフォーマンスをさらに制限し、完全に接続されたレイヤーは大量の計算を負担します。これらの問題に取り組むために、ConvMLPを提案します。これは視覚認識用の階層型畳み込みMLPであり、畳み込み層とMLPの軽量で段階的な共同設計です。特に、ConvMLP-Sは、9Mパラメーターと2.4G MAC(MLP-Mixer-B / 16のそれぞれ15%と19%)を使用してImageNet-1kで76.8%のトップ1精度を達成します。オブジェクト検出とセマンティックセグメンテーションに関する実験は、ConvMLPによって学習された視覚的表現がシームレスに転送され、より少ないパラメーターで競争力のある結果を達成できることをさらに示しています。コードと事前トレーニング済みモデルは、https://github.com/SHI-Labs/Convolutional-MLPsで公開されています。
MLP-based architectures, which consist of a sequence of consecutive multi-layer perceptron blocks, have recently been found to reach comparable results to convolutional and transformer-based methods. However, most adopt spatial MLPs which take fixed dimension inputs, therefore making it difficult to apply them to downstream tasks, such as object detection and semantic segmentation. Moreover, single-stage designs further limit performance in other computer vision tasks and fully connected layers bear heavy computation. To tackle these problems, we propose ConvMLP: a hierarchical Convolutional MLP for visual recognition, which is a light-weight, stage-wise, co-design of convolution layers, and MLPs. In particular, ConvMLP-S achieves 76.8% top-1 accuracy on ImageNet-1k with 9M parameters and 2.4G MACs (15% and 19% of MLP-Mixer-B/16, respectively). Experiments on object detection and semantic segmentation further show that visual representation learned by ConvMLP can be seamlessly transferred and achieve competitive results with fewer parameters. Our code and pre-trained models are publicly available at https://github.com/SHI-Labs/Convolutional-MLPs.
updated: Thu Sep 09 2021 17:52:57 GMT+0000 (UTC)
published: Thu Sep 09 2021 17:52:57 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト