Human-in-the-Loop Informed Deep Learning Rectal Tumor Segmentation on Pre-Treatment MRI

Michael Kong, Thomas DeSilvio, Leo Bao, Brennan Flannery, Benjamin N. Parker, Stephen Tang, Murad Labbad, Gregory O’Connor, Amit Gupta, Emily Steinhagen, Andrei S. Purysko, William Hall, David Liska, Eric L. Marderstein, Aaron Carroll, Marka Crittenden, Michael Gough, Kristina Young, Satish E. Viswanath

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Precise segmentation of rectal cancer tumors on routine MRI is critical for accurate clinical staging and downstream computational analyses. While deep learning-based segmentation algorithms have shown much promise in automating the otherwise tedious, subjective, and costly process of manual segmentation, they require significant amounts of manually annotated data for training. To address these limitations of deep learning-based segmentation models, we present a novel deep learning framework that incorporates human-in-the-loop (HITL) refinement to automatically delineate rectal tumors on multi-plane pre-treatment MR imaging. When evaluated on multiple holdout validation cohorts including a clinical trial dataset, the post-HITL segmentation model significantly outperformed the pre-HITL model with median dice similarity coefficient of 0.763 and Hausdorff distance of 28.4mm in comparison to 0.601 and 31.8mm, respectively. HITL refinement learning also significantly accelerated the manual annotation process by 20 minutes. HITL learning represents a feasible, effective, and efficient solution to semi-automated tumor segmentation on routine rectal cancer MRI scans.

Original languageEnglish
Title of host publicationMedical Imaging 2024
Subtitle of host publicationComputer-Aided Diagnosis
EditorsWeijie Chen, Susan M. Astley
PublisherSPIE
ISBN (Electronic)9781510671584
DOIs
StatePublished - 2024
EventMedical Imaging 2024: Computer-Aided Diagnosis - San Diego, United States
Duration: Feb 19 2024Feb 22 2024

Publication series

NameProgress in Biomedical Optics and Imaging - Proceedings of SPIE
Volume12927
ISSN (Print)1605-7422

Conference

ConferenceMedical Imaging 2024: Computer-Aided Diagnosis
Country/TerritoryUnited States
CitySan Diego
Period02/19/2402/22/24

Keywords

  • MRI
  • Rectal cancer
  • deep learning
  • human-in-the-loop
  • multi-plane
  • segmentation

Fingerprint

Dive into the research topics of 'Human-in-the-Loop Informed Deep Learning Rectal Tumor Segmentation on Pre-Treatment MRI'. Together they form a unique fingerprint.

Cite this