GenDiff: A Dose and Anatomy Aware Diffusion Model with Structural Prior Refinement for Low-Dose CT Reconstruction and Generalization

· Source: Computer Vision and Pattern Recognition · Field: Science & Research — Health & Medical Research, Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

GenDiff is a novel diffusion-based framework designed for low-dose computed tomography (LDCT) reconstruction, addressing the limitations of existing methods that struggle with varying radiation dose levels and anatomical regions. This framework jointly models continuous radiation dose and anatomical information within a unified network. Key components include a Dose-Anatomy Encoder for acquisition-aware embeddings, a dose- and anatomy-conditioned cold diffusion backbone for iterative refinement, a physics-consistency update, and a Structural Prior Refinement Module (SPRM) to preserve anatomical structures. Extensive experiments on multi-anatomy clinical datasets, including unseen ultra-low-dose conditions and out-of-distribution phantom/animal datasets, demonstrate GenDiff's superior performance over state-of-the-art convolutional neural network and diffusion-based methods. It achieves robust reconstruction quality across diverse clinical scenarios.

Key takeaway

For Machine Learning Engineers developing low-dose CT reconstruction models, GenDiff's approach of jointly modeling continuous radiation dose and anatomical information offers a robust solution. You should consider integrating dose-anatomy encoders and structural prior refinement to enhance generalization across diverse clinical datasets and ultra-low-dose conditions, improving diagnostic image quality. This method promises superior performance and robustness in practical clinical settings.

Key insights

GenDiff integrates dose and anatomy awareness into a diffusion model for robust low-dose CT reconstruction.

Principles

Method

GenDiff employs a Dose-Anatomy Encoder, a conditioned cold diffusion backbone, physics-consistency updates, and a Structural Prior Refinement Module (SPRM).

In practice

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.