DermoSegDiff: A Boundary-aware Segmentation Diffusion Model for Skin
Lesion Delineation
- URL: http://arxiv.org/abs/2308.02959v1
- Date: Sat, 5 Aug 2023 22:12:01 GMT
- Title: DermoSegDiff: A Boundary-aware Segmentation Diffusion Model for Skin
Lesion Delineation
- Authors: Afshin Bozorgpour and Yousef Sadegheih and Amirhossein Kazerouni and
Reza Azad and Dorit Merhof
- Abstract summary: We propose DermoSegDiff, a novel framework for skin lesion segmentation that incorporates boundary information during the learning process.
Our approach introduces a novel loss function that prioritizes the boundaries during training, gradually reducing the significance of other regions.
We also introduce a novel U-Net-based denoising network that proficiently integrates noise and semantic information inside the network.
- Score: 3.9548535445908928
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Skin lesion segmentation plays a critical role in the early detection and
accurate diagnosis of dermatological conditions. Denoising Diffusion
Probabilistic Models (DDPMs) have recently gained attention for their
exceptional image-generation capabilities. Building on these advancements, we
propose DermoSegDiff, a novel framework for skin lesion segmentation that
incorporates boundary information during the learning process. Our approach
introduces a novel loss function that prioritizes the boundaries during
training, gradually reducing the significance of other regions. We also
introduce a novel U-Net-based denoising network that proficiently integrates
noise and semantic information inside the network. Experimental results on
multiple skin segmentation datasets demonstrate the superiority of DermoSegDiff
over existing CNN, transformer, and diffusion-based approaches, showcasing its
effectiveness and generalization in various scenarios. The implementation is
publicly accessible on
\href{https://github.com/mindflow-institue/dermosegdiff}{GitHub}
Related papers
- SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidance [0.559239450391449]
Skin lesion segmentation is a crucial method for identifying early skin cancer.
We propose a hybrid architecture based on Mamba and CNN, called SkinMamba.
It maintains linear complexity while offering powerful long-range dependency modeling and local feature extraction capabilities.
arXiv Detail & Related papers (2024-09-17T05:02:38Z) - Coarse-to-Fine Proposal Refinement Framework for Audio Temporal Forgery Detection and Localization [60.899082019130766]
We introduce a frame-level detection network (FDN) and a proposal refinement network (PRN) for audio temporal forgery detection and localization.
FDN aims to mine informative inconsistency cues between real and fake frames to obtain discriminative features that are beneficial for roughly indicating forgery regions.
PRN is responsible for predicting confidence scores and regression offsets to refine the coarse-grained proposals derived from the FDN.
arXiv Detail & Related papers (2024-07-23T15:07:52Z) - HiDiff: Hybrid Diffusion Framework for Medical Image Segmentation [16.906987804797975]
HiDiff is a hybrid diffusion framework for medical image segmentation.
It can synergize the strengths of existing discriminative segmentation models and new generative diffusion models.
It excels at segmenting small objects and generalizing to new datasets.
arXiv Detail & Related papers (2024-07-03T23:59:09Z) - DiffSeg: A Segmentation Model for Skin Lesions Based on Diffusion Difference [2.9082809324784082]
We introduce DiffSeg, a segmentation model for skin lesions based on diffusion difference.
Its multi-output capability mimics doctors' annotation behavior, facilitating the visualization of segmentation result consistency and ambiguity.
We demonstrate the effectiveness of DiffSeg on the ISIC 2018 Challenge dataset, outperforming state-of-the-art U-Net-based methods.
arXiv Detail & Related papers (2024-04-25T09:57:52Z) - DiffVein: A Unified Diffusion Network for Finger Vein Segmentation and
Authentication [50.017055360261665]
We introduce DiffVein, a unified diffusion model-based framework which simultaneously addresses vein segmentation and authentication tasks.
For better feature interaction between these two branches, we introduce two specialized modules.
In this way, our framework allows for a dynamic interplay between diffusion and segmentation embeddings.
arXiv Detail & Related papers (2024-02-03T06:49:42Z) - Multi-task Explainable Skin Lesion Classification [54.76511683427566]
We propose a few-shot-based approach for skin lesions that generalizes well with few labelled data.
The proposed approach comprises a fusion of a segmentation network that acts as an attention module and classification network.
arXiv Detail & Related papers (2023-10-11T05:49:47Z) - Reliable Joint Segmentation of Retinal Edema Lesions in OCT Images [55.83984261827332]
In this paper, we propose a novel reliable multi-scale wavelet-enhanced transformer network.
We develop a novel segmentation backbone that integrates a wavelet-enhanced feature extractor network and a multi-scale transformer module.
Our proposed method achieves better segmentation accuracy with a high degree of reliability as compared to other state-of-the-art segmentation approaches.
arXiv Detail & Related papers (2022-12-01T07:32:56Z) - Fuzzy Attention Neural Network to Tackle Discontinuity in Airway
Segmentation [67.19443246236048]
Airway segmentation is crucial for the examination, diagnosis, and prognosis of lung diseases.
Some small-sized airway branches (e.g., bronchus and terminaloles) significantly aggravate the difficulty of automatic segmentation.
This paper presents an efficient method for airway segmentation, comprising a novel fuzzy attention neural network and a comprehensive loss function.
arXiv Detail & Related papers (2022-09-05T16:38:13Z) - BEFD: Boundary Enhancement and Feature Denoising for Vessel Segmentation [15.386077363312372]
We propose Boundary Enhancement and Feature Denoising (BEFD) module to facilitate the network ability of extracting boundary information in semantic segmentation.
By introducing Sobel edge detector, the network is able to acquire additional edge prior, thus enhancing boundary in an unsupervised manner for medical image segmentation.
arXiv Detail & Related papers (2021-04-08T13:44:47Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Cascaded Context Enhancement Network for Automatic Skin Lesion
Segmentation [10.648218637920035]
We formulate a cascaded context enhancement neural network for automatic skin lesion segmentation.
A new context aggregation module with a gate-based information integration approach is proposed.
We evaluate our approach on four public skin dermoscopy image datasets.
arXiv Detail & Related papers (2020-04-17T08:25:17Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.