Multiclass Segmentation of Pulmonary Diseases using Convolutional Neural Network

Abstract

Pulmonary disease has affected tens of millions of people in the world. This disease has also become the cause of death of millions of its sufferers every year. In addition, lung disease has also become the cause of other respiratory complications, which also causes the death of the sufferer. The diagnosis of pulmonary diseases through medical imaging is a significant challenge in computer vision and medical image processing. The difficulty is due to the wide variety in infected areas' shape, dimension, and location. Another challenge is to differentiate one lung disease from the other. Discriminating pulmonary diseases is a notable concern in the diagnosis of pulmonary disease. We have adopted the deep learning convolutional neural network in this study to address these challenges. Seven models were constructed using the Mask Region-based Convolutional Neural Network (Mask-RCNN) architecture to detect and segment infected areas within the lung region from CT scan imagery. The evaluation results show that the best model obtained scores of 91.98%, 85.25%, and 93.75% for DSC, MIoU, and mAP, respectively. The segmentation results are then visualized.