HyperAIHyperAI

Command Palette

Search for a command to run...

Swin Transformer:シフトされたウィンドウを用いた階層的ビジョンTransformer

Ze Liu Yutong Lin Yue Cao Han Hu Yixuan Wei Zheng Zhang Stephen Lin Baining Guo

概要

本稿では、コンピュータビジョンにおける汎用バックボーンとして機能可能な新しいビジョン変換器「Swin Transformer」を提案する。自然言語からビジョン領域へのTransformerの適用には、両領域間の本質的な差異が課題となる。具体的には、視覚的エンティティのスケール変動の大きさや、画像におけるピクセルの高解像度がテキストにおける単語と比較して顕著である点が挙げられる。これらの差異に対処するため、階層的なTransformerアーキテクチャを提案する。このアーキテクチャでは、\textbf{S}hifted \textbf{win}dows(シフト付きウィンドウ)を用いて表現を計算する。シフト付きウィンドウ方式は、自己注意(self-attention)の計算を重複しない局所的なウィンドウ内に限定することで、計算効率を向上させる一方で、ウィンドウ間の接続性も維持できる。この階層構造は、さまざまなスケールでのモデリングが可能であり、画像サイズに対して線形の計算複雑度を実現する。Swin Transformerのこれらの特徴により、画像分類(ImageNet-1Kでtop-1精度87.3%)、オブジェクト検出(COCO test-devでボックスAP 58.7、マスクAP 51.1)、セマンティックセグメンテーション(ADE20K valでmIoU 53.5)など、幅広いビジョンタスクに適応可能である。特にCOCOデータセットにおいて、従来の最先端技術を+2.7ボックスAPおよび+2.6マスクAPの大幅な差で上回り、ADE20Kでは+3.2 mIoUの向上を達成した。これは、Transformerベースのモデルがビジョンバックボーンとして極めて高い潜在能力を有していることを示している。また、階層構造とシフト付きウィンドウアプローチは、すべてMLP(全結合層)構造のアーキテクチャにも有益であることが実証された。コードおよびモデルは、~\url{https://github.com/microsoft/Swin-Transformer} にて公開されている。


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています