Interpretable 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 space to enhance its interpretability. Notably, attribute regularization aims to enc...

Full description

Saved in:
Bibliographic Details
Published in:arXiv.org 2024-08
Main Authors: Maxime Di Folco, Bercea, Cosmin I, Chan, Emily, Schnabel, Julia A
Format: Article
Language:English
Subjects:
Artificial intelligence
Image reconstruction
Machine learning
Magnetic resonance imaging
Medical imaging
Regularization
Representations
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary: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.
ISSN:2331-8422