arXiv:2501.15610eess.IVcs.CV2025-01被引 14

让放射科医生参与训练,提升CT金属伪影去除的临床适应性

Radiologist-in-the-Loop Self-Training for Generalizable CT Metal Artifact Reduction

  • 引入医生反馈优化伪标签生成,打破固定先验限制
  • 在多个真实临床数据集上实现最优泛化性能
  • 适合医疗AI研发者与放射科医生协作改进模型

CT图像中的金属伪影会严重降低图像质量,影响诊断准确性。基于模拟数据训练的监督式金属伪影去除(MAR)方法因存在显著域差距,在真实临床CT图像上表现不佳。现有先进的半监督方法虽利用预训练网络生成伪真值以缓解此问题,但依赖固定先验导致伪真值质量和数量受限,引发确认偏差并降低临床适用性。为此,本文提出一种新型放射科医生参与的自训练框架RISE-MAR,通过将医生反馈融入半监督学习过程,持续提升伪真值的质量与数量,增强模型在真实临床数据上的泛化能力。为保障质量,设计临床质量评估模型模拟医生判读,筛选高质量伪真值用于训练;为保障数量,自训练框架迭代生成更多高质量伪真值,扩展临床数据集,进一步提升模型泛化性。在多个临床数据集上的大量实验表明,RISE-MAR在泛化性能上优于当前最先进方法,推动了MAR模型向实际应用迈进。代码已开源:https://github.com/Masaaki-75/rise-mar。

原文摘要 · Abstract (English)

Metal artifacts in computed tomography (CT) images can significantly degrade image quality and impede accurate diagnosis. Supervised metal artifact reduction (MAR) methods, trained using simulated datasets, often struggle to perform well on real clinical CT images due to a substantial domain gap. Although state-of-the-art semi-supervised methods use pseudo ground-truths generated by a prior network to mitigate this issue, their reliance on a fixed prior limits both the quality and quantity of these pseudo ground-truths, introducing confirmation bias and reducing clinical applicability. To address these limitations, we propose a novel Radiologist-In-the-loop SElf-training framework for MAR, termed RISE-MAR, which can integrate radiologists' feedback into the semi-supervised learning process, progressively improving the quality and quantity of pseudo ground-truths for enhanced generalization on real clinical CT images. For quality assurance, we introduce a clinical quality assessor model that emulates radiologist evaluations, effectively selecting high-quality pseudo ground-truths for semi-supervised training. For quantity assurance, our self-training framework iteratively generates additional high-quality pseudo ground-truths, expanding the clinical dataset and further improving model generalization. Extensive experimental results on multiple clinical datasets demonstrate the superior generalization performance of our RISE-MAR over state-of-the-art methods, advancing the development of MAR models for practical application. Code is available at https://github.com/Masaaki-75/rise-mar.

CT图像金属伪影自训练放射科医生

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