最近、トランスフォーマーは、さまざまなビジョン タスクで有望なパフォーマンスを示しています。ただし、トランスフォーマーにとって、特に高解像度の視覚タスクの場合、グローバルな自己注意の高いコストは依然として課題です。局所的自己注意は、効率のために限られた領域内で注意計算を実行しますが、その結果、受容野が小さいため、コンテキスト モデリングが不十分になります。この作業では、階層型ビジョン トランスフォーマーのグローバル モデリング機能を強化する 2 つの新しい注意モジュール、つまりランダム サンプリング ウィンドウ (RS-Win) と重要領域ウィンドウ (IR-Win) を紹介します。具体的には、RS-Win はランダムな画像パッチをサンプリングしてウィンドウを構成し、均一な分布に従います。つまり、RS-Win のパッチは画像の任意の位置から取得できます。 IR-Win はアテンション マップの画像パッチの重みに従ってウィンドウを構成します。特に、RS-Win は、初期の高解像度段階であっても、モデル全体にわたってグローバルな情報を取得できます。 IR-Win を使用すると、自己注意モジュールが画像の重要な領域に焦点を合わせ、より有益な機能をキャプチャできます。これらの設計に組み込まれた RSIR-Win Transformer は、一般的なビジョン タスクで優れたパフォーマンスを発揮します。
Recently, Transformers have shown promising performance in various vision tasks. However, the high costs of global self-attention remain challenging for Transformers, especially for high-resolution vision tasks. Local self-attention runs attention computation within a limited region for the sake of efficiency, resulting in insufficient context modeling as their receptive fields are small. In this work, we introduce two new attention modules to enhance the global modeling capability of the hierarchical vision transformer, namely, random sampling windows (RS-Win) and important region windows (IR-Win). Specifically, RS-Win sample random image patches to compose the window, following a uniform distribution, i.e., the patches in RS-Win can come from any position in the image. IR-Win composes the window according to the weights of the image patches in the attention map. Notably, RS-Win is able to capture global information throughout the entire model, even in earlier, high-resolution stages. IR-Win enables the self-attention module to focus on important regions of the image and capture more informative features. Incorporated with these designs, RSIR-Win Transformer demonstrates competitive performance on common vision tasks.