arXiv reaDer
UniNet:畳み込み、トランスフォーマー、およびMLPを使用した統合アーキテクチャ検索
UniNet: Unified Architecture Search with Convolution, Transformer, and MLP
最近、トランスフォーマーと多層パーセプトロン(MLP)アーキテクチャーは、さまざまなビジョンタスクで印象的な結果を達成しています。これらの演算子を手動で組み合わせてビジュアルネットワークアーキテクチャを設計することを検討したいくつかの作業は、ある程度満足のいくパフォーマンスを達成できます。この論文では、畳み込み、トランスフォーマー、およびMLPの最適な組み合わせを共同で検索して、視覚的なタスクで高性能な一連の全オペレーターネットワークアーキテクチャを構築することを提案します。広く使用されているストライド畳み込みまたはプーリングベースのダウンサンプリングモジュールは、オペレーターを組み合わせてネットワークを形成するときにパフォーマンスのボトルネックになることを経験的に特定しています。トランスフォーマーとMLPオペレーターによってキャプチャされたグローバルコンテキストによりよく取り組むために、トランスフォーマーとMLPオペレーターによってエンコードされたグローバル情報によりよく適応できる2つの新しいコンテキストアウェアダウンサンプリングモジュールを提案します。この目的のために、統合された検索スペースですべての演算子とダウンサンプリングモジュールを共同で検索します。特に、検索されたネットワークUniNet(Unified Network)は、複数のパブリックビジュアルベンチマーク、ImageNet分類、COCOオブジェクト検出、およびADE20Kセマンティックセグメンテーション。
Recently, transformer and multi-layer perceptron (MLP) architectures have achieved impressive results on various vision tasks. A few works investigated manually combining those operators to design visual network architectures, and can achieve satisfactory performances to some extent. In this paper, we propose to jointly search the optimal combination of convolution, transformer, and MLP for building a series of all-operator network architectures with high performances on visual tasks. We empirically identify that the widely-used strided convolution or pooling based down-sampling modules become the performance bottlenecks when the operators are combined to form a network. To better tackle the global context captured by the transformer and MLP operators, we propose two novel context-aware down-sampling modules, which can better adapt to the global information encoded by transformer and MLP operators. To this end, we jointly search all operators and down-sampling modules in a unified search space. Notably, Our searched network UniNet (Unified Network) outperforms state-of-the-art pure convolution-based architecture, EfficientNet, and pure transformer-based architecture, Swin-Transformer, on multiple public visual benchmarks, ImageNet classification, COCO object detection, and ADE20K semantic segmentation.
updated: Fri Oct 08 2021 11:09:40 GMT+0000 (UTC)
published: Fri Oct 08 2021 11:09:40 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アソシエイト