论文标题
使用Yolo V5S结节检测和3D神经网络分类器从CT图像鉴定肺结核的分层方法
A hierarchical approach for pulmonary nodules identification from CT images using YOLO v5s nodule detection and 3D neural network classifier
论文作者
论文摘要
在第一步中,使用预训练的模型(YOLO)来检测所有可疑的点头。使用397个CT图像对YOLO模型进行了重新训练,以检测CT图像中的整个结节。为了最大化模型的灵敏度,在第一阶段设置了0.3的置信度(对象检测的概率阈值)(确保从输入CT图像中检测到整个可疑结节)。层次结构模型的目的是检测和对整个肺结节(从CT图像)进行较低的假阴性速率(来自CT图像)。鉴于第一步的结果,我们提出了一个3D CNN分类器,以分析和分类由Yolo模型检测到的可疑结节,以实现具有非常低的假阴性速率的结节检测框架。使用Luna 16数据集评估了该框架,该数据集由888个CT图像组成,其中包含肺中1186个结节和400000个非结节的位置。结果:由于Yolo模型中使用的置信度较低,因此检测到大量的假阳性。利用3D分类器,结节检测的准确性得到了显着增强。当使用50%的置信度水平时,YOLO模型检测到294个可疑结节(在321中),其中有107个假阳性(187个真实的阳性)。通过将置信度降低到30%,通过Yolo模型确定了459个可疑结节,其中138是假阳性,而321个是真正的阳性。当置信度水平为30%的YOLO模型的结果被送入3D CNN分类器中时,结论得出了98.4%和AUC的结节检测准确性,得出的结论是:拟议的框架导致一些假阴性和假阳性在CT图像中发现了一些假阴性和假阳性预测。所提出的方法将有助于检测CT图像作为决策支持工具的肺结核。
In the first step, a pre-trained model (YOLO) was used to detect all suspicious nod-ules. The YOLO model was re-trained using 397 CT images to detect the entire nodule in CT images. To maximize the sensitivity of the model, a confidence level (the probability threshold for object detection) of 0.3 was set for nodule detection in the first phase (ensuring the entire suspicious nodules are detected from the input CT images). The aim of the hierarchy model is to detect and classify the entire lung nodules (from CT images) with a low false-negative rate. Given the outcome of the first step, we proposed a 3D CNN classifier to analyze and classify the suspicious nodules detected by the YOLO model to achieve a nodule detection framework with a very low false-negative rate. This framework was evaluated using the LUNA 16 dataset, which consists of 888 CT images containing the location of 1186 nodules and 400000 non-nodules in the lung. Results: A large number of false positives were detected due to the low confidence level used in the YOLO model. Utilizing the 3D classifier, the accuracy of nodule detection was remarkably enhanced. The YOLO model detected 294 suspicious nodules (out of 321) when using a confidence level of 50%, wherein there were 107 false positives (187 true positives). By reducing the confidence level to 30%, 459 suspicious nodules were identified by the YOLO model, wherein 138 were false positives, and 321 were true positives. When the outcome of the YOLO model with a confidence level of 30% was fed into the 3D CNN classifier, a nodule detection accuracy of 98.4% and AUC of 98.9% were achieved Conclusion: The proposed framework resulted in a few false-negative and false-positives pre-dictions in nodule detection from CT images. The proposed approach would be helpful in detecting pulmonary nodules from CT images as a decision support tool.