用分割引导和不确定性量化提升质子治疗中CBCT转CT的精度与可信度。
Segmentation-Guided CT Synthesis with Pixel-Wise Conformal Uncertainty Bounds
- 基于pCT分割先验引导图像生成,增强解剖结构一致性。
- 引入像素级置信区间,提供可校准的不确定性估计。
- 适合需要高可靠合成CT的放疗临床应用,尤其质子治疗。
质子治疗中的精准剂量计算依赖高质量的CT图像。虽然计划CT(pCT)用于剂量规划,但锥形束CT(CBCT)在自适应放疗(ART)中被用于生成sCT以改善剂量计算。尽管成本低且辐射暴露少,CBCT存在严重伪影和图像质量差的问题,难以满足精确剂量学要求。基于深度学习的CBCT-to-CT转换成为有前景的方法,但现有方法常引入解剖不一致且缺乏可靠的不确定性估计,限制了临床应用。为此,我们提出STF-RUE框架,包含两个关键组件:一是利用pCT提取的分割先验增强解剖一致性的分割引导转换方法(STF);二是为预测CT添加像素级共形预测区间、提供稳健可靠性指标的共形预测方法(RUE)。在两个基准数据集上,使用UNet++和Fast-DDPM进行的综合实验表明,STF-RUE显著提升了转换精度,通过专为精确剂量计算设计的软组织聚焦指标评估。此外,该方法提供了更校准的不确定性集合,增强了对合成CT的信任。通过同时解决解剖保真度与不确定性量化问题,STF-RUE标志着向更安全高效的自适应质子治疗迈出重要一步。代码已公开于https://anonymous.4open.science/r/cbct2ct_translation-B2D9/。
原文摘要 · Abstract (English)
Accurate dose calculations in proton therapy rely on high-quality CT images. While planning CTs (pCTs) serve as a reference for dosimetric planning, Cone Beam CT (CBCT) is used throughout Adaptive Radiotherapy (ART) to generate sCTs for improved dose calculations. Despite its lower cost and reduced radiation exposure advantages, CBCT suffers from severe artefacts and poor image quality, making it unsuitable for precise dosimetry. Deep learning-based CBCT-to-CT translation has emerged as a promising approach. Still, existing methods often introduce anatomical inconsistencies and lack reliable uncertainty estimates, limiting their clinical adoption. To bridge this gap, we propose STF-RUE, a novel framework integrating two key components. First, STF, a segmentation-guided CBCT-to-CT translation method that enhances anatomical consistency by leveraging segmentation priors extracted from pCTs. Second, RUE, a conformal prediction method that augments predicted CTs with pixel-wise conformal prediction intervals, providing clinicians with robust reliability indicator. Comprehensive experiments using UNet++ and Fast-DDPM on two benchmark datasets demonstrate that STF-RUE significantly improves translation accuracy, as measured by a novel soft-tissue-focused metric designed for precise dose computation. Additionally, STF-RUE provides better-calibrated uncertainty sets for synthetic CT, reinforcing trust in synthetic CTs. By addressing both anatomical fidelity and uncertainty quantification, STF-RUE marks a crucial step toward safer and more effective adaptive proton therapy. Code is available at https://anonymous.4open.science/r/cbct2ct_translation-B2D9/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。