arXiv reaDer
ExCon:画像分類のための説明主導の教師あり対照学習
ExCon: Explanation-driven Supervised Contrastive Learning for Image Classification
対照的な学習により、画像分類などのタスクで学習された埋め込み表現の品質が大幅に向上しました。ただし、既存の対照的な拡張方法の主な欠点は、画像コンテンツの変更につながる可能性があり、そのセマンティクスに望ましくない変更が生じる可能性があることです。これは、ダウンストリームタスクでのモデルのパフォーマンスに影響を与える可能性があります。したがって、この論文では、画像のタスク関連の意味内容が保持されるように、対照学習で画像データを拡張できるかどうかを尋ねます。この目的のために、顕著性ベースの説明方法を活用して、対照的な学習のためにコンテンツを保持するマスクされた拡張を作成することを提案します。私たちの斬新な説明主導の教師あり対照学習(ExCon)方法論は、近くの画像埋め込みに同様のコンテンツと説明を持たせるという2つの目標を決定的に果たします。 ExConの影響を定量化するために、CIFAR-100およびTinyImageNetデータセットで実験を行います。 ExConは、分類、説明の質、敵対的なロバスト性、および分布シフトのコンテキストでのモデルの確率的予測のキャリブレーションの点で、バニラ教師あり対照学習よりも優れていることを示します。
Contrastive learning has led to substantial improvements in the quality of learned embedding representations for tasks such as image classification. However, a key drawback of existing contrastive augmentation methods is that they may lead to the modification of the image content which can yield undesired alterations of its semantics. This can affect the performance of the model on downstream tasks. Hence, in this paper, we ask whether we can augment image data in contrastive learning such that the task-relevant semantic content of an image is preserved. For this purpose, we propose to leverage saliency-based explanation methods to create content-preserving masked augmentations for contrastive learning. Our novel explanation-driven supervised contrastive learning (ExCon) methodology critically serves the dual goals of encouraging nearby image embeddings to have similar content and explanation. To quantify the impact of ExCon, we conduct experiments on the CIFAR-100 and the Tiny ImageNet datasets. We demonstrate that ExCon outperforms vanilla supervised contrastive learning in terms of classification, explanation quality, adversarial robustness as well as calibration of probabilistic predictions of the model in the context of distributional shift.
updated: Fri Dec 03 2021 02:15:27 GMT+0000 (UTC)
published: Sun Nov 28 2021 23:15:26 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アソシエイト