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

使用生成式人工智能研究医学影像模型与数据集

Using generative AI to investigate medical imagery models and datasets

Google(谷歌);Advanced Clinical(先进临床);Pro Unlimited;University of California San Francisco(加州大学旧金山分校);Weizmann Institute of Science(魏茨曼科学研究所)
Oran Lang, Doron Yaya-Stupp, Ilana Traynis, Heather Cole-Lewis, Chloe R. Bennett, Courtney Lyles, Charles Lau, Michal Irani, Christopher Semturs, Dale R. Webste… 展开作者

Oran Lang, Doron Yaya-Stupp, Ilana Traynis, Heather Cole-Lewis, Chloe R. Bennett, Courtney Lyles, Charles Lau, Michal Irani, Christopher Semturs, Dale R. Webster, Greg S. Corrado, Avinatan Hassidim, Yossi Matias, Yun Liu, Naama Hammel, Boris Babenko

AI总结:

本文提出一种基于StylEx生成模型的自动视觉解释方法,通过检测分类器敏感属性并结合跨学科专家形成假设,在三种医学影像模态上揭示已知特征、混杂因素和新属性,以增强AI模型可解释性并促进科学发现。

摘要:

AI模型已在许多医学影像任务中展现出潜力。然而,我们解释这些模型学到了哪些信号的能力严重不足。为了增强对基于AI模型的信任,解释是必要的,并且通过揭示数据中专家尚不知晓的信号,解释还可能促成新的科学发现。在本文中,我们提出了一种利用团队专业知识进行自动视觉解释的方法,通过生成关于图像中哪些视觉信号与任务相关的假设。我们提出以下4个步骤:(i)训练一个分类器以执行给定任务;(ii)训练一个分类器引导的基于StyleGAN的图像生成器(StylEx);(iii)自动检测并可视化分类器最敏感的顶级视觉属性;(iv)针对潜在机制提出假设,以激发未来研究。具体而言,我们将发现的属性呈现给一个跨学科专家小组,以便假设能够考虑健康的社会和结构性决定因素。我们在三种医学影像模态的八个预测任务上展示了结果:视网膜眼底照片、外眼照片和胸部X光片。我们展示了捕捉临床已知特征的属性示例、由生理机制之外的因素引起的混杂因素,并揭示了许多生理上合理的新属性。我们的方法有潜力使研究人员能够更好地理解、改进其评估,并从基于AI的模型中提取新知识。重要的是,我们强调,我们的框架生成的属性可以捕捉超越生理学或病理生理学的现象,反映医疗保健服务和社会文化因素的真实世界本质。最后,我们打算发布代码,使研究人员能够训练自己的StylEx模型并分析其预测任务。

英文摘要:

AI models have shown promise in many medical imaging tasks. However, our ability to explain what signals these models have learned is severely lacking. Explanations are needed in order to increase the trust in AI-based models, and could enable novel scientific discovery by uncovering signals in the data that are not yet known to experts. In this paper, we present a method for automatic visual explanations leveraging team-based expertise by generating hypotheses of what visual signals in the images are correlated with the task. We propose the following 4 steps: (i) Train a classifier to perform a given task (ii) Train a classifier guided StyleGAN-based image generator (StylEx) (iii) Automatically detect and visualize the top visual attributes that the classifier is sensitive towards (iv) Formulate hypotheses for the underlying mechanisms, to stimulate future research. Specifically, we present the discovered attributes to an interdisciplinary panel of experts so that hypotheses can account for social and structural determinants of health. We demonstrate results on eight prediction tasks across three medical imaging modalities: retinal fundus photographs, external eye photographs, and chest radiographs. We showcase examples of attributes that capture clinically known features, confounders that arise from factors beyond physiological mechanisms, and reveal a number of physiologically plausible novel attributes. Our approach has the potential to enable researchers to better understand, improve their assessment, and extract new knowledge from AI-based models. Importantly, we highlight that attributes generated by our framework can capture phenomena beyond physiology or pathophysiology, reflecting the real world nature of healthcare delivery and socio-cultural factors. Finally, we intend to release code to enable researchers to train their own StylEx models and analyze their predictive tasks.

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

↑