Author:
Kundur Niranjan C.,Anil Bellary Chiterki,Dhulavvagol Praveen M.,Ganiger Renuka,Ramadoss Balakrishnan
Abstract
Pneumonia is a severe respiratory disease with potentially life-threatening consequences if not promptly diagnosed and treated. Chest X-rays are commonly employed for pneumonia detection, but interpreting the images can pose challenges. This study explores the efficacy of four popular transfer learning models, namely VGG16, ResNet, InceptionNet, and DenseNet, alongside a custom CNN model for this task. The model performance is evaluated using Mean Absolute Error (MAE) as the performance metric. The findings reveal that VGG16 outperforms the other transfer learning models, achieving the lowest MAE (66.19). To optimize the model training process, a distributed training strategy utilizing TensorFlow's TPU (Tensor Processing Unit) strategy is implemented. The custom CNN model is parallelized using TPU's multiple instances available over the cloud, enabling efficient computation parallelization and significantly reducing model training times. The experimental results demonstrate a remarkable decrease of 68.36% and 54.74% in model training times for the CNN model when trained using TPU compared to training on a CPU and GPU, respectively.
Publisher
Engineering, Technology & Applied Science Research
Reference16 articles.
1. S. Kalgutkar et al., "Pneumonia Detection from Chest X-ray using Transfer Learning," in 2021 6th International Conference for Convergence in Technology (I2CT), Maharashtra, India, Apr. 2021.
2. S. V. Militante and B. G. Sibbaluca, "Pneumonia Detection Using Convolutional Neural Networks," International Journal of Scientific & Technology Research, vol. 9, no. 4, pp. 1332–1337, 2020.
3. R. Kundu, R. Das, Z. W. Geem, G.-T. Han, and R. Sarkar, "Pneumonia detection in chest X-ray images using an ensemble of deep learning models," PLOS ONE, vol. 16, no. 9, 2021, Art. no. e0256630.
4. A. Ranjan, C. Kumar, R. K. Gupta, and R. Misra, "Transfer Learning Based Approach for Pneumonia Detection Using Customized VGG16 Deep Learning Model," in Internet of Things and Connected Technologies, Cham, 2022, pp. 17–28.
5. T. Rahman et al., "Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection Using Chest X-ray," Applied Sciences, vol. 10, no. 9, Jan. 2020, Art. no. 3233.