TGPO-WRHNN: Two-stage Grad-CAM-guided PMRS Optimization and weighted-residual hypergraph neural network for pneumonia detection

Abstract

Recent studies based on chest X-ray images have shown that pneumonia can be effectively detected using deep convolutional neural network methods. However, these methods tend to introduce additional noise and extract only local feature information, making it difficult to express the relationship between data objects. This study proposes a Two-stage Grad-CAM-guided pre-trained model and removal scheme (PMRS) Optimization and weighted-residual hypergraph neural network model (TGPO-WRHNN). First, our model extracts high-dimensional features using the TGPO module to capture both global and local information from an image. Second, we propose a new distance-based hypergraph construction method (DBHC) to amplify the difference between distances and better distinguish the relation between nearby and distant neighbors. Finally, we introduce a weighted-residual hypergraph convolution module (WRHC) to ensure the model maintains excellent performance, even at deeper levels. Our model was tested on a dataset of chest X-ray images of pediatric patients aged 1 to 5 years at the Guangzhou Women and Children’s Medical Centre by 10-fold cross-validation. The results showed that the method achieved a maximum accuracy of 98.97%, precision of 98.86%, recall of 98.43%, F1 score of 98.64%, and AUC of 99.78%. Compared to other existing models, our model demonstrated improvements of 0.87%, 0.86%, 0.16%, and 0.38% in terms of accuracy, precision, F1 score, and AUC, respectively.