Main Session
Sep 30
QP 45 - AI Powered Dose Prediction and Segmentation

1266 - Dual-Model Deep Learning Framework for Accurate Benchmark Dose Prediction In Brain Stereotactic Radiosurgery

10:55am - 11:00am ET
Room 153

Presenter(s)

Wentao Wang, PhD, MS - Thomas Jefferson University, Philadelphia, PA

Z. Xiao1, W. Cao1, H. Liu1, Y. Chen1, R. Shah2, G. O'Neill2, M. Bloomfield2, A. Alluri3, W. Shi1, and W. Wang1; 1Department of Radiation Oncology, Sidney Kimmel Medical College & Cancer Center at Thomas Jefferson University, Philadelphia, PA, 2Sidney Kimmel Medical College at Thomas Jefferson University, Philadelphia, PA, 3Drexel University, Philadephia, PA

Purpose/Objective(s): Dose gradient in brain SRS directly impacts normal tissue toxicity, yet no standardized benchmark currently exists for cross-platform plan quality assessment. We developed a size-stratified deep learning framework to predict patient-specific benchmark dose, employing appropriately scaled volumes of interest (VOI) to ensure accurate dose gradient prediction across the full range of lesion sizes.

Materials/Methods: Benchmark Gamma Knife (GK) plans optimized for maximal dose conformality were generated for 461 patients (1,097 targets). Our deep learning framework consisting of two 3D U-Net models with attention mechanisms was trained: a small target model (for volume <1 cc, 756 targets, median 0.27 cc [range 0.002-1.0], 64×64×64 mm³ VOI, 5-fold cross-validation ensemble) and a large target model (for volume =1 cc, 341 targets, median 2.9 cc [range 1.0-75.3], 128×128×128 mm³ VOI). A custom loss function prioritized accuracy in the dose falloff region, i.e., the 1-cm ring around the target volume. Performance was evaluated on an independent test set of 137 patients with 209 targets (103 small, 106 large) and compared against two single-model baselines using paired Wilcoxon signed-rank tests.

Results: On 209 independent external test targets, the small target model (volume < 1 cc, n=103) achieved a mean absolute error (MAE) of 0.43±0.15 Gy in the dose falloff region, V50% Dice of 0.866, and |V12Gy error| of 0.09±0.13 cc. The large target model (volume = 1 cc, n=106) achieved an MAE of 0.35±0.17 Gy in the dose falloff region, V50% Dice of 0.909, and |V12Gy error| of 0.72±1.57 cc. Compared to single-model baselines, the dual-model approach achieved 30% lower intra-target MAE and significantly better Dice, and V12Gy accuracy overall (all p<0.001). The large VOI baseline model performed 26% worse in falloff MAE (p<0.001), while the small VOI baseline model showed equivalent falloff accuracy but significantly worse intra-target prediction.

Conclusion: The dual-model deep learning framework accurately predicted patient-specific benchmark dose distributions for brain SRS. It achieved statistically significant improvements relative to single-model alternatives, enabling reliable cross-platform plan-quality evaluation and direct utility for routine clinical SRS quality assurance for all lesion sizes.

*A common evaluation grid was achieved by using a symmetric resampling scheme for the baseline models. Interpolation artifacts and information loss may modestly affect measured performance.
Metric

*Small VOI model (64-mm Baseline)

*Large VOI model (128-mm Baseline)

2-MODEL (Ours)

Falloff MAE (Gy)

0.39±0.17

0.53±0.21

0.39±0.17

Inside PTV MAE (Gy)

3.78±2.15

3.97±2.55

2.65±1.91

Correlation

0.960±0.031

0.945±0.043

0.964±0.031

Dice (50% Rx)

0.881±0.122

0.858±0.122

0.888±0.121

|V12Gy Error| (cc)

0.43±1.29

0.75±2.93

0.41±1.16