Abstract
Chest rib fracture can be regarded as a kind of small objects to be detected with complex shape and large similarity to the surrounding background. Fatigue caused by work intensity is the main reason why radiologists’ detection efficiency and accuracy of rib fracture decrease over time. This work proposes an automatical detection method of rib fractures in chest CT image based on a centernet network with heatmap pyramid structure. Firstly, a hierarchical fusion hourglass network is constructed to perform feature extraction, and the multi-branch residual blocks and fusion strategy in it also play a positive role in improving the feature extraction ability. Secondly, a heatmap pyramid structure which can utilize multi-scale information is adopted to generate more accurate corner information. In addition, a non-local dual spatial attention module is designed to reduce the misclassification of corner and center points. Experimental results show that the proposed network achieves an accurate detection of rib fracture with a mean Average Precision (mAP) of more than 89%.
Similar content being viewed by others
Availability of data and materials
All authors are sure that all data and materials as well as software application support published claims and comply with field standards.
References
Schulze, C., Hoppe, H., Schweitzer, W., Schwendener, N., Grabherr, S., Jackowski, C.: Rib fractures at postmortem computed tomography (PMCT) validated against the autopsy. Forensic Sci. Int. 233(1–3), 90–98 (2013). https://doi.org/10.1016/j.forsciint.2013.08.025
Fang, L., Xiaojun, G., Cheng, L., Liang, J., Xiu, J., Ming, L.: The value of CT ribs unfolded algorithm in the diagnosis of rib fracture[J]. Chinese J. Radiol. (2017). https://doi.org/10.3760/CMA.J.ISSN.1005-1201.2017.09.013
Saba, L., Biswas, M., Kuppili, V., et al.: The present and future of deep learning in radiology. Eur. J. Radiol. 114, 14–24 (2019). https://doi.org/10.1016/j.ejrad.2019.02.038
Esteva, A., Kuprel, B., Novoa, R.A., et al.: Dermatologist-level classification of skin cancer with deep neural networks [published correction appears in Nature. Nature. 542(7639), 115–118 (2017). https://doi.org/10.1038/nature21056
Bonavita, I., Rafael-Palou, X., Ceresa, M., Piella, G., Ribas, V., González Ballester, M.A.: Integration of convolutional neural networks for pulmonary nodule malignancy assessment in a lung cancer classification pipeline. Comput Methods Programs Biomed. 185, 105172 (2020). https://doi.org/10.1016/j.cmpb.2019.105172
Li, X., Chen, H., Qi, X., Dou, Q., Fu, C.W., Heng, P.A.: H-DenseUNet: hybrid densely connected UNet for liver and tumor segmentation from CT volumes. IEEE Trans. Med. Imaging. 37(12), 2663–2674 (2018). https://doi.org/10.1109/TMI.2018.2845918
Raisuddin, A.M., Vaattovaara, E., Nevalainen, M., et al.: Critical evaluation of deep neural networks for wrist fracture detection. Sci. Rep. 11(1), 6006 (2021). https://doi.org/10.1038/s41598-021-85570-2
Nguyen, H. T., Pham, H. H., Nguyen, N. T., et al. VinDr-SpineXR: A deep learning framework for spinal lesions detection and classification from radiographs[C]//International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer, Cham, 2021: 291–301. doi: https://doi.org/10.1007/978-3-030-87240-3_28
Wei J, Yao J, Zhanga G, et al. Semi-supervised object detection based on single-stage detector for thighbone fracture localization[J]. arXiv preprint arXiv:2210.10998, 2022. doi:https://doi.org/10.48550/arXiv.2210.10998
Urakawa, T., Tanaka, Y., Goto, S., Matsuzawa, H., Watanabe, K., Endo, N.: Detecting intertrochanteric hip fractures with orthopedist-level accuracy using a deep convolutional neural network. Skeletal Radiol. 48(2), 239–244 (2019). https://doi.org/10.1007/s00256-018-3016-3
Wang, W., et al.: Attention mechanism-based deep learning method for hairline fracture detection in hand X-rays. Neural Comput. Appl. (2022). https://doi.org/10.1007/S00521-022-07412-0/TABLES/8
Guan, B., et al.: Automatic detection and localization of thighbone fractures in X-ray based on improved deep learning method. Comput. Vis. Image Underst. 216, 103345 (2022). https://doi.org/10.1016/j.cviu.2021.103345
Girshick, Ross, et al. "Rich feature hierarchies for accurate object detection and semantic segmentation." Proceedings of the IEEE conference on computer vision and pattern recognition. 2014. doi: https://doi.org/10.48550/arXiv.1311.2524
Girshick, Ross. "Fast r-cnn." Proceedings of the IEEE international conference on computer vision. 2015. doi: https://doi.org/10.48550/arXiv.1504.08083
Ren, S., He, K., Girshick, R., Sun, J.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017). https://doi.org/10.1109/TPAMI.2016.2577031
Cai, Z., and N, Va.: "Cascade r-cnn: Delving into high quality object detection." Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2018. doi: https://doi.org/10.48550/arXiv.1712.00726
Redmon, J,: et al. "You only look once: Unified, real-time object detection." Proceedings of the IEEE conference on computer vision and pattern recognition. 2016. doi: https://doi.org/10.48550/arXiv.1506.02640
Redmon, J. and Ali F.: "YOLO9000: better, faster, stronger." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017. doi: https://doi.org/10.48550/arXiv.1612.08242
Redmon, J, and Ali F.: "Yolov3: An incremental improvement." arXiv preprint arXiv:1804.02767 (2018). doi: https://doi.org/10.48550/arXiv.1804.02767
Bochkovskiy, A, Chien-Yao W, and Hong-Yuan ML.: "Yolov4: Optimal speed and accuracy of object detection." arXiv preprint arXiv:2004.10934 (2020). doi: https://doi.org/10.48550/arXiv.2004.10934
Law, H, and Jia D.: "Cornernet: Detecting objects as paired keypoints." Proceedings of the European conference on computer vision (ECCV). 2018. doi: https://doi.org/10.48550/arXiv.1808.01244
Duan, K,: et al. "Centernet: Keypoint triplets for object detection." Proceedings of the IEEE/CVF international conference on computer vision. 2019. doi: https://doi.org/10.48550/arXiv.1904.08189
Tian, Z.: et al. "Fcos: Fully convolutional one-stage object detection." Proceedings of the IEEE/CVF international conference on computer vision. 2019. doi: https://doi.org/10.48550/arXiv.1904.01355
Szegedy, C.: et al. "Going deeper with convolutions." Proceedings of the IEEE conference on computer vision and pattern recognition. 2015. doi: https://doi.org/10.48550/arXiv.1409.4842
Kingma, Diederik P., and Jimmy B.: "Adam: A method for stochastic optimization." arXiv preprint arXiv:1412.6980 (2014). doi: https://doi.org/10.48550/arXiv.1412.6980
Lin, T-Y.: et al. "Microsoft coco: Common objects in context." European conference on computer vision. Springer, Cham, 2014. doi: https://doi.org/10.48550/arXiv.1405.0312
Ge, Z.: et al. "Yolox: Exceeding yolo series in 2021." arXiv preprint arXiv: 2107.08430 (2021). doi: https://doi.org/10.48550/arXiv.2107.08430
Zhou, X., Vladlen K., and Philipp K.: "Probabilistic two-stage detection." arXiv preprint arXiv:2103.07461 (2021). doi: https://doi.org/10.48550/arXiv.2103.07461
Sun, P.: et al. "Sparse r-cnn: End-to-end object detection with learnable proposals." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021.doi: https://doi.org/10.48550/arXiv.2011.12450
Acknowledgements
The authors gratefully acknowledge the support of the Shanxi Bethune Hospital for providing the data.
Funding
The Natural Science for Youth Foundation of China under Grant 62001321, Fundamental Research Program of Shanxi Province under Grants 20210302124265, and the Excellent Graduate Innovation Project of TYUST under Grant XCX212028.
Author information
Authors and Affiliations
Contributions
All authors contributed to the study conception and design. Material preparation, data collection and analysis were performed by XZ, HS, YS and RL. The first draft of the manuscript was written by YS, and all authors commented on previous versions of the manuscript. All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare no conflicts of interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Su, Y., Zhang, X., Shangguan, H. et al. Rib fracture detection in chest CT image based on a centernet network with heatmap pyramid structure. SIViP 17, 2343–2350 (2023). https://doi.org/10.1007/s11760-022-02451-5
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11760-022-02451-5