涨潮是否能抬高所有船只?基于AI的心脏磁共振分割中的偏置缓解研究

Does a Rising Tide Lift All Boats? Bias Mitigation for AI-based CMR Segmentation

摘要 Abstract

人工智能(AI)在医学影像任务中的应用日益广泛,然而,这些模型可能存在偏置,特别是在训练数据集不平衡的情况下。例如,在心脏磁共振(CMR)图像分割模型中存在明显的种族偏置效应。尽管这一现象已在多篇文献中报道,但对该领域偏置缓解算法的有效性知之甚少。我们旨在研究常见的偏置缓解方法对基于AI的CMR分割模型中黑人与白人受试者之间偏置的影响。具体而言,我们采用过采样、重要性重加权以及分组分布鲁棒优化(Group DRO),并结合这些技术来减轻种族偏置。此外,受到关于AI驱动的CMR分割偏置根本原因的最新研究启发,我们还评估了在裁剪后的CMR图像上训练和评估的模型中相同的方法。结果表明,过采样能够有效减轻偏置,显著提高代表性不足的黑人群体的表现,同时不会显著降低多数白人群体的表现。分组分布鲁棒优化也能提升黑人群体的表现,但效果不显著,而重要性重加权则降低了黑人群体的表现。结合过采样和分组分布鲁棒优化同样能改善黑人群体的表现,但效果不显著。使用裁剪后的图像可以提高两个种族群体的表现,并减少偏置,而结合过采样技术进一步减少了偏置。

Artificial intelligence (AI) is increasingly being used for medical imaging tasks. However, there can be biases in the resulting models, particularly when they were trained using imbalanced training datasets. One such example has been the strong race bias effect in cardiac magnetic resonance (CMR) image segmentation models. Although this phenomenon has been reported in a number of publications, little is known about the effectiveness of bias mitigation algorithms in this domain. We aim to investigate the impact of common bias mitigation methods to address bias between Black and White subjects in AI-based CMR segmentation models. Specifically, we use oversampling, importance reweighing and Group DRO as well as combinations of these techniques to mitigate the race bias. Furthermore, motivated by recent findings on the root causes of AI-based CMR segmentation bias, we evaluate the same methods using models trained and evaluated on cropped CMR images. We find that bias can be mitigated using oversampling, significantly improving performance for the underrepresented Black subjects whilst not significantly reducing the majority White subjects' performance. Group DRO also improves performance for Black subjects but not significantly, while reweighing decreases performance for Black subjects. Using a combination of oversampling and Group DRO also improves performance for Black subjects but not significantly. Using cropped images increases performance for both races and reduces the bias, whilst adding oversampling as a bias mitigation technique with cropped images reduces the bias further.

涨潮是否能抬高所有船只?基于AI的心脏磁共振分割中的偏置缓解研究 - arXiv