AI End-to-End Radiation Treatment Planning Under One Second

Simon Arberet, Riqiang Gao, Martin Kraus, Florin C. Ghesu, Wilko Verbakel +7 more
3/6/2026
eess.IVcs.AIcs.LG

No confident code implementation found for this paper.

Abstract

Artificial intelligence-based radiation therapy (RT) planning has the potential to reduce planning time and inter-planner variability, improving efficiency and consistency in clinical workflows. Most existing automated approaches rely on multiple dose evaluations and corrections, resulting in plan generation times of several minutes. We introduce AIRT (Artificial Intelligence-based Radiotherapy), an end-to-end deep-learning framework that directly infers deliverable treatment plans from CT images and structure contours. AIRT generates single-arc VMAT prostate plans, from imaging and anatomical inputs to leaf sequencing, in under one second on a single Nvidia A100 GPU. The framework includes a differentiable dose feedback, an adversarial fluence map shaping, and a plan generation augmentation to improve plan quality and robustness. The model was trained on more than 10,000 intact prostate cases. Non-inferiority to RapidPlan Eclipse was demonstrated across target coverage and OAR sparing metrics. Target homogeneity (HI = 0.10 $\pm$ 0.01) and OAR sparing were similar to reference plans when evaluated using AcurosXB. These results represent a significant step toward ultra-fast standardized RT planning and a streamlined clinical workflow.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{arberet2026endtoend,
  title  = {AI End-to-End Radiation Treatment Planning Under One Second},
  author = {Simon Arberet and Riqiang Gao and Martin Kraus and Florin C. Ghesu and Wilko Verbakel and Mamadou Diallo and Anthony Magliari and Venkatesan Karuppusamy and Sushil Beriwal and REQUITE Consortium and Ali Kamen and Dorin Comaniciu},
  year   = {2026},
  eprint = {2603.06338},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.06338v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.06338.svg)](https://codeofpaper.com/paper/2603.06338)

Discussion