论文标题
MHSNET:多头和空间注意网络,具有假阳性降低的肺结节检测
MHSnet: Multi-head and Spatial Attention Network with False-Positive Reduction for Pulmonary Nodules Detection
论文作者
论文摘要
多年来,肺癌的死亡率在癌症中一直在癌症中排名较高。肺癌的早期发现对于预防疾病,治愈和死亡率降低至关重要。但是,现有的肺结核检测方法引入了过多的假阳性建议,以实现高灵敏度,这在临床情况下是不切实际的。在本文中,我们提出了多头检测和空间挤压和注意网络MHSNET,以检测肺结核,以帮助医生早期诊断为肺癌。具体来说,我们首先引入多头探测器和跳过连接,以自定义大小,形状和类型的各种结节,并捕获多尺度功能。然后,我们实施了一个空间注意模块,以使网络能够专注于不同的区域,这些区域受到经验丰富的临床医生筛查CT图像的不同启发,这会导致较少的假阳性建议。最后,我们使用线性回归模型提出了一个轻巧但有效的假阳性减少模块,以减少误报提案的数量,而前线网络上没有任何约束。与最先进的模型相比,广泛的实验结果表明,根据平均Froc,敏感性,尤其是错误的发现率(2.98%和2.18%,平均Froc和敏感性提高了2.98%和2.18%,在平均Froc和敏感性方面提高了5.62%和28.33%的降低,在虚假发现率和每阶段的平均候选人中降低了2.62%和28.33%)。假阳性减少模块将每次扫描产生的候选者的平均数量显着降低68.11%,假发现率提高了13.48%,这有望根据检测结果减少下游任务的分心建议。
The mortality of lung cancer has ranked high among cancers for many years. Early detection of lung cancer is critical for disease prevention, cure, and mortality rate reduction. However, existing detection methods on pulmonary nodules introduce an excessive number of false positive proposals in order to achieve high sensitivity, which is not practical in clinical situations. In this paper, we propose the multi-head detection and spatial squeeze-and-attention network, MHSnet, to detect pulmonary nodules, in order to aid doctors in the early diagnosis of lung cancers. Specifically, we first introduce multi-head detectors and skip connections to customize for the variety of nodules in sizes, shapes and types and capture multi-scale features. Then, we implement a spatial attention module to enable the network to focus on different regions differently inspired by how experienced clinicians screen CT images, which results in fewer false positive proposals. Lastly, we present a lightweight but effective false positive reduction module with the Linear Regression model to cut down the number of false positive proposals, without any constraints on the front network. Extensive experimental results compared with the state-of-the-art models have shown the superiority of the MHSnet in terms of the average FROC, sensitivity and especially false discovery rate (2.98% and 2.18% improvement in terms of average FROC and sensitivity, 5.62% and 28.33% decrease in terms of false discovery rate and average candidates per scan). The false positive reduction module significantly decreases the average number of candidates generated per scan by 68.11% and the false discovery rate by 13.48%, which is promising to reduce distracted proposals for the downstream tasks based on the detection results.