arXiv:2412.02900eess.IVcs.LG2024-12被引 4

提出因果生成模型MACAW,让医学影像模型能理解变量间的因果关系。

MACAW: A Causal Generative Model for Medical Imaging

  • 将复杂因果结构融入归一化流,实现可解释的生成建模
  • 能生成与衰老相关的脑结构变化等反事实图像,准确预测年龄
  • 支持不确定性估计,适合需要可解释性的临床研究

尽管深度学习在神经影像研究中表现优异,但尚未广泛应用于临床,原因之一是多数模型仅捕捉输入与输出之间的相关性,导致包含无关偏差且难以解释。本文提出一种名为掩码因果流(MACAW)的新因果生成架构。首先,将复杂因果结构整合进归一化流;其次,通过反事实预测识别因果变量对结果变量的影响;最后,推导并实现了显式贝叶斯分类,提供内在不确定性估计。方法在合成数据和超过23000名英国生物银行参与者的真实脑部MRI数据上验证,结果显示:(1) 能准确编码因果推理,并生成与衰老相关的脑结构变化反事实图像;(2) 仅用单张2D MRI切片即可准确预测受试者年龄;(3) 可生成假设不同个体特征(如年龄、性别、体重指数)下的新样本。代码已公开于https://github.com/vibujithan/macaw-2D.git。

原文摘要 · Abstract (English)

Although deep learning techniques show promising results for many neuroimaging tasks in research settings, they have not yet found widespread use in clinical scenarios. One of the reasons for this problem is that many machine learning models only identify correlations between the input images and the outputs of interest, which can lead to many practical problems, such as encoding of uninformative biases and reduced explainability. Thus, recent research is exploring if integrating a priori causal knowledge into deep learning models is a potential avenue to identify these problems. This work introduces a new causal generative architecture named Masked Causal Flow (MACAW) for neuroimaging applications. Within this context, three main contributions are described. First, a novel approach that integrates complex causal structures into normalizing flows is proposed. Second, counterfactual prediction is performed to identify the changes in effect variables associated with a cause variable. Finally, an explicit Bayesian inference for classification is derived and implemented, providing an inherent uncertainty estimation. The feasibility of the proposed method was first evaluated using synthetic data and then using MRI brain data from more than 23000 participants of the UK biobank study. The evaluation results show that the proposed method can (1) accurately encode causal reasoning and generate counterfactuals highlighting the structural changes in the brain known to be associated with aging, (2) accurately predict a subject's age from a single 2D MRI slice, and (3) generate new samples assuming other values for subject-specific indicators such as age, sex, and body mass index. The code for a toy dataset is available at the following link: https://github.com/vibujithan/macaw-2D.git.

医学影像因果模型生成模型可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。