Implementation of Spatial-Level Augmentation on Pneumonia Classification with Convolutional Neural Network

Wahyu Andi Saputra

doi:10.47065/bits.v4i2.2270

Wahyu Andi Saputra * Institut Teknologi Telkom, Purwokerto, Indonesia

(*) Corresponding Author

DOI: https://doi.org/10.47065/bits.v4i2.2270

Keywords: Covid-19; Augmentation; Spatial-level; Pneumonia Classification; CNN

Abstract

Pneumonia is a disease with symptoms of difficulty breathing, fever, dry cough, and chest pain. As an indication that a person has COVID-19, it is necessary to immediately identify pneumonia by a doctor whether the lung X-ray image of the patient is classified as pneumonia or not. The importance of early diagnosis can reduce the risk of death in patients. Convolutional Neural Network is one of the fields of study in the realm of Computer Science that can perform image-based object classification. CNN can be used to identify pneumonia based on thorax images based on the features or features displayed on the image. One of the important elements in CNN is the amount of data that can be used for training, validation, and testing. Generally, the more data entered, the more learning material for the CNN system so that the system can classify more accurately. This study aims to measure the accuracy of the CNN model on thoracic-pneumonia images with spatial level augmentation changes. Image augmentation is implemented to increase image variance with initial data of 5856 images. The applied augmentations are Affine, Flip, Pixel Dropout, Random Size Crop, and Shift Scale Rotate. The stages of this research are manually grouping images, implementing augmentation on images, applying training-validation-testing on CNN, and analyzing the output results of the developed system. By using 5 types of augmentation, the dataset used as learning material can be increased up to 5x the original amount. From the research carried out, it was found that the Random-sized Crop type augmentation gave the highest accuracy value of 94.719% or an increase of 3.808% from the non-augmentation testing data. From this research, it is hoped that studies related to augmentation can be a reference regarding the type of augmentation process and its results in finding the CNN accuracy value, especially in the case of pneumonia classification

Downloads

Download data is not yet available.

References

Moch. D. Nugraha, “Analisis Faktor Risiko Kematian Akibat Infeksi Covid-19,” Jurnal Ilmu Kesehatan Bhakti Husada: Health Sciences Journal, vol. 12, no. 2, pp. 204–214, 2021, doi: 10.34305/jikbh.v12i2.343.

H. A. Diah Handayani, Dwi Rendra Hadi, Fathiyah Isbaniah, Erlina Burhan, “Penyakit Virus Corona 2019,” Jurnal Respirologi, vol. 40, no. 2, p. 119, 2020, [Online]. Available: https://jurnalrespirologi.org/index.php/jri/article/download/101/110

M. N. Afifah, “8 Penyakit Bawaan atau Komorbid Covid-19 yang Perlu Diwaspadai,” Kompas.com, 2021. https://health.kompas.com/read/2021/07/31/120100668/8-penyakit-bawaan-atau-komorbid-covid-19-yang-perlu-diwaspadai?page=all (accessed Jan. 21, 2022).

Andrew M. Freeman and J. Townes R. Leigh, “Viral Pneumonia,” National Library of Medicine. StatPearls Publishing, Jan. 2022.

J. M. Galván and O. Rajas, “Review of Non-bacterial Infections in Respiratory Medicine : Viral Pneumonia ଝ,” vol. 51, no. 11, pp. 590–597, 2015.

L. Gattinoni et al., “COVID-19 pneumonia : pathophysiology and management,” European Respiratory Review, no. June, pp. 1–13, 2021, doi: 10.1183/16000617.0138-2021.

W. H. Organization, “Middle East respiratory syndrome coronavirus (MERS-CoV),” WHO WEB, 2019. https://www.who.int/news-room/fact-sheets/detail/middle-east-respiratory-syndrome-coronavirus-(mers-cov) (accessed Feb. 28, 2022).

Y. T. Y. Steve, “Penjelasan tentang Pneumonia & Novel Coronavirus COVID-19,” mountelizabeth, 2020. https://www.mountelizabeth.com.sg/id/healthplus/article/pneumonia-and-coronavirus (accessed Mar. 01, 2022).

N. Yudistira, A. W. Widodo, and B. Rahayudi, “Deteksi Covid-19 pada Citra Sinar-X Dada Menggunakan Deep Learning yang Efisien,” Jurnal Teknologi Informasi dan Ilmu Komputer, vol. 7, no. 6, p. 1289, 2020, doi: 10.25126/jtiik.2020763651.

M. T. Cicerone and C. H. Camp, “Potential Roles for Spectroscopic Coherent Raman Imaging for Histopathology and Biomedicine,” in Neurophotonics and Biomedical Spectroscopy, Elsevier, 2019, pp. 547–570. doi: 10.1016/B978-0-323-48067-3.00021-4.

A. Peryanto, A. Yudhana, and R. Umar, “Klasifikasi Citra Menggunakan Convolutional Neural Network dan K Fold Cross Validation,” Journal of Applied Informatics and Computing, vol. 4, no. 1, pp. 45–51, 2020, doi: 10.30871/jaic.v4i1.2017.

Y. Chen, L. Song, Y. Liu, L. Yang, and D. Li, “A review of the artificial neural network models for water quality prediction,” Applied Sciences (Switzerland), vol. 10, no. 17, 2020, doi: 10.3390/app10175776.

R. Yamashita, M. Nishio, R. K. G. Do, and K. Togashi, “Convolutional neural networks: an overview and application in radiology,” Insights Imaging, vol. 9, no. 4, pp. 611–629, Aug. 2018, doi: 10.1007/s13244-018-0639-9.

N. Nasrullah, J. Sang, M. S. Alam, M. Mateen, B. Cai, and H. Hu, “Automated Lung Nodule Detection and Classification Using Deep Learning Combined with Multiple Strategies.,” Sensors (Basel), vol. 19, no. 17, Aug. 2019, doi: 10.3390/s19173722.

W. Swastika, “Studi Awal Deteksi Covid-19 Menggunakan Citra Ct Berbasis Deep Preliminary Study of Covid-19 Detection Using Ct Image Based on,” Jurnal Teknologi Informasi dan Ilmu Komputer, vol. 7, no. 3, pp. 629–634, 2020, doi: 10.25126/jtiik.202073399.

X. Chen, “Image enhancement effect on the performance of convolutional neural networks,” no. June, pp. 1–40, 2019, [Online]. Available: www.bth.se

C. Luo, X. Li, J. Yin, J. He, D. gao, and J. Zhou, “How Does the Data Set and the Number of Categories Affect CNN-based Image Classification Performance?,” Journal of Software, vol. 14, no. 4, pp. 168–181, 2019, doi: 10.17706/jsw.14.4.168-181.

M. Xu, S. Yoon, A. Fuentes, and D. S. Park, “A Comprehensive Survey of Image Augmentation Techniques for Deep Learning,” May 2022, [Online]. Available: http://arxiv.org/abs/2205.01491

Su. Lau, “Image Augmentation for Deep Learning | by Suki Lau | Towards Data Science,” Towards Data Science, Jul. 11, 2017. https://towardsdatascience.com/image-augmentation-for-deep-learning-histogram-equalization-a71387f609b2 (accessed Aug. 31, 2022).

“Albumentations Documentation.” https://albumentations.ai/docs/ (accessed Sep. 01, 2022).

S. Adi Widiarto, W. Andi Saputra, and A. Ratna Dewi, “Klasifikasi Citra X-Ray Toraks Dengan Menggunakan Contrast Limited Adaptive Histogram Equalization Dan Convolutional Neural Network (Studi Kasus: Pneumonia).”

“Keras API reference.” https://keras.io/api/ (accessed Sep. 01, 2022).

H. X. Huynh, B. Q. Truong, K. T. Nguyen Thanh, and D. Q. Truong, “Plant Identification Using New Architecture Convolutional Neural Networks Combine with Replacing the Red of Color Channel Image by Vein Morphology Leaf,” Vietnam Journal of Computer Science, vol. 07, no. 02, pp. 197–208, May 2020, doi: 10.1142/S2196888820500116.

M. Dhouibi, A. K. ben Salem, and S. ben Saoud, “Optimization of CNN model for image classification,” in 2021 IEEE International Conference on Design & Test of Integrated Micro & Nano-Systems (DTS), Jun. 2021, pp. 1–6. doi: 10.1109/DTS52014.2021.9497988

Bila bermanfaat silahkan share artikel ini

Berikan Komentar Anda terhadap artikel Implementation of Spatial-Level Augmentation on Pneumonia Classification with Convolutional Neural Network