Customization of latent space in semi-supervised Variational AutoEncoder

Seunghwan An, Jong June Jeon

Research output: Contribution to journalArticlepeer-review

Abstract

We propose a novel semi-supervised learning method of Variational AutoEncoder (VAE), which yields a customized latent space through our EXplainable encoder Network (EXoN). The customization involves a manual design of the interpolation and structural constraint, such as proximity, which enhances the interpretability of the latent space. To improve the classification performance, we introduce a new semi-supervised classification method called SCI (Soft-label Consistency Interpolation). Combining the classification loss and the Kullback–Leibler divergence is crucial in constructing an explainable latent space. Additionally, the variability of the generated samples is determined by an active latent subspace, which effectively captures distinctive characteristics. We conduct experiments using the MNIST, SVHN, and CIFAR-10 datasets, and the results demonstrate that our approach yields an explainable latent space while significantly reducing the effort required to analyze representation patterns within the latent space.

Original languageEnglish
Pages (from-to)54-60
Number of pages7
JournalPattern Recognition Letters
Volume177
DOIs
StatePublished - Jan 2024

Keywords

  • Customization
  • Explainable latent space
  • Semi-supervised
  • Variational AutoEncoder

Fingerprint

Dive into the research topics of 'Customization of latent space in semi-supervised Variational AutoEncoder'. Together they form a unique fingerprint.

Cite this