Related Experiment Videos
An optimized EfficientNetB7 deep learning model for lung cancer tissue pathology image recognition
Xiongwen He1, Lili Fu1, Baicheng Xu1
1Department of Medical Oncology, The Second Affiliated Hospital of Hainan Medical University, Haikou, Hainan, China.
Background:
Lung cancer is the leading cause of cancer deaths worldwide. This study constructed and optimized a deep learning model to recognize lung cancer pathological images for intelligent classification.
Methods:
The LC25000 dataset was adopted for model training, and the LungHist700 dataset served as independent validation, with validation F1 score as the primary evaluation metric. Nine classic convolutional neural networks (DenseNet201, EfficientNetB7, EfficientNetV2S, InceptionResNetV2, MobileNetV3Large, NASNetLarge, ResNet50V2, VGG19, and Xception) were first benchmarked. Afterwards, three ensemble strategies including Equal-Weighted Averaging, Performance-Weighted Averaging and XGBoost were tested. Meanwhile, hyperparameter tuning was implemented on the optimal single model.
Result:
All nine baseline models obtained good performance on training and internal test datasets. Among individual models, EfficientNetB7 performed best with an F1 score of 0.891 and AUC of 0.749. Ensemble methods only reached a maximum F1 of 0.889, showing no obvious advantage over the best single model. By contrast, the hyperparameter-optimized EfficientNetB7 achieved remarkable generalization improvement, with accuracy 0.873, precision 0.864, recall 0.999, F1 score0.926 and AUC 0.871.
Conclusion:
The optimized model effectively recognizes lung cancer pathological images and supplies technical support for intelligent diagnosis. Multi-center large-sample clinical validation is required for future clinical transformation.