Fast and Sensitive Dose Prediction
Predicting treatment dose quickly while preserving the effects of clinically meaningful contour variation
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.
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.
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.
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
- Segmentation robustness helps ensure that the anatomical inputs and learned representations remain dependable under deployment shifts.
- Personalized contour review and correction uses dose prediction to distinguish harmless variation from alternatives that materially change treatment.
Dose prediction is therefore both a major modeling problem in its own right and the clinical bridge connecting segmentation output to radiotherapy decisions.