arXiv reaDer
敵対的な勢い-対照的な事前トレーニング
Adversarial Momentum-Contrastive Pre-Training
最近提案された敵対的な自己監視学習方法は、通常、堅牢な機能を抽出するために大きなバッチと長いトレーニングエポックを必要とします。これにより、リソースが限られているプラットフォームで大きな計算オーバーヘッドが発生します。ネットワークがより小さなバッチとより少ないエポックでより強力な特徴表現を学習するのを助けるために、この論文は、クリーンサンプルと敵対サンプルにそれぞれ対応する2つのメモリバンクを導入する新しい敵対運動量対照学習法を提案します。これらのメモリバンクをトレーニングプロセスに動的に組み込んで、過去のミニバッチ間の不変の機能を追跡できます。以前の敵対的な事前トレーニングモデルと比較して、私たちの方法は、より小さなバッチサイズとより少ないトレーニングエポックで優れたパフォーマンスを実現します。さらに、このモデルは、ダウンストリームの分類タスクで微調整された後、複数のベンチマークデータセットでいくつかの最先端の監視された防御方法よりも優れています。
Recently proposed adversarial self-supervised learning methods usually require big batches and long training epochs to extract robust features, which will bring heavy computational overhead on platforms with limited resources. In order to help the network learn more powerful feature representations in smaller batches and fewer epochs, this paper proposes a novel adversarial momentum contrastive learning method, which introduces two memory banks corresponding to clean samples and adversarial samples, respectively. These memory banks can be dynamically incorporated into the training process to track invariant features among historical mini-batches. Compared with the previous adversarial pre-training model, our method achieves superior performance with smaller batch size and less training epochs. In addition, the model outperforms some state-of-the-art supervised defensive methods on multiple benchmark datasets after being fine-tuned on downstream classification tasks.
updated: Sat May 28 2022 15:45:22 GMT+0000 (UTC)
published: Thu Dec 24 2020 07:49:10 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アソシエイト