arXiv reaDer
難しい例を使用した自己教師あり事前トレーニングにより、視覚的表現が向上します
Self-supervised Pre-training with Hard Examples Improves Visual Representations
自己教師あり事前トレーニング(SSP)は、ランダムな画像変換を使用して、視覚表現学習用のトレーニングデータを生成します。この論文では、最初に、既存のSSPメソッドを疑似ラベルの予測学習として統合するモデリングフレームワークを紹介します。次に、ランダム画像変換によって生成されたものよりも疑似ラベルを予測するのが難しいトレーニング例を生成する新しいデータ拡張方法を提案します。具体的には、敵対的なトレーニングとCutMixを使用して、MoCo-v2とDeepCluster-v2の拡張ビューとして使用されるハードサンプル(HEXA)を作成し、それぞれ2つのバリアントHEXA_MoCoとHEXA_DClusterを作成します。私たちの実験では、ImageNetでモデルを事前トレーニングし、複数の公開ベンチマークで評価します。私たちの評価は、2つの新しいアルゴリズムのバリエーションが元のアルゴリズムのバリエーションを上回り、限られたタスクの監視が微調整に利用できる幅広いタスクで新しい最先端を達成することを示しています。これらの結果は、ハードな例が事前トレーニング済みモデルの一般化を改善するのに役立つことを証明しています。
Self-supervised pre-training (SSP) employs random image transformations to generate training data for visual representation learning. In this paper, we first present a modeling framework that unifies existing SSP methods as learning to predict pseudo-labels. Then, we propose new data augmentation methods of generating training examples whose pseudo-labels are harder to predict than those generated via random image transformations. Specifically, we use adversarial training and CutMix to create hard examples (HEXA) to be used as augmented views for MoCo-v2 and DeepCluster-v2, leading to two variants HEXA_MoCo and HEXA_DCluster, respectively. In our experiments, we pre-train models on ImageNet and evaluate them on multiple public benchmarks. Our evaluation shows that the two new algorithm variants outperform their original counterparts, and achieve new state-of-the-art on a wide range of tasks where limited task supervision is available for fine-tuning. These results verify that hard examples are instrumental in improving the generalization of the pre-trained models.
updated: Fri Dec 25 2020 02:44:22 GMT+0000 (UTC)
published: Fri Dec 25 2020 02:44:22 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アソシエイト