arXiv reaDer
負のデータ拡張を伴う対照学習による新規性の検出
Novelty Detection via Contrastive Learning with Negative Data Augmentation
新規性の検出は、クエリの例が学習したトレーニング分布と異なるかどうかを判断するプロセスです。以前の方法は、生成的敵対的ネットワーク(GAN)を介して通常のサンプルの表現を学習しようとします。しかし、彼らは不安定な訓練、モードの低下、そして低い識別能力に苦しむでしょう。最近、ノベルティ検出における自己教師あり学習のために、さまざまな口実タスク(回転予測やクラスタリングなど)が提案されています。ただし、学習された潜在的特徴はまだ識別力が低いです。新しいデコーダー-エンコーダーフレームワークを導入することにより、このような問題を克服します。まず、生成ネットワーク(別名デコーダー)は、初期化された潜在ベクトルを画像にマッピングすることによって表現を学習します。特に、このベクトルは、モードドロップの問題を回避するために、トレーニングデータの分布全体を考慮することによって初期化されます。第二に、対照的なネットワーク(別名エンコーダー)は、相互情報量の推定を通じて「比較することを学ぶ」ことを目的としています。これは、負のデータ拡張戦略を使用して、生成ネットワークがより識別力のある表現を取得するのに直接役立ちます。広範な実験により、私たちのモデルは最先端のノベルティ検出器よりも大幅に優れており、CIFAR10やDCASEなどのノベルティ検出ベンチマークで新しい最先端の結果を達成していることが示されています。さらに、私たちのモデルは、他の敵対者ベースの新規性検出方法と比較して、非敵対的方法でのトレーニングに対してより安定しています。
Novelty detection is the process of determining whether a query example differs from the learned training distribution. Previous methods attempt to learn the representation of the normal samples via generative adversarial networks (GANs). However, they will suffer from instability training, mode dropping, and low discriminative ability. Recently, various pretext tasks (e.g. rotation prediction and clustering) have been proposed for self-supervised learning in novelty detection. However, the learned latent features are still low discriminative. We overcome such problems by introducing a novel decoder-encoder framework. Firstly, a generative network (a.k.a. decoder) learns the representation by mapping the initialized latent vector to an image. In particular, this vector is initialized by considering the entire distribution of training data to avoid the problem of mode-dropping. Secondly, a contrastive network (a.k.a. encoder) aims to ``learn to compare'' through mutual information estimation, which directly helps the generative network to obtain a more discriminative representation by using a negative data augmentation strategy. Extensive experiments show that our model has significant superiority over cutting-edge novelty detectors and achieves new state-of-the-art results on some novelty detection benchmarks, e.g. CIFAR10 and DCASE. Moreover, our model is more stable for training in a non-adversarial manner, compared to other adversarial based novelty detection methods.
updated: Fri Jun 18 2021 07:26:15 GMT+0000 (UTC)
published: Fri Jun 18 2021 07:26:15 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アソシエイト