An illustrated tour of

Differentiable
Photon Transport

Chris von Csefalvay

Go from zero to actual differentiable photon transport in CUDA/Warp without losing your sanity to physics, your livelihood to GPU rental and your will to live to maths.

Volume I

Foundations

We introduce differentiable photon transport, and try to do so in the most painless but educational way possible: by constructing, step by step, a differentiable C-arm.

  1. 01IntroductionDefines the questions and boundaries of differentiable photon transport, beginning with a C-arm and an image, then asking which quantities we want to recover and what a useful derivative would tell us about them.
  2. 02TransmissionEstablishes the transmission model used throughout the volume, following attenuation along a single path and keeping the parameters, units and assumptions visible as that physical account becomes an executable model.
  3. 03Frames and motionFixes the coordinate frames and motion conventions for later chapters, giving the source, detector and object a shared account of where everything is so that a change in pose means the same thing in the geometry and the code.
  4. 04Volumes and raysConnects volumetric representations with line-integral operators, tracing a ray through a sampled attenuation field and examining how interpolation, boundaries and numerical choices shape the projection we ask the GPU to compute.
  5. 05Projection gradientsDevelops derivatives of the projection operation, asking how an image changes when its inputs move and how independent numerical checks can distinguish a useful gradient from one that merely looks plausible.
  6. 06Pose recoveryStudies pose recovery with differentiable image formation, working backwards from an image to geometry while considering the choice of image comparison, the initial estimate and the ambiguities that a single view cannot resolve.
  7. 07Acquisition mismatchExamines acquisition changes that can confound geometric inference, looking at why two X-rays may differ even when nothing has moved and what a recovery method needs to account for before it starts adjusting the object's pose.
  8. 08Spectra and detectorsExtends the model to spectra and detector response, moving beyond a single photon energy to consider how the source spectrum and the detector's sensitivity affect the predicted image and the parameters we hope to infer from it.
  9. 09Monte Carlo transportIntroduces scattering and stochastic transport estimators, following photons beyond the straight path from source to detector and considering how sampling choices, variance and computational cost enter the task of estimating an image.
  10. 10Transport gradientsTreats derivatives through the full transport process, considering what changes when paths are sampled, which dependencies must be included and how to assess correctness without mistaking sampling noise for a modelling or implementation error.
  11. 112D–3D registrationAligns a known volume with measured radiographs, extending pose recovery to calibrated acquisitions, multiple views and independent evaluation when the anatomy and image formation do not agree perfectly.
  12. 12Volume reconstructionRecovers a three-dimensional attenuation field from projections, developing volume optimisation and regularisation before examining sparse views, limited angles and material decomposition with spectral measurements.
  13. 13Acquisition designChooses measurements that improve registration or reconstruction, using task-specific uncertainty and error objectives to select viewpoints and allocate exposure within geometric and acquisition constraints.
  14. AX-ray physicsCollects the X-ray concepts required by the main text, keeping physical quantities, imaging terminology and useful conventions within reach for readers who know their way around code but have no reason to know their way around a radiology department.