arXiv reaDer
MSG-Transformer: メッセンジャー トークンの操作によるローカル空間情報の交換
MSG-Transformer: Exchanging Local Spatial Information by Manipulating Messenger Tokens
トランスフォーマーは、視覚認識のためのニューラル ネットワークを設計する新しい方法論を提供しました。畳み込みネットワークと比較して、トランスフォーマーは各段階でグローバルな特徴を参照する能力を楽しんでいますが、アテンション モジュールは、高解像度の視覚データを処理するトランスフォーマーのアプリケーションを妨げるより高い計算オーバーヘッドをもたらします。このペーパーは、効率性と柔軟性の間の矛盾を軽減することを目的としており、メッセンジャー (MSG) として機能する各リージョンに特化したトークンを提案します。したがって、これらの MSG トークンを操作することにより、リージョン間で視覚情報を柔軟に交換でき、計算の複雑さが軽減されます。次に、MSG トークンを MSG-Transformer という名前のマルチスケール アーキテクチャに統合します。標準的な画像分類とオブジェクト検出では、MSG-Transformer は競争力のあるパフォーマンスを実現し、GPU と CPU の両方で推論が加速されます。コードは https://github.com/hustvl/MSG-Transformer で入手できます。
Transformers have offered a new methodology of designing neural networks for visual recognition. Compared to convolutional networks, Transformers enjoy the ability of referring to global features at each stage, yet the attention module brings higher computational overhead that obstructs the application of Transformers to process high-resolution visual data. This paper aims to alleviate the conflict between efficiency and flexibility, for which we propose a specialized token for each region that serves as a messenger (MSG). Hence, by manipulating these MSG tokens, one can flexibly exchange visual information across regions and the computational complexity is reduced. We then integrate the MSG token into a multi-scale architecture named MSG-Transformer. In standard image classification and object detection, MSG-Transformer achieves competitive performance and the inference on both GPU and CPU is accelerated. The code will be available at https://github.com/hustvl/MSG-Transformer.
updated: Mon May 31 2021 17:16:42 GMT+0000 (UTC)
published: Mon May 31 2021 17:16:42 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アソシエイト