RelativeFlow: Taming Medical Image Denoising Learning with Noisy Reference
Summary
RelativeFlow is a novel flow matching framework designed to overcome the "noisy reference problem" in medical image denoising (MID), where absolutely clean images are unavailable for supervision. Existing methods like Simulated-Supervised Discriminative Learning (SimSDL), Self-Supervised Learning (SSL), and Simulated-Supervised Generative Learning (SimSGL) either treat noisy references as clean targets, leading to suboptimal convergence or reference-biased learning, or impose restrictive noise assumptions. RelativeFlow reformulates flow matching by decomposing the absolute noise-to-clean mapping into relative noisier-to-noisy mappings. It achieves this through two main components: Consistent Transport (CoT), a displacement map that ensures relative flows compose a unified absolute flow, and Simulation-based Velocity Field (SVF), which constructs a learnable velocity field using modality-specific degradation operators. Experiments on Computed Tomography (CT) and Magnetic Resonance (MR) denoising demonstrate that RelativeFlow significantly outperforms current methods.
Key takeaway
For Computer Vision Engineers developing medical image denoising solutions, RelativeFlow offers a robust approach to overcome the limitations of noisy training data. By reformulating flow matching with Consistent Transport and Simulation-based Velocity Fields, you can achieve superior denoising performance on modalities like CT and MR, even when only heterogeneous noisy references are available. Consider integrating this framework to improve the consistency and quality of your denoised medical images.
Key insights
RelativeFlow uses flow matching to learn unified denoising from heterogeneous noisy medical image references.
Principles
- Decompose absolute noise-to-clean mapping into relative noisier-to-noisy mappings.
- Constrain relative flows to progressively compose a unified absolute flow.
- Utilize modality-specific degradation operators for velocity field construction.
Method
RelativeFlow employs Consistent Transport (CoT) to unify relative flows and a Simulation-based Velocity Field (SVF) based on medical image degradation processes to learn denoising across modalities like CT and MR.
In practice
- Apply RelativeFlow to CT denoising tasks.
- Implement RelativeFlow for MR image denoising.
- Use provided code for training and evaluation.
Topics
- Medical Image Denoising
- Noisy Reference Problem
- RelativeFlow Framework
- Flow Matching
- Consistent Transport
Code references
Best for: Computer Vision Engineer, AI Scientist, Research Scientist, Machine Learning Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.