arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.16005cs.CVcs.LG

通过域同化实现医学影像的对抗鲁棒迁移学习

Adversarial-Robust Transfer Learning for Medical Imaging via Domain Assimilation

Missouri University of Science and Technology(密苏里科技大学)

Xiaohui Chen, Tie Luo

AI总结:

本文针对医学影像迁移学习中自然与医学图像域差异导致的对抗脆弱性问题,提出引入纹理与颜色适应及纹理保持的域同化方法,有效降低攻击效力并提升模型鲁棒性与可信度。

摘要:

在医学影像领域,已有大量研究致力于挖掘其在发现患者关键诊断特征方面的潜力。人工智能驱动的医学诊断依赖复杂的机器学习和深度学习模型,从医学图像中分析、检测和识别疾病。尽管这些模型以高准确率为特征的性能表现卓越,但它们仍面临可信度问题。对原始图像引入细微扰动,即可使攻击者操纵预测输出,将其重定向到其他有目标或无目标的类别。此外,公开可用的医学图像稀缺,构成了可靠训练的瓶颈,这使得当代算法依赖于基于大量自然图像预训练的模型——这种做法被称为迁移学习。然而,自然图像与医学图像之间存在显著的域差异,这导致迁移学习得到的AI模型对对抗攻击表现出更高的脆弱性。本文提出了一种域同化方法,在迁移学习中引入纹理和颜色适应,随后加入纹理保持组件以抑制不期望的失真。我们系统分析了在不同数据模态下迁移学习面对各种对抗攻击时的性能,总体目标是增强模型在医学影像任务中的鲁棒性和安全性。结果表明,该方法在降低攻击效力方面具有高效性,有助于在生物医学应用中实现更可信的迁移学习。

英文摘要:

In the field of Medical Imaging, extensive research has been dedicated to leveraging its potential in uncovering critical diagnostic features in patients. Artificial Intelligence (AI)-driven medical diagnosis relies on sophisticated machine learning and deep learning models to analyze, detect, and identify diseases from medical images. Despite the remarkable performance of these models, characterized by high accuracy, they grapple with trustworthiness issues. The introduction of a subtle perturbation to the original image empowers adversaries to manipulate the prediction output, redirecting it to other targeted or untargeted classes. Furthermore, the scarcity of publicly available medical images, constituting a bottleneck for reliable training, has led contemporary algorithms to depend on pretrained models grounded on a large set of natural images -- a practice referred to as transfer learning. However, a significant {\em domain discrepancy} exists between natural and medical images, which causes AI models resulting from transfer learning to exhibit heightened {\em vulnerability} to adversarial attacks. This paper proposes a {\em domain assimilation} approach that introduces texture and color adaptation into transfer learning, followed by a texture preservation component to suppress undesired distortion. We systematically analyze the performance of transfer learning in the face of various adversarial attacks under different data modalities, with the overarching goal of fortifying the model's robustness and security in medical imaging tasks. The results demonstrate high effectiveness in reducing attack efficacy, contributing toward more trustworthy transfer learning in biomedical applications.

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

↑