arXiv reaDer
ドロップキー
DropKey
このホワイト ペーパーでは、Vision Transformer の自己注意層のドロップアウト手法の分析と改善に焦点を当てます。これは、以前の研究では驚くほど無視されていましたが、重要です。特に、私たちは 3 つの中心的な質問について研究を行っています。文献でアテンション ウェイトをドロップするのとは異なり、ドロップアウト操作をアテンション マトリックス計算の前に進め、キーをドロップアウト ユニットとして設定し、新しいドロップアウト ビフォア ソフトマックス スキームを生成することを提案します。このスキームが注意の重みの正則化と確率の両方の機能を維持し、特定のパターンへの過剰適合の問題を軽減し、モデルを強化して重要な情報をグローバルにキャプチャするのに役立つことを理論的に検証します。次に、連続するレイヤーでドロップ率をスケジュールする方法は?すべてのレイヤーで一定のドロップ率を利用するのとは対照的に、自己注意レイヤーのスタックに沿ってドロップ率を徐々に減少させる新しい減少スケジュールを提示します。提案されたスケジュールが低レベルの機能でのオーバーフィッティングや高レベルのセマンティクスでの欠落を回避できることを実験的に検証し、モデルトレーニングの堅牢性と安定性を向上させます。第三に、構造化されたドロップアウト操作を CNN として実行する必要があるかどうか。ドロップアウト操作のパッチ ベースのブロック バージョンを試みたところ、この CNN の便利なトリックは ViT には必須ではないことがわかりました。上記の 3 つの質問を検討した結果、Key をドロップ ユニットと見なし、ドロップ率のスケジュールの減少を利用して、ViT を一般的な方法で改善する新しい DropKey メソッドを提示します。包括的な実験により、T2T や VOLO などのさまざまな ViT アーキテクチャや、画像分類、物体検出、人間と物体の相互作用の検出、人体形状の復元などのさまざまなビジョン タスクに対する DropKey の有効性が実証されています。コードは承認後にリリースされます。
In this paper, we focus on analyzing and improving the dropout technique for self-attention layers of Vision Transformer, which is important while surprisingly ignored by prior works. In particular, we conduct researches on three core questions: First, what to drop in self-attention layers? Different from dropping attention weights in literature, we propose to move dropout operations forward ahead of attention matrix calculation and set the Key as the dropout unit, yielding a novel dropout-before-softmax scheme. We theoretically verify that this scheme helps keep both regularization and probability features of attention weights, alleviating the overfittings problem to specific patterns and enhancing the model to globally capture vital information; Second, how to schedule the drop ratio in consecutive layers? In contrast to exploit a constant drop ratio for all layers, we present a new decreasing schedule that gradually decreases the drop ratio along the stack of self-attention layers. We experimentally validate the proposed schedule can avoid overfittings in low-level features and missing in high-level semantics, thus improving the robustness and stableness of model training; Third, whether need to perform structured dropout operation as CNN? We attempt patch-based block-version of dropout operation and find that this useful trick for CNN is not essential for ViT. Given exploration on the above three questions, we present the novel DropKey method that regards Key as the drop unit and exploits decreasing schedule for drop ratio, improving ViTs in a general way. Comprehensive experiments demonstrate the effectiveness of DropKey for various ViT architectures, e.g. T2T and VOLO, as well as for various vision tasks, e.g., image classification, object detection, human-object interaction detection and human body shape recovery. Codes will be released upon acceptance.
updated: Tue Mar 07 2023 04:23:53 GMT+0000 (UTC)
published: Thu Aug 04 2022 13:24:04 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アソシエイト