Multi-class Tissue Segmentation of CT images using an Ensemble Deep Learning method.

Abstract

Microwave ablation (MWA) therapy is a well-known technique for locally destroying lung tumors with the help of computed tomography (CT) images. However, tumor recurrence occurs because of insufficient ablation of the tumor. In order to perform an accurate treatment of lung cancer, there is a demand to determine the tumor area precisely. To address the problem at hand, which involves accurately segmenting organs and tumors in CT images obtained during MWA therapy, physicians could benefit from a semantic segmentation method. However, such a method typically requires a large number of images to achieve optimal results through deep learning techniques. To overcome this challenge, our team developed four different (multiple) U-Net based semantic segmentation models that work in conjunction with one another to produce a more precise segmented image, even when working with a relatively small dataset. By combining the highest weight value of segmentation from multiple methods into a single output, we can achieve a more reliable and accurate segmentation outcome. Our approach proved successful in segmenting four different tissue structures, including lungs, lung tumors, and ablated tissues in CT medical images. The Intersection over Union (IoU) is employed to quantitatively evaluate the proposed method. The method shows the highest average IoU, with 0.99 for the background, 0.98 for the lung, 0.77 for the ablated, and 0.54 for the tumor tissue. The results show that employing multiple DL methods is superior to that of individual base-learner models for all four different tissue structures, even in the presence of the relatively small dataset.Clinical relevance- An essential issue of tumor ablation therapy is to know when the entire tumor tissue has completely been destroyed. However, as it is difficult to distinguish between destroyed and living tumor, this is hardly reliable in clinical practice during MWA therapy, especially when working with a small dataset. Improved AI segmentation methods can help to improve performance to reduce recurrence.