RelativeFlow: Taming Medical Image Denoising Learning with Noisy Reference

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Medical Imaging · Depth: Expert, medium

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

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

Topics

Code references

Best for: Computer Vision Engineer, AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.