Fast and Sensitive Dose Prediction

Predicting treatment dose quickly while preserving the effects of clinically meaningful contour variation

Dose Prediction Model Sensitivity Treatment Planning
Watch the demos

The Research Question

Radiotherapy planning depends on accurate three-dimensional dose calculation, but repeated treatment-planning calculations are too slow for large evaluation studies, rapid comparison of contour alternatives, or interactive review. This axis asks whether learned models can predict dose fast enough for those settings while remaining accurate, robust, and above all sensitive to clinically meaningful contour changes.

Sensitivity is distinct from overall prediction accuracy. A model can reproduce the broad dose distribution yet smooth away the local effect of the contour variation that a quality-assurance workflow needs to detect.

Demos: dose prediction in motion

A treatment plan says how much radiation lands where. Producing one properly takes a planning system and hours of work, which is why nobody re-plans a treatment just to check whether an organ was outlined well. These clips come from a model that predicts the same dose distribution in about fifteen seconds, close enough to be useful while the contours are still editable.

In every clip the planning CT is the grey background, the heat map is dose in Gy, and the outlined structures are the tumour target and the organs at risk being protected.

Use the fullscreen control on any clip to read the per-panel numbers.

Does the prediction move when a contour moves?

This is the question the whole idea rests on. Ten plausible outlines of the left optic nerve are drawn in turn. The left panel is the plan a treatment planning system re-optimised for each one, and the right is the model’s prediction for the same outline.

Sensitivity to left optic nerve contour variation Each frame is one redrawing of the nerve, solid, with the original dashed beside it. Under each panel is the mean dose that outline receives; the subtitle gives the shift from the original for the plan and for the prediction. Across the ten outlines those two shifts correlate at 0.926, while the Dice overlap between outlines correlates at only −0.471 — geometry alone does not tell you which edits matter. Full details on the project page.

What happens on anatomy the model never saw

Targets that wrap around a structure, or that break into several separate lesions, were absent from the training set. Each clip shows the treatment plan, the original model, and the same model retrained with six such cases.

A concave target In this case the dose score improves from 0.67 to 0.60 Gy after retraining. The failure was in the training distribution rather than the architecture, and six extra cases recover most of the conformity that was lost.
A target split into several lesions The same three panels for the other unfamiliar shape. Curating the training set for shape coverage turns out to be cheaper than collecting more data in general.

How close the prediction gets

Sweeping through a case from the base of the skull upwards, with the treatment plan on the left and the prediction on the right.

The hardest case in the test set The largest target of the twenty test cases, with seven organs at risk receiving more than 20 Gy, and the worst dose score of the set. Even here the predicted distribution tracks the plan closely enough to reason with. Three more sweeps are on the project page.

These clips come from the DeepDoseSens study. Its full write-up, covering both the ISBI 2023 conference paper and the Cancers 2023 journal extension, with the model, the per-organ results and the code, lives on its own page: How sensitive is a dose prediction model to the contours you feed it?


Connection to the PhD Thesis

Part Two of my PhD thesis established the technical foundation for this axis. It evaluated dose-prediction accuracy, sensitivity to inter-expert contour variation, behavior on unusual cases, and the translation of local sensitivity into ASTRA’s review maps.

The completed work established that:

  • learned dose prediction can operate on a timescale suitable for rapid evaluation;
  • sensitivity to contour changes must be tested separately from global dose accuracy;
  • out-of-distribution behavior requires deliberate validation and mitigation; and
  • local dose sensitivity can guide attention to consequential contour regions.

Each study keeps its own write-up next to its code, rather than being restated here:

  • How sensitive is a dose prediction model to the contours you feed it?, IEEE ISBI 2023 and its Cancers 2023 journal extension. The cascaded 3D U-Net, the sensitivity experiment, the out-of-distribution stress tests, and every table recomputed from the archived volumes.
  • ASTRA, IEEE EMBC 2023. Turning local dose sensitivity into review maps that point a clinician at the contour regions worth checking.

The full list, with links, is on the publications page.

Current Directions

I am continuing this axis through:

  • architectures that preserve fine-grained anatomical sensitivity while modeling global treatment context;
  • robust prediction across tumour sites, institutions, planning systems, and treatment techniques;
  • calibrated uncertainty for unusual anatomies and out-of-distribution cases;
  • faster feedback for iterative or adaptive treatment planning;
  • evaluation of contour alternatives by predicted dose and treatment objectives; and
  • integration of dose prediction with personalized contour proposals and interactive correction.

Relationship to the Other Axes

Dose prediction is therefore both a major modeling problem in its own right and the clinical bridge connecting segmentation output to radiotherapy decisions.

References