用改进的配准方法提升MRI/CBCT生成合成CT的解剖一致性。
Why Registration Quality Matters: Enhancing sCT Synthesis with IMPACT-Based Registration
- 基于IMPACT特征相似性进行图像配准,提升解剖对齐精度。
- 在局部测试集上,新配准方法使合成CT的MAE降低、结构更真实。
- 揭示了配准偏差对模型评估的影响,适合关注泛化性的研究者。
我们在SynthRAD2025挑战赛(任务1和2)中,使用KonfAI框架构建统一的sCT生成流程,从MRI与CBCT生成合成CT。模型采用2.5D U-Net++结构,编码器为ResNet-34,跨解剖区域联合训练并按区域微调。损失函数结合像素级L1损失与基于SAM和TotalSegmentator的感知损失IMPACT-Synth,以增强结构保真度。训练采用AdamW优化器(初始学习率0.001,每25k步减半),输入为基于体部掩码的块状归一化数据(MRI为320×320,CBCT为256×256),仅使用随机翻转作为增强。无后处理。最终预测采用测试时增强与五折集成。最佳模型依据验证集MAE选择。对比两种配准策略:(i) Elastix+互信息,与挑战流水线一致;(ii) IMPACT,基于预训练分割网络的特征相似性。在本地测试集上,IMPACT配准获得更准确且解剖一致的配准,带来更低的MAE与更真实的结构。但在公开验证集上,使用Elastix对齐数据的模型得分更高,反映出评估流水线对特定配准策略的偏好。这表明配准误差会传播至监督学习中,影响训练与评估,可能因牺牲解剖真实性而虚高性能指标。推广使用解剖一致的IMPACT可缓解此偏差,助力开发更鲁棒、泛化的sCT合成模型。
原文摘要 · Abstract (English)
We participated in the SynthRAD2025 challenge (Tasks 1 and 2) with a unified pipeline for synthetic CT (sCT) generation from MRI and CBCT, implemented using the KonfAI framework. Our model is a 2.5D U-Net++ with a ResNet-34 encoder, trained jointly across anatomical regions and fine-tuned per region. The loss function combined pixel-wise L1 loss with IMPACT-Synth, a perceptual loss derived from SAM and TotalSegmentator to enhance structural fidelity. Training was performed using AdamW (initial learning rate = 0.001, halved every 25k steps) on patch-based, normalized, body-masked inputs (320x320 for MRI, 256x256 for CBCT), with random flipping as the only augmentation. No post-processing was applied. Final predictions leveraged test-time augmentation and five-fold ensembling. The best model was selected based on validation MAE. Two registration strategies were evaluated: (i) Elastix with mutual information, consistent with the challenge pipeline, and (ii) IMPACT, a feature-based similarity metric leveraging pretrained segmentation networks. On the local test sets, IMPACT-based registration achieved more accurate and anatomically consistent alignments than mutual-information-based registration, resulting in improved sCT synthesis with lower MAE and more realistic anatomical structures. On the public validation set, however, models trained with Elastix-aligned data achieved higher scores, reflecting a registration bias favoring alignment strategies consistent with the evaluation pipeline. This highlights how registration errors can propagate into supervised learning, influencing both training and evaluation, and potentially inflating performance metrics at the expense of anatomical fidelity. By promoting anatomically consistent alignment, IMPACT helps mitigate this bias and supports the development of more robust and generalizable sCT synthesis models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。