Search papers, labs, and topics across Lattice.
This paper introduces PyDoseRT Proton, a GPU-accelerated pencil-beam dose calculation engine that integrates a convolutional residual-correction network to enhance proton dose prediction accuracy. By employing a double-Gaussian kernel calibrated through a two-stage fitting process, the system achieves Monte Carlo-level precision while maintaining computational efficiency. The results demonstrate impressive performance metrics, including a mean absolute error (MAE) of 0.0066 for beamlet predictions and a gamma pass rate of 98.30%, indicating its potential for clinical application in proton therapy.
Achieving Monte Carlo-level accuracy in proton dose calculations while significantly reducing computational time could revolutionize treatment planning in radiation therapy.
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.