Segmentation of Non-Small Cell Lung Carcinomas: Introducing DRU-Net and Multi-Lens Distortion.

Journal: Journal of imaging
Published Date:

Abstract

The increased workload in pathology laboratories today means automated tools such as artificial intelligence models can be useful, helping pathologists with their tasks. In this paper, we propose a segmentation model (DRU-Net) that can provide a delineation of human non-small cell lung carcinomas and an augmentation method that can improve classification results. The proposed model is a fused combination of truncated pre-trained DenseNet201 and ResNet101V2 as a patch-wise classifier, followed by a lightweight U-Net as a refinement model. Two datasets (Norwegian Lung Cancer Biobank and Haukeland University Lung Cancer cohort) were used to develop the model. The DRU-Net model achieved an average of 0.91 Dice similarity coefficient. The proposed spatial augmentation method (multi-lens distortion) improved the Dice similarity coefficient from 0.88 to 0.91. Our findings show that selecting image patches that specifically include regions of interest leads to better results for the patch-wise classifier compared to other sampling methods. A qualitative analysis by pathology experts showed that the DRU-Net model was generally successful in tumor detection. Results in the test set showed some areas of false-positive and false-negative segmentation in the periphery, particularly in tumors with inflammatory and reactive changes. In summary, the presented DRU-Net model demonstrated the best performance on the segmentation task, and the proposed augmentation technique proved to improve the results.

Authors

  • Soroush Oskouei
    Department of Circulation and Medical Imaging, Norwegian University of Science and Technology (NTNU), NO-7491 Trondheim, Norway.
  • Marit Valla
    Department of Clinical and Molecular Medicine, Norwegian University of Science and Technology (NTNU), NO-7491 Trondheim, Norway.
  • André Pedersen
    Department of Health Research, SINTEF, Trondheim, Norway.
  • Erik Smistad
    Department of Circulation and Medical Imaging, Norwegian University of Science and Technology, PO Box 8905, 7491 Trondheim, Norway.
  • Vibeke Grotnes Dale
    Department of Clinical and Molecular Medicine, Norwegian University of Science and Technology (NTNU), NO-7491 Trondheim, Norway.
  • Maren Høibø
    Department of Clinical and Molecular Medicine, Norwegian University of Science and Technology (NTNU), NO-7491 Trondheim, Norway.
  • Sissel Gyrid Freim Wahl
    Department of Pathology, St. Olavs Hospital, Trondheim University Hospital, NO-7030 Trondheim, Norway.
  • Mats Dehli Haugum
    Department of Pathology, St. Olavs Hospital, Trondheim University Hospital, NO-7030 Trondheim, Norway.
  • Thomas Langø
  • Maria Paula Ramnefjell
    Centre for Cancer Biomarkers CCBIO, Department of Clinical Medicine, University of Bergen, NO-5007 Bergen, Norway.
  • Lars Andreas Akslen
    Centre for Cancer Biomarkers CCBIO, Department of Clinical Medicine, University of Bergen, NO-5007 Bergen, Norway.
  • Gabriel Kiss
    Department of Computer Science, Faculty of Information Technology and Electrical Engineering, Norwegian University of Science and Technology, Trondheim 7034, Norway.
  • Hanne Sorger
    Department of Circulation and Medical Imaging, Norwegian University of Science and Technology (NTNU), NO-7491 Trondheim, Norway.

Keywords

No keywords available for this article.