arXiv reaDer
画像の分類、生成などのためのジェネレーティブマックスマハラノビス分類器
Generative Max-Mahalanobis Classifiers for Image Classification, Generation and More
Grathwohlらの共同エネルギーベースモデル(JEM)。は、標準のソフトマックス分類器が、同時分布p(x、y)のエネルギーベースモデル(EBM)として再解釈できることを示しています。結果として得られるモデルを最適化して、キャリブレーション、堅牢性、および分布外検出を改善すると同時に、最近のGANベースのアプローチの品質に匹敵するサンプルを生成できます。ただし、JEMが利用するsoftmax分類器は本質的に識別力があり、その潜在的な特徴空間は確率分布として適切に定式化されていないため、画像生成の可能性が妨げられ、トレーニングが不安定になる可能性があります。生成的分類器はデータ生成プロセスを明示的にモデル化するため、線形判別分析(LDA)などの生成的分類器が画像生成に適している可能性があると仮定します。したがって、この論文では、画像の分類と生成のためのLDA分類器を調査します。特に、LDAの特殊なケースであるMax-Mahalanobis Classifier(MMC)は、私たちの目標に非常によく適合しています。私たちの生成MMC(GMMC)は、画像の分類と生成のために、識別的、生成的、または共同でトレーニングできることを示しています。複数のデータセットでの広範な実験により、GMMCは、キャリブレーション、敵対的ロバスト性、および分布外検出においてJEMを大幅に上回りながら、最先端の識別および生成パフォーマンスを達成することが示されています。ソースコードはhttps://github.com/sndnyang/GMMCで入手できます。
Joint Energy-based Model (JEM) of Grathwohl et al. shows that a standard softmax classifier can be reinterpreted as an energy-based model (EBM) for the joint distribution p(x,y); the resulting model can be optimized to improve calibration, robustness, and out-of-distribution detection, while generating samples rivaling the quality of recent GAN-based approaches. However, the softmax classifier that JEM exploits is inherently discriminative and its latent feature space is not well formulated as probabilistic distributions, which may hinder its potential for image generation and incur training instability. We hypothesize that generative classifiers, such as Linear Discriminant Analysis (LDA), might be more suitable for image generation since generative classifiers model the data generation process explicitly. This paper therefore investigates an LDA classifier for image classification and generation. In particular, the Max-Mahalanobis Classifier (MMC), a special case of LDA, fits our goal very well. We show that our Generative MMC (GMMC) can be trained discriminatively, generatively, or jointly for image classification and generation. Extensive experiments on multiple datasets show that GMMC achieves state-of-the-art discriminative and generative performances, while outperforming JEM in calibration, adversarial robustness, and out-of-distribution detection by a significant margin. Our source code is available at https://github.com/sndnyang/GMMC.
updated: Thu Feb 25 2021 13:35:35 GMT+0000 (UTC)
published: Fri Jan 01 2021 00:42: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アソシエイト