arXiv:2603.00535cs.CV2026-03

用检索增强方法提升无配对CBCT转CT的精度与稳定性。

RAFM: Retrieval-Augmented Flow Matching for Unpaired CBCT-to-CT Translation

  • 通过检索构建伪配对,改进无配对流匹配的训练稳定性。
  • 在真实无配对条件下,各项指标均优于现有方法。
  • 适合需要高精度sCT生成的放疗影像处理场景。

锥形束CT(CBCT)在放疗中常用,但存在严重伪影和不可靠的亨氏单位(HU)值,限制其直接用于剂量计算。从CBCT生成合成CT(sCT)至关重要,但因时间差、解剖变异和配准误差,成对的CBCT-CT数据常不可靠或缺失。本文将校正流(RF)引入无配对CBCT-to-CT转换,虽理论支持分布级耦合与确定性传输,但在小样本和有限批大小下实际效果未被充分探索。直接使用随机或批内伪配对会导致语义不匹配的监督信号,引发训练不稳定。为此,提出检索增强流匹配(RAFM),利用冻结的DINOv3编码器和全局CT记忆库构建检索引导的伪配对,提升耦合质量并稳定训练。在SynthRAD2023数据集上,采用严格主体级真无配对协议的实验表明,RAFM在FID、MAE、SSIM、PSNR和SegScore上均优于现有方法。代码已开源。

原文摘要 · Abstract (English)

Cone-beam CT (CBCT) is routinely acquired in radiotherapy but suffers from severe artifacts and unreliable Hounsfield Unit (HU) values, limiting its direct use for dose calculation. Synthetic CT (sCT) generation from CBCT is therefore an important task, yet paired CBCT--CT data are often unavailable or unreliable due to temporal gaps, anatomical variation, and registration errors. In this work, we introduce rectified flow (RF) into unpaired CBCT-to-CT translation in medical imaging. Although RF is theoretically compatible with unpaired learning through distribution-level coupling and deterministic transport, its practical effectiveness under small medical datasets and limited batch sizes remains underexplored. Direct application with random or batch-local pseudo pairing can produce unstable supervision due to semantically mismatched endpoint samples. To address this challenge, we propose Retrieval-Augmented Flow Matching (RAFM), which adapts RF to the medical setting by constructing retrieval-guided pseudo pairs using a frozen DINOv3 encoder and a global CT memory bank. This strategy improves empirical coupling quality and stabilizes unpaired flow-based training. Experiments on SynthRAD2023 under a strict subject-level true-unpaired protocol show that RAFM outperforms existing methods across FID, MAE, SSIM, PSNR, and SegScore. The code is available at https://github.com/HiLab-git/RAFM.git.

医学图像无配对翻译流匹配sCT生成

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