Pulmonary nodule detection in x-ray images by feature augmentation and context aggregation

Chenglin Liu,Zhi Wu,Binquan Wang,Ming Zhu
DOI: https://doi.org/10.1088/1361-6560/ad2013
IF: 3.5
2024-01-20
Physics in Medicine and Biology
Abstract:Recent developments in X-ray image based pulmonary nodule detection have achieved remarkable results. However, existing methods are focused on transferring off-the-shelf coarse-grained classification models and fine-grained detection models rather than developing a dedicated framework optimized for nodule detection. In this paper, we propose PN-DetX, which as we know is the first dedicated pulmonary nodule detection framework. PN-DetX incorporates feature fusion and self-attention into X-ray based pulmonary nodule detection tasks, achieving improved detection performance. Specifically, PN-DetX adopts CSPDarknet backbone to extract features, and utilizes feature augmentation module to fuse features from different levels followed by context aggregation module to aggregate semantic information. To evaluate the efficacy of our method, we collect a LArge-scale Pulmonary NOdule Detection dataset, LAPNOD, comprising 2,954 X-ray images along with expert-annotated ground truths. As we know, this is the first large-scale chest X-ray pulmonary nodule detection dataset. Experiments demonstrates that our method outperforms baseline by 3.8% mAP and 5.1% AP0.5. The generality of our approach is also evaluated on other publicly available dataset. We aspire for our method to serve as an inspiration for future research in the field of pulmonary nodule detection. The data and codes will be made in public.
engineering, biomedical,radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?