Detection, Localization of Cardiomegaly and TB Disease of CXR Images using Deep Learning

Abstract

INTRODUCTION: Tuberculosis (TB) continues to pose a significant worldwide public health concern, as it stands as the primary contributor to mortality stemming from infectious illnesses. Cardiomegaly, characterized by an enlarged heart, poses medical concern as well.OBJECTIVES: Timely identification of Cardiomegaly is vital for effective management. Chest X-ray diagnosis is an easily available method with less radiation exposure to detect several lung infections and heart enlargement. Utilizing computer-aided diagnostic systems can aid in the early detection of lung conditions and the enlargement of the heart.METHODS: We worked on different state-of-the-art CNN architectures such as VGG, DenseNet and EfficientNet with customization over dataset generated from combination of multiple publicly available datasets, which consists of 12939 annotated images across three different categories, one being normal and other two being TB and cardiomegaly diseases..RESULTS: EfficientNetB5 with optimization has shown excellent results amongst others in classifying Tuberculosis and Cardiomegaly with a remarkable accuracy of 97%.CONCLUSION: The proposed model is ready for clinical diagnosis and triaging of X-ray images. Our solution also offers efficient ways to show the presence of the above diseases using Grad-CAM technique.