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

Abstract

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 2954 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% AP 0.5. The generality of our approach is also evaluated on the publicly available dataset NODE21. We aspire for our method to serve as an inspiration for future research in the field of pulmonary nodule detection. The dataset and codes will be made in public.