Any2Any 3D Diffusion Models with Knowledge Transfer: A Radiotherapy Planning Study
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
- 2025MAISI-v2: Accelerated 3D High-Resolution Medical Image Synthesis with Rectified Flow and Region-specific Contrastive Loss7 Aug
- 2025Text2CT: Towards 3D CT Volume Generation from Free-text Descriptions Using Diffusion Model7 May
- 2024MAISI: Medical AI for Synthetic Imaging13 Sep
- 2023DiffDP: Radiotherapy Dose Prediction via a Diffusion Model19 Jul
- 2023DoseDiff: Distance-aware Diffusion Model for Dose Prediction in Radiotherapy28 Jun
Other references · 46
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
34 more
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2023T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Modelscited by 1,351
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
- 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
- 2023MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generationcited by 536
- 2023BiomedCLIP: a multimodal biomedical foundation model pretrained from fifteen million scientific image-text pairscited by 520
- 2023Medical SAM Adapter: Adapting Segment Anything Model for Medical Image Segmentationcited by 490
- 2024OmniGen: Unified Image Generationcited by 394
- 2023Human Preference Score: Better Aligning Text-to-Image Models with Human Preferencecited by 287
- 2019ResUNet-a: a deep learning framework for semantic segmentation of remotely sensed datacited by 261
- 2023Using Human Feedback to Fine-tune Diffusion Models without Any Reward Modelcited by 252
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2023PMC-CLIP: Contrastive Language-Image Pre-training using Biomedical Documentscited by 212
- 2023Any-to-Any Generation via Composable Diffusioncited by 204
- 2022Versatile Diffusion: Text, Images and Variations All in One Diffusion Modelcited by 199
- 2023One Transformer Fits All Distributions in Multi-Modal Diffusion at Scalecited by 187
- 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 162
- 2024Learning Multi-dimensional Human Preference for Text-to-Image Generationcited by 106
- 2023Foundational Models in Medical Imaging: A Comprehensive Survey and Future Visioncited by 76
- 2024Adapting Visual-Language Models for Generalizable Anomaly Detection in Medical Imagescited by 69
- 2024OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flowscited by 46
- 2024PixWizard: Versatile Image-to-Image Visual Assistant with Open-Language Instructionscited by 31
- 2020OpenKBP: The open-access knowledge-based planning grand challengecited by 30
- 2018Three-Dimensional Radiotherapy Dose Prediction on Head and Neck Cancer Patients with a Hierarchically Densely Connected U-net Deep Learning Architecturecited by 18
- 2018Three-Dimensional Dose Prediction for Lung IMRT Patients with Deep Neural Networks: Robust Learning from Heterogeneous Beam Configurationscited by 12
- 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
- 2024Multi-Agent Reinforcement Learning Meets Leaf Sequencing in Radiotherapycited by 6
- 2022Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Functioncited by 5
- 2025Jodi: Unification of Visual Generation and Understanding via Joint Modelingcited by 5
- 2021DeepDoseNet: A Deep Learning model for 3D Dose Prediction in Radiation Therapycited by 3
- 2025Automating RT Planning at Scale: High Quality Data For AI Trainingcited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2023Diffusion-based Radiotherapy Dose Prediction Guided by Inter-slice Aware Structure Encoding
- 2021LENAS: Learning-based Neural Architecture Search and Ensemble for 3D Radiotherapy Dose Prediction
- 2023SP-DiffDose: A Conditional Diffusion Model for Radiation Dose Prediction Based on Multi-Scale Fusion of Anatomical Structures, Guided by SwinTransformer and Projector
- 2025Conditional Diffusion Model with Anatomical-Dose Dual Constraints for End-to-End Multi-Tumor Dose Prediction
- 2026GenDiff: A Dose and Anatomy Aware Diffusion Model with Structural Prior Refinement for Low-Dose CT Reconstruction and Generalization
- 2026Optimizing 3D Diffusion Models for Medical Imaging via Multi-Scale Reward Learning
- 2025Patient-specific AI for generation of 3D dosimetry imaging from two 2D-planar measurements
- 2026DoseBridge: Denoising Diffusion Bridge Model for Dose Prediction in Lung Intensity-Modulated Proton Therapy