Generalizable Cone Beam CT Esophagus Segmentation Using Physics-Based Data Augmentation

TL;DR

This paper introduces a physics-based data augmentation technique for training deep learning models to accurately segment the esophagus in both planning and cone-beam CT scans, improving robustness and generalizability.

Contribution

The study presents a novel physics-based augmentation method that enhances esophagus segmentation accuracy across different CT modalities using synthetic artifact generation.

Findings

Achieved 0.81 Dice score on pCTs and 0.74 on CBCTs.

Model trained on synthetic data generalizes well to real clinical data.

Improved segmentation robustness across imaging modalities.

Abstract

Automated segmentation of esophagus is critical in image guided/adaptive radiotherapy of lung cancer to minimize radiation-induced toxicities such as acute esophagitis. We developed a semantic physics-based data augmentation method for segmenting esophagus in both planning CT (pCT) and cone-beam CT (CBCT) using 3D convolutional neural networks. 191 cases with their pCT and CBCTs from four independent datasets were used to train a modified 3D-Unet architecture with a multi-objective loss function specifically designed for soft-tissue organs such as esophagus. Scatter artifacts and noise were extracted from week 1 CBCTs using power law adaptive histogram equalization method and induced to the corresponding pCT followed by reconstruction using CBCT reconstruction parameters. Moreover, we leverage physics-based artifact induced pCTs to drive the esophagus segmentation in real weekly CBCTs. Segmentations were evaluated using geometric Dice and Hausdorff distance as well as dosimetrically using mean esophagus dose and D5cc. Due to the physics-based data augmentation, our model trained just on the synthetic CBCTs was robust and generalizable enough to also produce state-of-the-art results on the pCTs and CBCTs, achieving 0.81 and 0.74 Dice overlap. Our physics-based data augmentation spans the realistic noise/artifact spectrum across patient CBCT/pCT data and can generalize well across modalities with the potential to improve the accuracy of treatment setup and response analysis.