ENHANCING THE EFFECTIVENESS OF TUBERCULOSIS DIAGNOSIS MODELS BASED ON VISUALIZATION TECHNIQUES | Vinh | TNU Journal of Science and Technology

ENHANCING THE EFFECTIVENESS OF TUBERCULOSIS DIAGNOSIS MODELS BASED ON VISUALIZATION TECHNIQUES

About this article

Received: 28/05/24                Revised: 26/06/24                Published: 26/06/24

Authors

1. Nguyen Trong Vinh, Lac Hong University
2. Pham Thi Bay, Lilama 2 International Technology College
3. Lam Thanh Hien, Lac Hong University
4. Do Nang Toan Email to author, Institute of Information Technology - Vietnam Academy of Science and Technology

Abstract


The robust development of deep learning models has solved many practical problems, including supporting disease diagnosis through images. Although much progress has been achieved, explaining the decisions made by deep learning models remains a significant challenge. In some cases, these models use information outside the diagnostic area. This paper proposes a solution through visualization in diagnosing tuberculosis from chest X-ray images. It highlights the regions within the images that the deep learning model utilizes, aiming to detect discrepancies within the training dataset images. This enables data normalization and the application of techniques to improve the model's accuracy. The proposal has been implemented in trials and has shown effectiveness with deep learning models for chest X-ray images, aiding in the diagnosis of tuberculosis. It not only improves the reliability but also increases the accuracy of the model by adjusting and normalizing the training data based on real findings from the visualization process.

Keywords


Imaging diagnosis; Data visualization; Deep learning; Score-CAM; Transfer learning

References


[1] VRPACS Software, 2022. [Online]. Available: http://vnpacs.vn. [Accessed April 03, 2024].

[2] T. M. Ha, H. T. Lam, V. D. Ngo, and T. N. Do, "Detecting Tuberculosis from Vietnamese X-Ray Imaging Using Transfer Learning Approach," Computers, Materials & Continua, vol.74, pp. 5001-5016, 2022.

[3] Y. LeCun, Y. Bengio, and G. Hinton, "Deep learning," Nature, vol. 521, no. 7553, pp. 436-444, 2015.

[4] I. Goodfellow, Y. Bengio, and A. Courville, Deep Learning, MIT Press, 2016.

[5] B. Zhou, A. Khosla, A. Lapedriza, A. Oliva, and A. Torralba, "Learning Deep Features for Discriminative Localization," IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016, pp. 2921-2929.

[6] T. Rahman, A. Khandakar, M. A. Kadir, K. R. Islam, K. F. Islam, R. Mazhar, T. Hamid, M. T. Islam, Z. B. Mahbub, M. A. Ayari, and M. E. H. Chowdhury, "Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualization," IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2020, pp. 191586–191601.

[7] K. Simonyan, A. Vedaldi, and A. Zisserman, "Deep inside convolutional networks: Visualising image classification models and saliency maps," CoRR, 2013, doi: 10.48550/arXiv.1312.6034.

[8] M. D. Zeiler and R. Fergus, "Visualizing and understanding convolutional networks," in ECCV, 2014, pp. 818–833.

[9] Z. C. Lipton, "The Mythos of Model Interpretability," Communications of the ACM, vol. 61, no. 10, June 2016, doi: 10.1145/3233231.

[10] R. R. Selvaraju et al., "Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization," International Conference on Computer Vision (ICCV), 2017, doi: 10.1109/ICCV. 2017.74.

[11] R. R. Selvaraju, S. Lee, Y. Shen, H. Jin, S. Ghosh, L. Heck, D. Batra, and D. Parikh, "Taking a hint: Leveraging explanations to make vision and language models more grounded," in Proceedings of the International Conference on Computer Vision (ICCV), 2019, pp. 2591-2600.

[12] H. Wang, Z. Wang, M. Du, F. Yang, Z. Zhang, S. Ding et al., "Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networks," in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2020, pp. 24-25.

[13] H. Jung and Y. Oh, "Towards Better Explanations of Class Activation Mapping," International Conference on Computer Vision (ICCV), 2021, pp. 1336 - 1344.

[14] N. Pandey, “Chest Xray Masks and Labels,” 2020. [Online]. Available: https://www.kaggle.com/ nikhilpandey360/chestxray-masks-and-labels. [Accessed April 03, 2024].




DOI: https://doi.org/10.34238/tnu-jst.10483

Refbacks

  • There are currently no refbacks.
TNU Journal of Science and Technology
Rooms 408, 409 - Administration Building - Thai Nguyen University
Tan Thinh Ward - Thai Nguyen City
Phone: (+84) 208 3840 288 - E-mail: jst@tnu.edu.vn
Based on Open Journal Systems
©2018 All Rights Reserved