Search papers, labs, and topics across Lattice.
This paper introduces a weakly-supervised image quality transfer (IQT) framework that addresses the challenge of correcting geometric distortions in prostate diffusion-weighted imaging (DWI) without requiring paired clinical scans. By leveraging image quality assessment signals to create latent quality prototypes, the method simulates realistic distortions, allowing for effective training of a second IQT model for distortion correction. Experimental results show that this approach not only generates images that closely mimic real-world artifacts but also outperforms existing unpaired methods in clinical task performance metrics such as PI-RADS and Gleason score classification.
Realistic distortion simulation enables a novel framework that significantly enhances prostate DWI correction without the need for costly paired data.
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.
