Abstract:Large-scale, big-variant, high-quality data are crucial for developing robust and successful deep-learning models for medical applications since they potentially enable better generalization performance and avoid overfitting. However, the scarcity of high-quality labeled data always presents significant challenges. This paper proposes a novel approach to address this challenge by developing controllable diffusion models for medical image synthesis, called DiffBoost. We leverage recent diffusion probabilistic models to generate realistic and diverse synthetic medical image data that preserve the essential characteristics of the original medical images by incorporating edge information of objects to guide the synthesis process. In our approach, we ensure that the synthesized samples adhere to medically relevant constraints and preserve the underlying structure of imaging data. Due to the random sampling process by the diffusion model, we can generate an arbitrary number of synthetic images with diverse appearances. To validate the effectiveness of our proposed method, we conduct an extensive set of medical image segmentation experiments on multiple datasets, including Ultrasound breast (+13.87%), CT spleen (+0.38%), and MRI prostate (+7.78%), achieving significant improvements over the baseline segmentation methods. The promising results demonstrate the effectiveness of our \textcolor{black}{DiffBoost} for medical image segmentation tasks and show the feasibility of introducing a first-ever text-guided diffusion model for general medical image segmentation tasks. With carefully designed ablation experiments, we investigate the influence of various data augmentations, hyper-parameter settings, patch size for generating random merging mask settings, and combined influence with different network architectures. Source code are available at <a class="link-external link-https" href="https://github.com/NUBagciLab/DiffBoost" rel="external noopener nofollow">this https URL</a>.

Rethinking Deep Active Learning for Medical Image Segmentation: A Diffusion and Angle-Based Framework

Diminishing Uncertainty Within the Training Pool: Active Learning for Medical Image Segmentation

PCDAL: A Perturbation Consistency-Driven Active Learning Approach for Medical Image Segmentation and Classification

A Deep Active Learning Framework with Information Guided Label Generation for Medical Image Segmentation.

Annotation-efficient deep learning for automatic medical image segmentation

Attention, Suggestion and Annotation: A Deep Active Learning Framework for Biomedical Image Segmentation

HiDiff: Hybrid Diffusion Framework for Medical Image Segmentation

HAL-IA: A Hybrid Active Learning framework using Interactive Annotation for medical image segmentation

Enhancing Medical Image Segmentation with Deep Learning and Diffusion Models

OneSeg: Self-learning and One-shot Learning based Single-slice Annotation for 3D Medical Image Segmentation

DSAL: Deeply Supervised Active Learning from Strong and Weak Labelers for Biomedical Image Segmentation.

Predictive Accuracy-Based Active Learning for Medical Image Segmentation

Integrating Deep Metric Learning with Coreset for Active Learning in 3D Segmentation

DiffBoost: Enhancing Medical Image Segmentation via Text-Guided Diffusion Model

A dynamic interactive learning framework for automated 3D medical image segmentation

Diffuse-UDA: Addressing Unsupervised Domain Adaptation in Medical Image Segmentation with Appearance and Structure Aligned Diffusion Models

IOADA: an Optimal Automated Augmentation Algorithm for Medical Image Segmentation

Retrieval-augmented Few-shot Medical Image Segmentation with Foundation Models

Breaking the Barrier: Selective Uncertainty-based Active Learning for Medical Image Segmentation

An Annotation Sparsification Strategy for 3D Medical Image Segmentation via Representative Selection and Self-Training

Model-data-driven adversarial active learning for brain tumor segmentation