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

在胸部X光分类中消除虚假相关性

Unlearning Spurious Correlations in Chest X-ray Classification

University College Dublin(都柏林大学)

Misgina Tsighe Hagos, Kathleen M. Curran, Brian Mac Namee

AI总结:

本研究针对医学图像分类中多数据源引发的虚假相关性问题,提出利用XBL方法结合低要求的特征标注反馈机制,有效消除Covid-19胸部X射线数据集中的虚假相关性,构建鲁棒模型。

摘要:

医学图像分类模型通常使用来自多个数据源的训练数据集进行训练。虽然利用多个数据源对于实现模型泛化至关重要,但必须认识到,这些来源的多样性固有地引入了意外的混杂因素及其他挑战,这可能会影响模型的准确性和透明度。在医学图像分类中,尤其是在肌肉骨骼图像分类中,一个显著的混杂因素是青春期观察到的骨骼成熟诱导的骨生长。我们使用Covid-19胸部X射线数据集训练了一个深度学习模型,并展示了由于意外的混杂区域,该数据集如何导致虚假相关性。eXplanation Based Learning (XBL) 是一种深度学习方法,它通过利用模型解释来交互式地消除虚假相关性,从而超越了可解释性。这是通过集成交互式用户反馈(特别是特征标注)来实现的。在我们的研究中,我们采用了两种低要求的手动反馈机制来实施基于XBL的方法,以有效消除这些虚假相关性。我们的结果强调了XBL在存在混杂因素的情况下构建鲁棒模型的巨大潜力。

英文摘要:

Medical image classification models are frequently trained using training datasets derived from multiple data sources. While leveraging multiple data sources is crucial for achieving model generalization, it is important to acknowledge that the diverse nature of these sources inherently introduces unintended confounders and other challenges that can impact both model accuracy and transparency. A notable confounding factor in medical image classification, particularly in musculoskeletal image classification, is skeletal maturation-induced bone growth observed during adolescence. We train a deep learning model using a Covid-19 chest X-ray dataset and we showcase how this dataset can lead to spurious correlations due to unintended confounding regions. eXplanation Based Learning (XBL) is a deep learning approach that goes beyond interpretability by utilizing model explanations to interactively unlearn spurious correlations. This is achieved by integrating interactive user feedback, specifically feature annotations. In our study, we employed two non-demanding manual feedback mechanisms to implement an XBL-based approach for effectively eliminating these spurious correlations. Our results underscore the promising potential of XBL in constructing robust models even in the presence of confounding factors.

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

↑