Enhanced lung image segmentation using deep learning

Neural Comput Appl. 2022 Jan 3:1-15. doi: 10.1007/s00521-021-06719-8. Online ahead of print.

Abstract

With the advances in technology, assistive medical systems are emerging with rapid growth and helping healthcare professionals. The proactive diagnosis of diseases with artificial intelligence (AI) and its aligned technologies has been an exciting research area in the last decade. Doctors usually detect tuberculosis (TB) by checking the lungs' X-rays. Classification using deep learning algorithms is successfully able to achieve accuracy almost similar to a doctor in detecting TB. It is found that the probability of detecting TB increases if classification algorithms are implemented on segmented lungs instead of the whole X-ray. The paper's novelty lies in detailed analysis and discussion of U-Net + + results and implementation of U-Net + + in lung segmentation using X-ray. A thorough comparison of U-Net + + with three other benchmark segmentation architectures and segmentation in diagnosing TB or other pulmonary lung diseases is also made in this paper. To the best of our knowledge, no prior research tried to implement U-Net + + for lung segmentation. Most of the papers did not even use segmentation before classification, which causes data leakage. Very few used segmentations before classification, but they only used U-Net, which U-Net + + can easily replace because accuracy and mean_iou of U-Net + + are greater than U-Net accuracy and mean_iou , discussed in results, which can minimize data leakage. The authors achieved more than 98% lung segmentation accuracy and mean_iou 0.95 using U-Net + + , and the efficacy of such comparative analysis is validated.

Keywords: Data preprocessing; Deep learning; Lung segmentation; Segmentation models; TB dataset.