arXiv:2508.02043cs.CV2025-08被引 1

用双约束扩散模型一键生成多肿瘤放疗计划,精度提升超30%。

Conditional Diffusion Model with Anatomical-Dose Dual Constraints for End-to-End Multi-Tumor Dose Prediction

  • 基于解剖-剂量双重约束的条件扩散模型,融合影像与治疗参数
  • 平均误差低至0.101-0.154,脊髓最大剂量误差<0.1Gy,DICE达0.927
  • 生成速度仅22秒/例,适合临床部署,首次实现端到端多瘤种预测

放疗计划制定常依赖耗时且依赖经验的试错过程,现有深度学习方法在泛化性、精度和临床适用性方面存在局限。为此,我们提出ADDiff-Dose——一种基于解剖-剂量双重约束的条件扩散模型,用于端到端多肿瘤剂量预测。该模型采用LightweightVAE3D压缩高维CT数据,并在逐步加噪与去噪框架中整合靶区与危及器官(OAR)掩码、束流参数等多模态输入,通过多头注意力机制注入条件特征,使用包含MSE、条件项和KL散度的复合损失函数,确保剂量准确性与临床约束合规性。在大规模公开数据集(2,877例)及三个外部机构队列(共450例)上评估显示,其显著优于传统基线,平均绝对误差(MAE)为0.101–0.154(对比UNet的0.316与GAN模型的0.169),DICE系数达0.927(提升6.8%),脊髓最大剂量误差控制在0.1 Gy以内。单例计划生成平均耗时仅22秒。消融实验表明,结构编码器使临床剂量约束符合率提升28.5%。据我们所知,这是首个将条件扩散模型应用于放疗剂量预测的研究,提供了一种可泛化、高效的自动化治疗规划方案,有望大幅缩短规划时间并优化临床流程。

原文摘要 · Abstract (English)

Radiotherapy treatment planning often relies on time-consuming, trial-and-error adjustments that heavily depend on the expertise of specialists, while existing deep learning methods face limitations in generalization, prediction accuracy, and clinical applicability. To tackle these challenges, we propose ADDiff-Dose, an Anatomical-Dose Dual Constraints Conditional Diffusion Model for end-to-end multi-tumor dose prediction. The model employs LightweightVAE3D to compress high-dimensional CT data and integrates multimodal inputs, including target and organ-at-risk (OAR) masks and beam parameters, within a progressive noise addition and denoising framework. It incorporates conditional features via a multi-head attention mechanism and utilizes a composite loss function combining MSE, conditional terms, and KL divergence to ensure both dosimetric accuracy and compliance with clinical constraints. Evaluation on a large-scale public dataset (2,877 cases) and three external institutional cohorts (450 cases in total) demonstrates that ADDiff-Dose significantly outperforms traditional baselines, achieving an MAE of 0.101-0.154 (compared to 0.316 for UNet and 0.169 for GAN models), a DICE coefficient of 0.927 (a 6.8% improvement), and limiting spinal cord maximum dose error to within 0.1 Gy. The average plan generation time per case is reduced to 22 seconds. Ablation studies confirm that the structural encoder enhances compliance with clinical dose constraints by 28.5%. To our knowledge, this is the first study to introduce a conditional diffusion model framework for radiotherapy dose prediction, offering a generalizable and efficient solution for automated treatment planning across diverse tumor sites, with the potential to substantially reduce planning time and improve clinical workflow efficiency.

放疗规划扩散模型多肿瘤预测医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。