arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.08282eess.IVcs.CV

通过属性正则化实现心脏MRI的可解释表示学习

Interpretable Representation Learning of Cardiac MRI via Attribute Regularization

Institute of Machine Learning in Biomedical Imaging, Helmholtz Munich(亥姆霍兹慕尼黑生物医学成像机器学习研究所);Helmholtz AI, Helmholtz Munich(亥姆霍兹慕尼黑亥姆霍兹人工智能研究所);School of Computation, Information and Technology, Technical University of Munich(慕尼黑工业大学计算、信息与技术学院);School of Biomedical Engineering and Imaging Sciences, King’s College London(伦敦国王学院生物医学工程与影像科学学院)

Maxime Di Folco, Cosmin I. Bercea, Emily Chan, Julia A. Schnabel

AI总结:

提出属性正则化软内省变分自编码器,在对抗训练框架中结合属性正则化,解决了变分自编码器重建模糊问题,同时保持心脏MRI潜在空间的可解释性。

摘要:

可解释性在医学成像中至关重要,可确保临床医生能够理解并信任人工智能模型。近期已有几种方法考虑在潜在空间中编码属性以增强其可解释性。值得注意的是,属性正则化旨在沿潜在表示的维度编码一组属性。然而,该方法基于变分自编码器并存在重建模糊的问题。在本文中,我们提出了一种属性正则化软内省变分自编码器,它在对抗训练的变分自编码器框架内结合了潜在空间的属性正则化。我们在UK Biobank的短轴心脏磁共振图像上证明,所提方法能够解决变分自编码器方法的重建模糊问题,同时保留潜在空间的可解释性。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑