その驚異的な性能により、さまざまな視覚タスクでトランスフォーマーを利用することに関心が高まっています。ただし、既存のアプローチは主に内部モデル アーキテクチャ設計の最適化に重点を置いており、多くの場合、かなりの試行錯誤と大きな負担が伴います。この作業では、一般的なビジョン トランスフォーマーのパフォーマンスを向上させる、決定ストリーム キャリブレーションと呼ばれる新しいパラダイムを提案します。これを実現するために、さまざまなトークンと多次元の関連係数との相関関係を調べることにより、学習手順における情報伝播メカニズムに光を当てます。さらに分析すると、1) 最終決定はフォアグラウンド ターゲットのトークンに関連付けられ、フォアグラウンド ターゲットのトークン機能は可能な限り次のレイヤーに送信され、バックグラウンド領域の不要なトークン機能は削除されることがわかりました。順伝播で徐々に。 2) 各カテゴリは、トークン内の特定の疎次元のみに関連付けられています。上記の発見に基づいて、2 段階のキャリブレーション スキーム、つまり ViT-Calibrator を設計しました。これには、トークン伝搬キャリブレーション ステージと次元伝搬キャリブレーション ステージが含まれます。一般的に使用されるデータセットに関する広範な実験により、提案されたアプローチが有望な結果を達成できることが示されています。ソースコードは付録に記載されています。
A surge of interest has emerged in utilizing Transformers in diverse vision tasks owing to its formidable performance. However, existing approaches primarily focus on optimizing internal model architecture designs that often entail significant trial and error with high burdens. In this work, we propose a new paradigm dubbed Decision Stream Calibration that boosts the performance of general Vision Transformers. To achieve this, we shed light on the information propagation mechanism in the learning procedure by exploring the correlation between different tokens and the relevance coefficient of multiple dimensions. Upon further analysis, it was discovered that 1) the final decision is associated with tokens of foreground targets, while token features of foreground target will be transmitted into the next layer as much as possible, and the useless token features of background area will be eliminated gradually in the forward propagation. 2) Each category is solely associated with specific sparse dimensions in the tokens. Based on the discoveries mentioned above, we designed a two-stage calibration scheme, namely ViT-Calibrator, including token propagation calibration stage and dimension propagation calibration stage. Extensive experiments on commonly used datasets show that the proposed approach can achieve promising results. The source codes are given in the supplements.