Expert-level pediatric brain tumor segmentation in a limited data scenario with stepwise transfer learning

Aidan Boyd, Zezhong Ye, Sanjay Prabhu, Michael C Tjong, Yining Zha, Anna Zapaishchykova, Sridhar Vajapeyam, Hasaan Hayat, Rishi Chopra, Kevin X Liu, Ali Nabavidazeh, Adam Resnick, Sabine Mueller, Daphne Haas-Kogan, Hugo J W L Aerts, Tina Poussaint, Benjamin H Kann

Research output: Working paper / PreprintPreprint

Abstract

PURPOSE: Artificial intelligence (AI)-automated tumor delineation for pediatric gliomas would enable real-time volumetric evaluation to support diagnosis, treatment response assessment, and clinical decision-making. Auto-segmentation algorithms for pediatric tumors are rare, due to limited data availability, and algorithms have yet to demonstrate clinical translation. METHODS: We leveraged two datasets from a national brain tumor consortium (n=184) and a pediatric cancer center (n=100) to develop, externally validate, and clinically benchmark deep learning neural networks for pediatric low-grade glioma (pLGG) segmentation using a novel in-domain, stepwise transfer learning approach. The best model [via Dice similarity coefficient (DSC)] was externally validated and subject to randomized, blinded evaluation by three expert clinicians wherein clinicians assessed clinical acceptability of expert- and AI-generated segmentations via 10-point Likert scales and Turing tests. RESULTS: The best AI model utilized in-domain, stepwise transfer learning (median DSC: 0.877 [IQR 0.715-0.914]) versus baseline model (median DSC 0.812 [IQR 0.559-0.888]; <0.05). On external testing (n=60), the AI model yielded accuracy comparable to inter-expert agreement (median DSC: 0.834 [IQR 0.726-0.901] vs. 0.861 [IQR 0.795-0.905], =0.13). On clinical benchmarking (n=100 scans, 300 segmentations from 3 experts), the experts rated the AI model higher on average compared to other experts (median Likert rating: 9 [IQR 7-9]) vs. 7 [IQR 7-9], <0.05 for each). Additionally, the AI segmentations had significantly higher ( <0.05) overall acceptability compared to experts on average (80.2% vs. 65.4%). Experts correctly predicted the origins of AI segmentations in an average of 26.0% of cases. CONCLUSIONS: Stepwise transfer learning enabled expert-level, automated pediatric brain tumor auto-segmentation and volumetric measurement with a high level of clinical acceptability. This approach may enable development and translation of AI imaging segmentation algorithms in limited data scenarios.
Original languageEnglish
PublisherMedRxiv
DOIs
Publication statusPublished - 18 Sept 2023

Keywords

  • Automatic Tumor Segmentation
  • Deep Neural Network
  • MRI
  • Pediatric Low-Grade Glioma
  • nnUNet

Cite this