用弱监督方法生成逼真前列腺弥散加权像畸变,提升自动校正效果。
Learning to Distort: Weakly-Supervised Image Quality Transfer for Prostate DWI Correction

- 用图像质量标签构建特征空间原型,实现弱监督图像质量迁移。
- 通过原型流匹配生成真实畸变,模拟临床常见伪影。
- 生成数据训练校正模型,在PI-RADS和格里森评分上表现更优。
单次采集的前列腺弥散加权成像(DWI)常受几何畸变影响,干扰可靠诊断。由于缺乏配对的畸变与未畸变临床扫描,自动化校正方法面临挑战。本文提出一种新颖的弱监督图像质量迁移(IQT)框架,从无畸变图像向畸变图像迁移,利用图像质量评估(IQA)信号进行监督。不同于需要昂贵体素级配对数据的传统方法或无配对算法,本方法利用图像级质量标签(畸变/无畸变)在预训练特征空间中建立潜在质量原型。考虑到模拟真实畸变比直接无配对校正更可靠,我们设计了一种弱监督原型流匹配算法,显式正则化生成轨迹朝向畸变原型,生成能模拟临床退化的磁敏感伪影。利用这些真实感配对数据,可训练第二阶段的IQT模型以正向方向实现畸变校正。实验表明,所生成图像成功模拟真实伪影的诊断干扰,使校正模型性能更强。除了定性比较,还通过在分布内和外部数据集上评估PI-RADS与格里森评分分类的下游任务性能,定量对比了我们的方法与现有无配对方法(如CycleGAN、UNIT-DDPM、OT-FM)作为正向或逆向替代方案的表现。
原文摘要 · Abstract (English)
Single-shot echo-planar prostate diffusion-weighted imaging (DWI) is frequently complicated by geometric distortions, which impact the ability to derive reliable diagnoses from such images. Developing automated correction methods is challenged by the absence of paired distorted and undistorted clinical scans. In this paper, we first propose a novel weakly-supervised image quality transfer (IQT) framework from undistorted to distorted images that utilizes image quality assessment (IQA) signals to supervise the transfer process. Unlike traditional methods that require expensive, voxel-wise paired data or resort to developing unpaired algorithms, our approach utilizes image-level quality labels (here, distorted vs. undistorted) to establish latent quality prototypes within a pre-trained feature space. Recognizing that simulating realistic distortions is more reliable than direct unpaired correction, we describe a weakly-supervised prototype flow matching algorithm to explicitly regularize generative trajectories towards distorted prototypes, producing realistic susceptibility artifacts that mimic clinical degradations. By synthesizing these realistic pairs, we enable a second IQT model to be trained in the forward direction for distortion correction. Experimental results demonstrate that our generated images successfully mimic the diagnostic interference of real-world artifacts, which leads to more capable distortion correction IQT models. In addition to qualitative comparisons, we also conduct exhaustive quantitative evaluations that compare our approach with existing unpaired approaches (e.g., CycleGAN, UNIT-DDPM, and OT-FM) - as either forward or reverse alternatives - by assessing clinical downstream task performance in PI-RADS and Gleason score classification, using both in-distribution and external data sets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。