Interpretable Representation Learning of Cardiac MRI via Attribute Regularization

Autor:	Di Folco, Maxime, Bercea, Cosmin I., Chan, Emily, Schnabel, Julia A.
Rok vydání:	2024
Předmět:	Electrical Engineering and Systems Science - Image and Video Processing Computer Science - Computer Vision and Pattern Recognition
Druh dokumentu:	Working Paper
Popis:	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 space to enhance its interpretability. Notably, attribute regularization aims to encode a set of attributes along the dimensions of a latent representation. However, this approach is based on Variational AutoEncoder and suffers from blurry reconstruction. In this paper, we propose an Attributed-regularized Soft Introspective Variational Autoencoder that combines attribute regularization of the latent space within the framework of an adversarially trained variational autoencoder. We demonstrate on short-axis cardiac Magnetic Resonance images of the UK Biobank the ability of the proposed method to address blurry reconstruction issues of variational autoencoder methods while preserving the latent space interpretability. Comment: arXiv admin note: substantial text overlap with arXiv:2312.08915
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2406.08282 Zobrazit plný text záznamu View this record from Arxiv