Evaluating Utility of Memory Efficient Medical Image Generation: A Study on Lung Nodule Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Khadra, Kathrin, Türkbey, Utku
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910652785754112
author Khadra, Kathrin
Türkbey, Utku
author_facet Khadra, Kathrin
Türkbey, Utku
contents The scarcity of publicly available medical imaging data limits the development of effective AI models. This work proposes a memory-efficient patch-wise denoising diffusion probabilistic model (DDPM) for generating synthetic medical images, focusing on CT scans with lung nodules. Our approach generates high-utility synthetic images with nodule segmentation while efficiently managing memory constraints, enabling the creation of training datasets. We evaluate the method in two scenarios: training a segmentation model exclusively on synthetic data, and augmenting real-world training data with synthetic images. In the first case, models trained solely on synthetic data achieve Dice scores comparable to those trained on real-world data benchmarks. In the second case, augmenting real-world data with synthetic images significantly improves segmentation performance. The generated images demonstrate their potential to enhance medical image datasets in scenarios with limited real-world data.
format Preprint
id arxiv_https___arxiv_org_abs_2410_12542
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Evaluating Utility of Memory Efficient Medical Image Generation: A Study on Lung Nodule Segmentation
Khadra, Kathrin
Türkbey, Utku
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
The scarcity of publicly available medical imaging data limits the development of effective AI models. This work proposes a memory-efficient patch-wise denoising diffusion probabilistic model (DDPM) for generating synthetic medical images, focusing on CT scans with lung nodules. Our approach generates high-utility synthetic images with nodule segmentation while efficiently managing memory constraints, enabling the creation of training datasets. We evaluate the method in two scenarios: training a segmentation model exclusively on synthetic data, and augmenting real-world training data with synthetic images. In the first case, models trained solely on synthetic data achieve Dice scores comparable to those trained on real-world data benchmarks. In the second case, augmenting real-world data with synthetic images significantly improves segmentation performance. The generated images demonstrate their potential to enhance medical image datasets in scenarios with limited real-world data.
title Evaluating Utility of Memory Efficient Medical Image Generation: A Study on Lung Nodule Segmentation
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2410.12542