A Deep Learning Method for Pneumonia Detection Based on Fuzzy Non-Maximum Suppression

Hongli Wu,Huijuan Lu,Mingzhu Ping,Wenjie Zhu,Zhao Li
DOI: https://doi.org/10.1109/tcbb.2023.3247483
2023-01-01
IEEE/ACM Transactions on Computational Biology and Bioinformatics
Abstract:Pneumonia is one of the largest causes of death in the world. Deep learning techniques can assist doctors to detect the areas of pneumonia in the chest X-rays images. However, existing methods lack sufficient consideration for the large variation scale and the blurred boundary of the pneumonia area. Here, we present a deep learning method based on Retinanet for pneumonia detection. First, we introduce Res2Net into Retinanet to get the multi-scale feature of pneumonia. Then, we proposed a novel predicted boxes fusion algorithm, named Fuzzy Non-Maximum Suppression (FNMS), which gets a more robust predicted box by fusing the overlapping detection boxes. Finally, we get the performance outperforms than existing methods by integrating two models with different backbones. We report the experimental result in the single model case and the model ensemble case. In the single model case, RetinaNet with FNMS algorithm and Res2Net backbone is better than RetinaNet and other models. In the model ensemble case, the final score of predicted boxes that fused by the FNMS algorithm is better than NMS, Soft-NMS, and weighted boxes fusion. Experimental results on the pneumonia detection dataset verify the superiority of the FNMS algorithm and the proposed method in the pneumonia detection task.
What problem does this paper attempt to address?