English

SW-VAE: Weakly Supervised Learn Disentangled Representation Via Latent Factor Swapping

Machine Learning 2022-09-23 v1 Artificial Intelligence

Abstract

Representation disentanglement is an important goal of representation learning that benefits various downstream tasks. To achieve this goal, many unsupervised learning representation disentanglement approaches have been developed. However, the training process without utilizing any supervision signal have been proved to be inadequate for disentanglement representation learning. Therefore, we propose a novel weakly-supervised training approach, named as SW-VAE, which incorporates pairs of input observations as supervision signals by using the generative factors of datasets. Furthermore, we introduce strategies to gradually increase the learning difficulty during training to smooth the training process. As shown on several datasets, our model shows significant improvement over state-of-the-art (SOTA) methods on representation disentanglement tasks.

Keywords

Cite

@article{arxiv.2209.10623,
  title  = {SW-VAE: Weakly Supervised Learn Disentangled Representation Via Latent Factor Swapping},
  author = {Jiageng Zhu and Hanchen Xie and Wael Abd-Almageed},
  journal= {arXiv preprint arXiv:2209.10623},
  year   = {2022}
}
R2 v1 2026-06-28T01:51:05.910Z