mlsanityData to 14 Oct 2026

Any2Any 3D Diffusion Models with Knowledge Transfer: A Radiotherapy Planning Study

arXiv 2605.09622 ↗First posted 10 May 2026cs.CV cs.AIcited by 1

On the map: Diffusion models → Medical imaging and segmentation

Abstract (arXiv)

Voxel-wise dose prediction is a critical yet challenging task in practical radiotherapy (RT) planning, as bespoke models trained from scratch often struggle to generalize across diverse clinical settings. Meanwhile, generative models trained on billion-scale datasets from vision domains have achieved impressive performance. Herein, we propose DiffKT3D, a unified Any2Any 3D diffusion framework that leverages prior knowledge from pretrained video diffusion models for efficient and clinically meaningful dose prediction. To enable flexible conditioning across multiple clinical modalities (CT, anatomical structures, body, beam settings, etc.), we introduce an Any2Any conditional paradigm utilizing modality-specific embeddings without cross-attention overhead. Further, we design a novel reinforcement learning (RL) post-training mechanism guided by a clinically-informed Scorecard explicitly tailored to institutional treatment preferences. Compared with winner of GDP-HMM challenge, DiffKT3D sets a new state-of-the-art in dose prediction by reducing voxel-level MAE from 2.07 to 1.93. In addition, DiffKT3D achieves superior image quality and preference match. These results demonstrate that transferring diffusion priors via modality-aware conditioning and clinically aligned RL post-training can provide a robust and generalizable solution for RT planning across various clinical scenarios.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 51 of its 78 references that are in the map. First posted on arXiv 10 May 2026.

Builds on

51 of 78 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025MAISI-v2: Accelerated 3D High-Resolution Medical Image Synthesis with Rectified Flow and Region-specific Contrastive Loss7 Aug
  2. 2025Text2CT: Towards 3D CT Volume Generation from Free-text Descriptions Using Diffusion Model7 May
  3. 2024MAISI: Medical AI for Synthetic Imaging13 Sep
  4. 2023DiffDP: Radiotherapy Dose Prediction via a Diffusion Model19 Jul
  5. 2023DoseDiff: Distance-aware Diffusion Model for Dose Prediction in Radiotherapy28 Jun

Other references · 46

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  3. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  4. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  5. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  6. 2022Training language models to follow instructions with human feedbackcited by 15,971
  7. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  8. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  9. 2022Scalable Diffusion Models with Transformerscited by 6,054
  10. 2022Flow Matching for Generative Modelingcited by 5,702
  11. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
  12. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
34 more
  1. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
  2. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
  3. 2023T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Modelscited by 1,351
  4. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
  5. 2023Training Diffusion Models with Reinforcement Learningcited by 982
  6. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
  7. 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
  8. 2023MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generationcited by 536
  9. 2023BiomedCLIP: a multimodal biomedical foundation model pretrained from fifteen million scientific image-text pairscited by 520
  10. 2023Medical SAM Adapter: Adapting Segment Anything Model for Medical Image Segmentationcited by 490
  11. 2024OmniGen: Unified Image Generationcited by 394
  12. 2023Human Preference Score: Better Aligning Text-to-Image Models with Human Preferencecited by 287
  13. 2019ResUNet-a: a deep learning framework for semantic segmentation of remotely sensed datacited by 261
  14. 2023Using Human Feedback to Fine-tune Diffusion Models without Any Reward Modelcited by 252
  15. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
  16. 2023PMC-CLIP: Contrastive Language-Image Pre-training using Biomedical Documentscited by 212
  17. 2023Any-to-Any Generation via Composable Diffusioncited by 204
  18. 2022Versatile Diffusion: Text, Images and Variations All in One Diffusion Modelcited by 199
  19. 2023One Transformer Fits All Distributions in Multi-Modal Diffusion at Scalecited by 187
  20. 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 162
  21. 2024Learning Multi-dimensional Human Preference for Text-to-Image Generationcited by 106
  22. 2023Foundational Models in Medical Imaging: A Comprehensive Survey and Future Visioncited by 76
  23. 2024Adapting Visual-Language Models for Generalizable Anomaly Detection in Medical Imagescited by 69
  24. 2024OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flowscited by 46
  25. 2024PixWizard: Versatile Image-to-Image Visual Assistant with Open-Language Instructionscited by 31
  26. 2020OpenKBP: The open-access knowledge-based planning grand challengecited by 30
  27. 2018Three-Dimensional Radiotherapy Dose Prediction on Head and Neck Cancer Patients with a Hierarchically Densely Connected U-net Deep Learning Architecturecited by 18
  28. 2018Three-Dimensional Dose Prediction for Lung IMRT Patients with Deep Neural Networks: Robust Learning from Heterogeneous Beam Configurationscited by 12
  29. 2019Incorporating human and learned domain knowledge into training deep neural networks: A differentiable dose volume histogram and adversarial inspired framework for generating Pareto optimal dose distributions in radiation therapycited by 10
  30. 2024Multi-Agent Reinforcement Learning Meets Leaf Sequencing in Radiotherapycited by 6
  31. 2022Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Functioncited by 5
  32. 2025Jodi: Unification of Visual Generation and Understanding via Joint Modelingcited by 5
  33. 2021DeepDoseNet: A Deep Learning model for 3D Dose Prediction in Radiation Therapycited by 3
  34. 2025Automating RT Planning at Scale: High Quality Data For AI Trainingcited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2023Diffusion-based Radiotherapy Dose Prediction Guided by Inter-slice Aware Structure Encoding
  2. 2021LENAS: Learning-based Neural Architecture Search and Ensemble for 3D Radiotherapy Dose Prediction
  3. 2023SP-DiffDose: A Conditional Diffusion Model for Radiation Dose Prediction Based on Multi-Scale Fusion of Anatomical Structures, Guided by SwinTransformer and Projector
  4. 2025Conditional Diffusion Model with Anatomical-Dose Dual Constraints for End-to-End Multi-Tumor Dose Prediction
  5. 2026GenDiff: A Dose and Anatomy Aware Diffusion Model with Structural Prior Refinement for Low-Dose CT Reconstruction and Generalization
  6. 2026Optimizing 3D Diffusion Models for Medical Imaging via Multi-Scale Reward Learning
  7. 2025Patient-specific AI for generation of 3D dosimetry imaging from two 2D-planar measurements
  8. 2026DoseBridge: Denoising Diffusion Bridge Model for Dose Prediction in Lung Intensity-Modulated Proton Therapy