Attribute Regularized Soft Introspective Variational Autoencoder for Interpretable Cardiac Disease Classification
Interpretability is essential in medical imaging to ensure that clinicians can comprehend and trust artificial intelligence models. In this paper, we propose a novel interpretable approach that combines attribute regularization of the latent space within the framework of an adversarially trained variational autoencoder. Comparative experiments on a cardiac MRI dataset demonstrate the ability of the proposed method to address blurry reconstruction issues of variational autoencoder methods and improve latent space interpretability. Additionally, our analysis of a downstream task reveals that the classification of cardiac disease using the regularized latent space heavily relies on attribute regularized dimensions, demonstrating great interpretability by connecting the used attributes for prediction with clinical observations.
Code (1)
Tasks
AttributeSimilar Papers 제목 키워드 기반
Attribute Regularized Soft Introspective VAE: Towards Cardiac Attribute Regularization Through MRI Domains
Deep generative models have emerged as influential instruments for data generation and manipulation. Enhancing the controllability of these models by selectively modifying data attributes has been a recent focus. Variati…
AttributeImage GenerationInterpretable Representation Learning of Cardiac MRI via Attribute Regularization
Interpretability is essential in medical imaging to ensure that clinicians can comprehend and trust artificial intelligence models. Several approaches have been recently considered to encode attributes in the latent spac…
AttributeRepresentation LearningSoft-IntroVAE: Analyzing and Improving the Introspective Variational Autoencoder
The recently introduced introspective variational autoencoder (IntroVAE) exhibits outstanding image generations, and allows for amortized inference using an image encoder. The main idea in IntroVAE is to train a VAE adve…
Image GenerationOut-of-Distribution DetectionIntroVAC: Introspective Variational Classifiers for Learning Interpretable Latent Subspaces
Learning useful representations of complex data has been the subject of extensive research for many years. With the diffusion of Deep Neural Networks, Variational Autoencoders have gained lots of attention since they pro…
DecoderSemi-FairVAE: Semi-supervised Fair Representation Learning with Adversarial Variational Autoencoder
Adversarial learning is a widely used technique in fair representation learning to remove the biases on sensitive attributes from data representations. It usually requires to incorporate the sensitive attribute labels as…
AttributeFairnessRepresentation Learning