Neural Fast Monte Carlo for PBS Proton Therapy
A differentiable pencil-beam engine with a zero-init residual corrector, on the DoseRAD2026 open benchmark (~81,000 Geant4 beamlets, 85 energies).
Monte Carlo is the reference dose calculation in PBS proton therapy, but too slow for a planning loop. This project builds a differentiable pencil-beam engine (85-energy LUTs, water-equivalent-depth ray-tracing, two-stage MC calibration) and an xLSTM residual corrector with zero-init heads — at step zero its contribution is exactly the physics, so the neural network can only improve on it, never break it.
On 12 held-out patients (277 beamlets, three terrain types), the analytical engine alone reaches 93.3 % (water), 94.7 % (moderate) and 89.4 % (hard) median γ 1%/1mm pass rate at the 10% cut, with ~0.3 mm R80 error — an honest FAIL against the formal 98.8 % gate, with the engine's nuclear-tail limitations documented and deferred to the corrector. Development is test-driven with fail-closed provenance; the results-first repository documents the method and current numbers.