arXiv subjects

Gerd Heilemann

Publications and source records attributed to Gerd Heilemann.

4 recordsLinked to original sources

arXiv2026

PyDoseRT Proton: A GPU Pencil-Beam Engine with a Convolutional Residual-Correction Network for Fast Proton Dose Calculation

Architecture category. Hybrid method: a physics-based analytical pencil-beam (PB) dose engine followed by a 3-D convolutional residual-correction network (RepVGG-U-Net). We addressed the DoseRAD2026 proton dose-prediction task with PyDoseRT Proton, a GPU-accelerated engine implemented in PyTorch and augmented by a learned residual toward Monte Carlo (MC) accuracy. A double-Gaussian PB kernel was calibrated to GATE/Geant4 integrated depth doses in water in two stages: a classical per-energy curve fit, then a gradient-based fit of the full 3-D dose through the PyTorch physics engine as it retains a differentiable execution path for gradient-based optimization of dose-dependent objectives. The engine computes each beamlet on a beam's-eye-view (BEV) lattice with variance-preserving Gaussian splitting, an analytic nuclear halo, and a Fermi-Eyges heterogeneity term, then rotates the result into the patient frame. Additionally, a compact residual U-Net predicts an additive correction in BEV space. It is conditioned on voxelwise material-label embeddings, a discrete energy embedding and spot size. The same model was used for all anatomical sites (thoracic and abdominal). It was trained with a patient-space L1 objective emphasizing the scored high-dose region and multi-scale BEV deep supervision. The submitted CT configuration obtained preliminary-test beamlet MAE 0.0066, image-z IDD distance 0.0025, plan MAE 0.0049, 98.30\% gamma pass rate (1\%/1 mm), and DVH error 0.460.

physics.med-ph↗

arXiv2026

PyDoseRT Photon: Physics-Guided Pencil-Beam Dose Calculation with Neural Priors and Residual Correction for CT and MRI

We present a hybrid, physics-based analytical pencil-beam (PB) dose engine, augmented by two small frozen neural physics priors, followed by a 3-D convolutional residual-correction network (U-Net). We address the DoseRAD2026 (https://doserad2026.grand-challenge.org/) photon dose-prediction task with PyDoseRT Photon, a GPU PB engine implemented in PyTorch and corrected toward Monte Carlo (MC) accuracy in three learned stages of decreasing physical specificity. The engine reproduces the challenge's head-less MC source exactly where it can and models the patient with a beam-quality-indexed pencil kernel evaluated at the field's fluence-weighted radiological depth, and TERMA source scaling at the interaction site. Two tiny neural priors are trained through the frozen engine and then frozen themselves: a 39k-parameter 2-D fluence correction and a 48-parameter lateral heterogeneity correction mixing mass-conserving Gaussian redistribution operators. A compact 3-D U-Net (1.36M parameters) with a sequential refinement branch then predicts, per control point (CP) in the beam's-eye-view (BEV) frame, a bounded multiplicative gain and additive residual from seven channels. All learned stages are zero-initialized, so training starts from the analytical solution. For MRI, an nnU-Net regression model synthesizes a CT that enters the identical pipeline, with consecutively, the same trained corrector as the CT track. Design choices were driven by the challenge ranking, in which runtime carries double weight. The submitted method evaluated on our local CT and MR validation dataset achieved CP MAE 0.0086 and 0.0098, IDD distance 0.0011 and 0.0013, plan MAE 0.0024 and 0.0047, gamma pass rate (1%/1mm) 99.12% and 96.95%, with runtimes of 46s and 49s, respectively.

physics.med-ph↗

arXiv2025

A physics-informed, plug-and-play dose engine for gradient-based radiotherapy treatment planning

Radiotherapy treatment planning remains a time-intensive iterative process requiring expert intervention in commercial treatment planning system (TPS). While machine learning approaches have demonstrated promise, most remain depedent on TPS-based dose calculation or surrogate dose models, preventing direct optimization of deliverable treatment plan parameters. We propose PyDoseRT (PDRT), a physics-informed, GPU-accelerated dose engine implemented in PyTorch that computes dose distributions directly from treatment delivery parameters (i.e., MLC leaf positions, jaw positions, gantry angles, and monitor units). The engine preserves gradient information throughout the dose computation pipeline, enabling gradient-based optimization of hardware-constrained treatment plans without the reliance on a commercial TPS. PDRT was evaluated on 19 and 162 clinical VMAT prostate cancer plans from two hospitals (with different treatment machines). When recalculating clinical plans, PDRT achieved high 3D gamma pass rates (mean 96.8% for 2%/2 mm and 98.9% for 3%/3 mm, depending on cohort). All optimized plans converged to clinically acceptable solutions and passed deliverability verification when imported into a commercial TPS. This physics-informed framework eliminates TPS dependency for radiotherapy optimization research by enabling gradient-based planning while ensuring that delivery parameters remain in the machine-feasible range. The gradient-enabled dose engine allows exploration of novel optimization strategies and objective functions while maintaining clinical validity. The proposed approach provides a research platform for investigating real-time adaptive radiotherapy concepts, automated planning workflows, and TPS-independent optimization strategies, and democratizing radiotherapy research, by exposing gradient-enabled, hardware-aware, open-source dose computation.

physics.med-ph↗

arXiv2022

OpenKBP-Opt: An international and reproducible evaluation of 76 knowledge-based planning pipelines

We establish an open framework for developing plan optimization models for knowledge-based planning (KBP) in radiotherapy. Our framework includes reference plans for 100 patients with head-and-neck cancer and high-quality dose predictions from 19 KBP models that were developed by different research groups during the OpenKBP Grand Challenge. The dose predictions were input to four optimization models to form 76 unique KBP pipelines that generated 7600 plans. The predictions and plans were compared to the reference plans via: dose score, which is the average mean absolute voxel-by-voxel difference in dose a model achieved; the deviation in dose-volume histogram (DVH) criterion; and the frequency of clinical planning criteria satisfaction. We also performed a theoretical investigation to justify our dose mimicking models. The range in rank order correlation of the dose score between predictions and their KBP pipelines was 0.50 to 0.62, which indicates that the quality of the predictions is generally positively correlated with the quality of the plans. Additionally, compared to the input predictions, the KBP-generated plans performed significantly better (P<0.05; one-sided Wilcoxon test) on 18 of 23 DVH criteria. Similarly, each optimization model generated plans that satisfied a higher percentage of criteria than the reference plans. Lastly, our theoretical investigation demonstrated that the dose mimicking models generated plans that are also optimal for a conventional planning model. This was the largest international effort to date for evaluating the combination of KBP prediction and optimization models. In the interest of reproducibility, our data and code is freely available at https://github.com/ababier/open-kbp-opt.

physics.med-ph↗