Surf-CDM: Score-Based Surface Cold-Diffusion Model For Medical Image Segmentation

Fahim Ahmed Zaman; Mathews Jacob; Amanda Chang; Kan Liu; Milan Sonka; Xiaodong Wu

doi:10.48550/arxiv.2312.12649

Back

Surf-CDM: Score-Based Surface Cold-Diffusion Model For Medical Image Segmentation

Preprint

Open access

Surf-CDM: Score-Based Surface Cold-Diffusion Model For Medical Image Segmentation

Fahim Ahmed Zaman, Mathews Jacob, Amanda Chang, Kan Liu, Milan Sonka and Xiaodong Wu

arXiv.org

12/19/2023

DOI: 10.48550/arxiv.2312.12649

Files and links (1)

url

https://doi.org/10.48550/arxiv.2312.12649View

Preprint (Author's original)This preprint has not been evaluated by subject experts through peer review. Preprints may undergo extensive changes and/or become peer-reviewed journal articles. Open Access

Abstract

Diffusion models have shown impressive performance for image generation, often times outperforming other generative models. Since their introduction, researchers have extended the powerful noise-to-image denoising pipeline to discriminative tasks, including image segmentation. In this work we propose a conditional score-based generative modeling framework for medical image segmentation which relies on a parametric surface representation for the segmentation masks. The surface re-parameterization allows the direct application of standard diffusion theory, as opposed to when the mask is represented as a binary mask. Moreover, we adapted an extended variant of the diffusion technique known as the "cold-diffusion" where the diffusion model can be constructed with deterministic perturbations instead of Gaussian noise, which facilitates significantly faster convergence in the reverse diffusion. We evaluated our method on the segmentation of the left ventricle from 65 transthoracic echocardiogram videos (2230 echo image frames) and compared its performance to the most popular and widely used image segmentation models. Our proposed model not only outperformed the compared methods in terms of segmentation accuracy, but also showed potential in estimating segmentation uncertainties for further downstream analyses due to its inherent generative nature.

Computer Science - Computer Vision and Pattern Recognition

Details

Title: Subtitle: Surf-CDM: Score-Based Surface Cold-Diffusion Model For Medical Image Segmentation
Creators: Fahim Ahmed Zaman - University of Iowa
Mathews Jacob - University of Iowa
Amanda Chang - University of Iowa
Kan Liu - Washington University in St. Louis
Milan Sonka - University of Iowa
Xiaodong Wu - University of Iowa
Resource Type: Preprint
Publication Details: arXiv.org
DOI: 10.48550/arxiv.2312.12649
eISSN: 2331-8422
Language: English
Date posted: 12/19/2023
Academic Unit: Roy J. Carver Department of Biomedical Engineering; Radiology; Electrical and Computer Engineering; Iowa Technology Institute; Iowa Neuroscience Institute; Cardiovascular Medicine; Radiation Oncology; The Iowa Institute for Biomedical Imaging; Fraternal Order of Eagles Diabetes Research Center; Injury Prevention Research Center; Internal Medicine; Ophthalmology and Visual Sciences
Record Identifier: 9984532356402771

Metrics

99 Record Views