Age Encoded Adversarial Learning for Pediatric CT Segmentation

Saba Heidari Gheshlaghi1, Chi Nok Enoch Kan2, Taly Gilat Schmidt3

  • 1Department of Computer Science, Marquette University, Milwaukee, WI 53233, USA.

Insights

This study introduces a new deep learning framework, CFG-SegNet, to improve organ segmentation in pediatric CT scans, addressing data limitations. The method enhances accuracy for organs like the liver and heart, crucial for medical diagnoses.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Computer Vision

Background:

  • Accurate organ segmentation in CT images is vital for disease diagnosis, treatment planning, and radiation therapy.
  • Data scarcity, especially in pediatric CT segmentation due to radiation sensitivity, poses a significant challenge.
  • Existing methods struggle with limited datasets, impacting diagnostic and therapeutic precision.

Purpose of the Study:

  • To develop a novel segmentation framework, CFG-SegNet, to overcome data limitations in pediatric CT organ segmentation.
  • To incorporate an auxiliary classifier generative adversarial network (ACGAN) that conditions on age for enhanced feature generation.
  • To improve the accuracy and robustness of organ segmentation in low-data scenarios.

Main Methods:

  • Proposed a conditional feature generation segmentation network (CFG-SegNet) integrating an age-conditioned ACGAN.
  • Utilized a single loss function and 2.5D segmentation batches for training.
  • Experimented on a dataset of 359 pediatric subjects (5 days to 16 years).

Main Results:

  • CFG-SegNet achieved high average Dice Similarity Coefficients (DSC): 0.681 (prostate), 0.619 (uterus), 0.912 (liver), and 0.832 (heart).
  • Demonstrated improved segmentation accuracy compared to U-Net by 2.7% (prostate), 2.6% (uterus), 2.8% (liver), and 3.4% (heart).
  • The framework showed superior performance in segmenting organs with limited training data.

Conclusions:

  • The proposed CFG-SegNet effectively addresses data limitations in pediatric CT organ segmentation.
  • The age-conditioned ACGAN enhances feature generation, leading to more precise segmentation.
  • This framework offers a promising solution for improving medical image analysis in data-scarce pediatric populations.

Related Concept Videos