KochDet: BiFPN-based deep architecture for tuberculosis diagnosis

Patrick Ryan Sales dos Santos,Vitória de Carvalho Brito,Antônio Oseas de Carvalho Filho,João Otávio Bandeira Diniz
DOI: https://doi.org/10.1016/j.bspc.2024.106056
IF: 5.1
2024-02-05
Biomedical Signal Processing and Control
Abstract:Background and Objective: Tuberculosis is a bacterial infection caused by Koch's bacillus, and its transmission occurs through the air. The disease mainly affects the lungs and is considered the second leading cause of death from infection globally, behind only COVID-19. Nevertheless, tuberculosis is curable, and early diagnosis plays a key role in the success of treatment and in preventing the spread of the disease. Sputum smear microscopy has traditionally been the main method for diagnosing and monitoring tuberculosis treatment. In this context, the literature has developed computational approaches to help specialists diagnose tuberculosis by analyzing bacilloscopy images. Methodology: In this work, we propose using EfficientDet and exploring its backbones in the bacilli identification task. We also evaluated four different color representations, and in each experiment, we applied the Grid Search to define the best optimizer and resolution of the images. Results: We evaluated the results through cross-validation with k-fold = 5. The results were promising, with a mAP of 0.650, BACC of 0.839, IoU of 0.523, recall of 0.925, precision of 0.694, and f1-score of 0.774. Conclusion: The proposed method, using the EfficientDet detection network with its B3 to B5 backbones in the LUV color model, stands out as the most suitable approach for identifying MTBs. It shows that it has the potential to support the diagnosis of tuberculosis.
engineering, biomedical
What problem does this paper attempt to address?