论文标题

毫不奇怪:通过对抗攻击来增强低剂量CT扫描的良好肺结节检测

No Surprises: Training Robust Lung Nodule Detection for Low-Dose CT Scans by Augmenting with Adversarial Attacks

论文作者

Liu, Siqi, Setio, Arnaud Arindra Adiyoso, Ghesu, Florin C., Gibson, Eli, Grbic, Sasa, Georgescu, Bogdan, Comaniciu, Dorin

论文摘要

在早期检测恶性肺结核可以允许医疗干预措施,从而增加肺癌患者的存活率。使用计算机视觉技术检测结节可以提高解释胸部CT进行肺癌筛查的灵敏度和速度。许多研究使用CNN检测候选结节。尽管已经证明这种方法表现出胜过基于图像处理的常规方法的检测准确性,但已知CNN也仅限于训练集中代表性不足的样本,而易于易于察觉到不可察觉的噪声扰动。通过扩展数据集或模型,无法轻易解决此类限制。在这项工作中,我们建议将对抗性合成结节和对抗性攻击样本添加到训练数据中,以改善肺结核检测系统的概括和鲁棒性。为了生成来自可区分结节合成器的结节的硬示例,我们使用预计的梯度下降(PGD)来搜索有界社区内的潜在代码,以生成结节以减少检测器响应。为了使网络对意外的噪声扰动更加健壮,我们使用PGD来搜索可以触发网络以产生过度信心错误的噪声模式。通过评估包含来自三个放射科医生共有注释的两个不同基准数据集,我们表明所提出的技术可以改善实际CT数据上的检测性能。为了了解常规网络和提议的增强网络的局限性,我们还通过喂食不同类型的人工产生的斑块来对假阳性还原网络进行应力测试。我们表明,增强网络对于代表性不足的结节以及对噪声扰动的抗性都更为强大。

Detecting malignant pulmonary nodules at an early stage can allow medical interventions which may increase the survival rate of lung cancer patients. Using computer vision techniques to detect nodules can improve the sensitivity and the speed of interpreting chest CT for lung cancer screening. Many studies have used CNNs to detect nodule candidates. Though such approaches have been shown to outperform the conventional image processing based methods regarding the detection accuracy, CNNs are also known to be limited to generalize on under-represented samples in the training set and prone to imperceptible noise perturbations. Such limitations can not be easily addressed by scaling up the dataset or the models. In this work, we propose to add adversarial synthetic nodules and adversarial attack samples to the training data to improve the generalization and the robustness of the lung nodule detection systems. To generate hard examples of nodules from a differentiable nodule synthesizer, we use projected gradient descent (PGD) to search the latent code within a bounded neighbourhood that would generate nodules to decrease the detector response. To make the network more robust to unanticipated noise perturbations, we use PGD to search for noise patterns that can trigger the network to give over-confident mistakes. By evaluating on two different benchmark datasets containing consensus annotations from three radiologists, we show that the proposed techniques can improve the detection performance on real CT data. To understand the limitations of both the conventional networks and the proposed augmented networks, we also perform stress-tests on the false positive reduction networks by feeding different types of artificially produced patches. We show that the augmented networks are more robust to both under-represented nodules as well as resistant to noise perturbations.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源