Image & Video

MLFFM-SegDiff: New Diffusion Model Boosts Skin Lesion Segmentation Accuracy

New AI model achieves 92.07% Dice score on lesion segmentation, outperforming U-Net and SwinUNETR.

Deep Dive

Skin lesion segmentation is critical for computer-aided dermatology, but dermoscopic images suffer from blurred boundaries, low contrast, shape variations, and artifacts like hair and shadows. Diffusion models have shown promise in medical image segmentation due to their progressive denoising and distribution modeling. However, existing diffusion-based methods struggle with cross-level feature interaction and boundary detail recovery, limiting their accuracy on challenging cases.

To overcome these limitations, the team proposes MLFFM-SegDiff, built on a diffusion framework with three key innovations: a dual-path U-Net encoder that enhances interaction between noisy mask features and dermoscopic image features; a Multi-Level Feature Fusion Module (MLFFM) that improves skip connections via attention, scale alignment, and adaptive cross-level fusion; and a boundary-sensitive loss function. These designs let the decoder jointly leverage shallow boundary cues and deep semantic representations, improving mask reconstruction quality. On the ISIC2018, PH2, and HAM10000 datasets, MLFFM-SegDiff achieves an average Jaccard index of 0.8546 and Dice coefficient of 0.9207, outperforming DermoSegDiff, U-Net, and SwinUNETR across Accuracy, F1-score, Jaccard, Recall, and Dice. The code will be released after publication.

Key Points
  • Dual-path U-Net encoder enables better interaction between noisy mask features and input image features.
  • Multi-Level Feature Fusion Module (MLFFM) improves skip connections with attention, scale alignment, and adaptive fusion.
  • Achieves state-of-the-art results: Jaccard 0.8546 and Dice 0.9207 on benchmark datasets ISIC2018, PH2, and HAM10000.

Why It Matters

More accurate skin lesion segmentation enables earlier melanoma detection and reduces false positives in dermatology AI.

📬 Get the top 10 AI stories daily