Poster
in
Workshop: 3rd Workshop on Interpretable Machine Learning in Healthcare (IMLH)
SepVAE: a contrastive VAE to separate pathological patterns from healthy ones.
Robin Louiset · Edouard Duchesnay · Antoine Grigis · Benoit Dufumier · Pietro Gori
Keywords: [ Contrastive Analysis ] [ Variational Auto-Encoders ] [ Medical Imaging ]
Contrastive Analysis VAEs (CA-VAEs) are a family of Variational auto-encoders (VAEs) that aims at separating the common factors of variation between a background dataset (BG) (i.e., healthy subjects) and a target dataset (TG) (i.e., patients) from the ones that only exist in the target dataset. To do so, these methods separate the latent space into a set of salient features (i.e., proper to the target dataset) and a set of common features (i.e., exist in both datasets). Currently, all models fail to effectively prevent the sharing of information between latent spaces and to capture all salient factors of variation.To this end, we introduce two crucial regularization losses: a disentangling term between common and salient representations and a classification term between background and target samples in the salient space. We show a better performance than previous CA-VAEs methods on three medical applications and a natural images dataset (CelebA). Code and datasets are available at https://anonymous.4open.science/r/sep_vae-0D94/.