Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Diffusion Models with Implicit Guidance for Medical Anomaly Detection

About

Diffusion models have advanced unsupervised anomaly detection by improving the transformation of pathological images into pseudo-healthy equivalents. Nonetheless, standard approaches may compromise critical information during pathology removal, leading to restorations that do not align with unaffected regions in the original scans. Such discrepancies can inadvertently increase false positive rates and reduce specificity, complicating radiological evaluations. This paper introduces Temporal Harmonization for Optimal Restoration (THOR), which refines the de-noising process by integrating implicit guidance through temporal anomaly maps. THOR aims to preserve the integrity of healthy tissue in areas unaffected by pathology. Comparative evaluations show that THOR surpasses existing diffusion-based methods in detecting and segmenting anomalies in brain MRIs and wrist X-rays. Code: https://github.com/ci-ber/THOR_DDPM.

Cosmin I. Bercea, Benedikt Wiestler, Daniel Rueckert, Julia A. Schnabel• 2024

Related benchmarks

TaskDatasetResultRank
Tumor SegmentationKiTS Kidney Tumor 23
DSC38.39
12
Tumor SegmentationMSD Lung Tumor
DSC27.73
9
Tumor SegmentationMSD Brain
DSC18.35
9
Tumor SegmentationBraTS 23
DSC17.04
9
Tumor SegmentationMSD Pancreas Tumor
DSC29.45
9
Tumor SegmentationMSD Liver Tumor
DSC47.68
9
Tumor SegmentationMSD Colon Tumor
DSC18.36
9
Tumor SegmentationMSD Hepatic Vessel Tumor
DSC41.33
9
Tumor SegmentationIn-house MRI Liver Tumor
DSC0.1647
9
Brain MRI Stroke SegmentationATLAS and IXI
Average Dice29.74
6
Showing 10 of 11 rows

Other info

Code

Follow for update