Deep learning vs. machine learning in the detection of pulmonary nodules with chest X-rays
摘要
This study compares lung nodule detection with nine machine learning models, deep (DL) and non-deep (ML), using the same chest X-ray database.
MethodsThree Support Vector Machine models, four Random Forest, one SqueezeNet, and one YOLO were trained and validated using the same computing facilities and the same anatomical region as input, consisting of the segmentation of both lungs. The images were further pre-processed by a common algorithm before the ML models and by the same UNet convolutional network before the DL models. The ML models were trained based on a feature vector calculated on the detection of candidate nodules, while the DL models were trained directly from the image segmentation. The best ML model and the DL models were also tested using an external database. The models were compared based on accuracy, precision, sensitivity, and F1 score.
ResultsUnder the present standardized experimental conditions, the ML models were found to demonstrate a better balance between metrics, and the best model had generalization power. The DL models showed biases due to overfitting to the training data. The shortcomings of the training conditions for the DL models were analyzed.
ConclusionAll tested models performed better than humans. ML models performed better under limited training data, and DL showed theoretical potential but requires further optimization.