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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Tumor Segmentation | KiTS Kidney Tumor 23 | DSC38.39 | 12 | |
| Tumor Segmentation | MSD Lung Tumor | DSC27.73 | 9 | |
| Tumor Segmentation | MSD Brain | DSC18.35 | 9 | |
| Tumor Segmentation | BraTS 23 | DSC17.04 | 9 | |
| Tumor Segmentation | MSD Pancreas Tumor | DSC29.45 | 9 | |
| Tumor Segmentation | MSD Liver Tumor | DSC47.68 | 9 | |
| Tumor Segmentation | MSD Colon Tumor | DSC18.36 | 9 | |
| Tumor Segmentation | MSD Hepatic Vessel Tumor | DSC41.33 | 9 | |
| Tumor Segmentation | In-house MRI Liver Tumor | DSC0.1647 | 9 | |
| Brain MRI Stroke Segmentation | ATLAS and IXI | Average Dice29.74 | 6 |