A pilot exploratory study of the potentials of deep learning methods in cancer image segmentation and classification

Abstract

Background: Tumor classification and feature quantification from H&E histology images are critical tasks for cancer diagnosis, cancer research, and treatment. However, both tasks involve tedious and time-consuming manual examination of histology images. We explored the possibilities of using deep learning methods to perform segmentation and classification of histology images of cancer tissue for their potential in computeraided tumor diagnosis and other clinical and research applications. Specifically, we tested selected deep learning methods for their performance in the segmentation of stroma and glandular objects in tumor image data and in the classification of tumor images. We automated these tasks to help facilitate downstream tumor image analysis, reduce the labor load of pathologists, and provide them with a second opinion on their analysis. Methods: We modified a patch-based U-Net model and trained it to perform stroma detection and segmentation in cancer tissue. Then the semantic segmentation capabilities of the U-Net model were compared with that of a DeepLabV3+ model. We also explored the possible use of transfer learning to train a patch-based model to classify cancer tissue images as carcinoma and sarcoma and to further classify them as carcinoma subtypes. Results:In spite of the limited dataset available for the pilot study, we found that the unconventional DeepLabV3+ model performed biomedical image segmentation more effectively than U-Net when k-fold cross-validation was utilized, but U-Net still showed promise as an effective and efficient model when we used a customized validation approach. We believe that the DeepLabV3+ model can perform segmentation with even more accuracy if computation resource constraints are removed or if more data is used to augment the result. In terms of tumor classification, our selected models also consistently achieve test accuracies above 80%, with a model trained using transfer learning with VGG-16 network as the feature extractors, or convolutional base performing best. For multi-class tumor subtype classification, we also observed promising test accuracies from our models, and a customized post-processing method provided even higher prediction accuracy on test set images and this method can be further investigated. Conclusions:This pilot exploratory study provided strong evidence for the powerful potentials of deep learning models for segmentation and classification of tumor image data.

Ensemble Lung Segmentation System Using Deep Neural Networks

“…Recently, deep learning-based semantic segmentation is providing state-of-the-art performance in multiple applications, including medical imaging applications [8], [12], [13]. DeepLabV3+ has been a favorite segmentation model in the field of medical imaging [14]. DeepLabV3+ has beendeveloped by Google's DeepLab as discussed in L.C.…”