physics.med-phSep 4, 2026

BEAM3R: Beam's-eye-view architecture with Mamba-3 for implicit dose reconstruction

Authors: Chen Cheng, Michael Ferraro, James Grover, David E J Waddington, Emily Hewson

Organizations: Image X Institute, The University of Sydney, Sydney, Australia

Abstract

To enable accurate and rapid photon control point and proton beamlet dose calculation in the DoseRAD2026 challenge, we present BEAM3R, a dose estimation framework operating in beam's-eye-view (BEV). Our core innovation combines a Mamba-3 state-space depth-sequence core with physics-based transport conditioning to model long-range depth transport without expensive 3D convolutions. BEAM3R shares a 2D CNN encoder-decoder architecture for photon and proton dose tasks, processing per-plane BEV slices. Proton beamlets are conditioned on water equivalent thickness and remaining range, encoding the parameters determining Bragg peak position. Photon models use a bidirectional Mamba-3 core to capture dose contributions from materials downstream of the calculation point, while the proton model uses a forward core with learned energy-prefix tokens and a Bragg-peak refinement module. To reduce interpolation artifacts and support high spatial resolution, we introduce axial grid alignment of BEV lattices with CT slices and an implicit super-resolution representation via sub-pixel phase packing, evaluated by a differentiable Triton-accelerated resampler that reconstructs packed cubic B-spline coefficients directly in CT space. For MRI-based tasks, synthetic CTs (sCT) are generated by a patch-based conditional GAN with a SwinUNETR backbone. On the preliminary DoseRAD2026 test set, CT-to-photon and CT-to-proton models achieved 1%/1 mm local gamma pass rates of 96.8% and 96.0%, with stratified plan-level MAEs of 0.0041 and 0.0079. Substituting sCT reduced gamma pass rates to 89.7% for photon and 75.4% proton plan level doses, with stratified plan-level MAEs of 0.0093 and 0.0336. Standardised runtimes were 23.4 s and 18.4 s for CT-to-photon and CT-to-proton prediction, increasing to 39.7 s and 42.8 s for the corresponding MRI-based pipelines.

Figures & tables

Explore similar work

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

    Sep 1, 2026Lukas Zimmermann, Hermann Fuchs, Attila Simkó +1HealthcareRadiotherapy Dose Prediction

  2. DoseRAD2026 Challenge dataset: AI accelerated photon and proton dose calculation for radiotherapy

    Apr 14, 2026Fan Xiao, Nikolaos Delopoulos, Niklas Wahl +13Radiotherapy Dose PredictionMedical Image Benchmarks

  3. DoseBridge: Denoising Diffusion Bridge Model for Dose Prediction in Lung Intensity-Modulated Proton Therapy

    Aug 10, 2026Zerun Zhang, Xiaoda Cong, Xiangkun Xu +2HealthcareRadiotherapy Dose Prediction