Learning with a generative adversarial network from a positive unlabeled dataset for image classification

Abstract : In this paper, we propose a new approach which addresses the Positive Unlabeled learning challenge for image classification. Its functioning is based on GAN abilities in order to generate fake images samples whose distribution gets closer to negative samples distribution included in the unlabeled dataset available, while being different to the distribution of the unlabeled positive samples. Then we train a CNN classifier with the positive samples and the fake generated samples, as it would be done with a classic Positive Negative dataset. The tests performed on three different image classification datasets show that the system is stable up to an acceptable fraction of positive samples present in the unlabeled dataset. Although very different, this method outperforms the state of the art PU learning on the RGB dataset CIFAR-10.
Liste complète des métadonnées

Cited literature [17 references]  Display  Hide  Download

https://hal.archives-ouvertes.fr/hal-01811008
Contributor : Florent Chiaroni <>
Submitted on : Friday, June 8, 2018 - 2:15:34 PM
Last modification on : Thursday, November 15, 2018 - 1:24:01 PM
Document(s) archivé(s) le : Sunday, September 9, 2018 - 2:40:35 PM

File

Positive-GAN.pdf
Files produced by the author(s)

Identifiers

Citation

F Chiaroni, Mohamed-Cherif Rahal, Nicolas Hueber, Frédéric Dufaux. Learning with a generative adversarial network from a positive unlabeled dataset for image classification. IEEE International Conference on Image Processing, Oct 2018, Athens, Greece. ⟨10.1109/icip.2018.8451831 ⟩. ⟨hal-01811008⟩

Share

Metrics

Record views

306

Files downloads

823