Source-linked AI summary

EA-LiteUNet: An Edge-Adaptive and Resource-Efficient U-Net for Boundary-Sensitive Dermoscopic Image Segmentation

Wang Jiangtao, Nur Intan Raihana Ruhaiyem, Fu Panpan, Yang Yu, Huang Yan

arXiv:2608.15537v1cs.CVcs.AI

TL;DR

Dermoscopic segmentation requires accurate target-region delineation while maintaining overall accuracy. EA-LiteUNet is an edge-adaptive, resource-efficient U-shaped architecture that achieves Dice 92.08% and HD95 12.89 pixels on ISIC2018.

  • Problem

    Dermoscopic segmentation requires identifying target regions while maintaining high overall accuracy.

  • Method

    EA-LiteUNet is an edge-adaptive and resource-efficient U-shaped architecture for dermoscopic image segmentation.

  • Results

    92.08% Dice and 12.89 pixels HD95 are achieved on ISIC2018, with the best reported results across key metrics.

  • Takeaways & Limitations

    EA-LiteUNet can better identify target regions while maintaining high overall accuracy on ISIC2018.

  • Takeaways & Limitations

    Validation is restricted to dermoscopic skin-lesion datasets, so generalization to other imaging modalities and anatomical targets remains unverified.

Abstract

from arXiv · show

Accurate boundary delineation remains a persistent challenge in dermoscopic image segmentation because of blurred lesion margins, heterogeneous textures, and complex background artifacts. From a signal-processing perspective, lesion boundaries represent high-frequency components that are highly susceptible to aliasing, noise amplification, and information loss. Consequently, repeated downsampling and feature transformations in conventional convolutional architectures often lead to severely degraded boundary representations. To address these limitations, we propose EA-LiteUNet, an edge-adaptive and computationally efficient U-Net variant specifically designed for boundary-sensitive medical image segmentation. The architecture integrates three core mechanisms: (1) boundary-aware representation learning to suppress aliasing and preserve high-frequency structural details; (2) attention-guided feature modulation to selectively enhance boundary-relevant responses across multi-scale features; and (3) a resource-adaptive inference strategy to dynamically balance segmentation accuracy and computational efficiency. Extensive evaluations across three public dermoscopic datasets demonstrate that EA-LiteUNet consistently achieves superior boundary precision. Specifically, on the ISIC 2018 dataset, the method significantly reduces the 95% Hausdorff Distance (HD95) to 12.89 pixels while maintaining a robust Dice score of 92.08%. Notably, this strong performance is achieved with an ultralightweight configuration of merely 0.29M parameters and 1.17 GFLOPs. Ablation studies further validate the complementary effects of these components, confirming their contribution to enhanced boundary fidelity and stable optimization.

4.3 Implementation details

EA-LiteUNet was implemented in PyTorch with standardized 256×256 inputs, normalization, and augmentation, then evaluated across dermoscopic datasets. It achieved strong boundary-sensitive segmentation performance, including leading results on ISIC2018 and PH2 with lightweight computation.

  • Implementation: EA-LiteUNet was implemented in PyTorch and trained on an NVIDIA GeForce RTX 4060 GPU.
  • Implementation: 256×256 input resizing, normalization, random rotation, flipping, and brightness–contrast jitter were applied before training.
  • ISIC2018 results: 92.08% Dice, 12.89-pixel HD95, and 94.67% sensitivity were achieved on ISIC2018, outperforming the compared segmentation models across key metrics.
  • ISIC2017 results: 89.12% Dice, 14.23-pixel HD95, and 93.07% sensitivity were achieved on ISIC2017, with improvements over LB-UNet of 0.27% Dice, 0.21% sensitivity, and 4.34 pixels in HD95.
  • Efficiency: 0.29M parameters and 1.17 GFLOPs characterize the EA-LiteUNet configuration listed for ISIC2017, alongside 89.52±0.12 Dice, 95.89±0.13 specificity, and 14.23±0.11 HD95.

(1) Region-Based Convergence (Dice)

Region-based convergence improves from the baseline to the full configuration: model-7 reaches higher Dice values, stabilizes earlier, and exhibits reduced variance. These trends indicate that representation stabilization and attention-guided modulation improve regional consistency during training.

  • Model-1 (baseline) shows larger Dice oscillations and slower stabilization, reflecting unstable region prediction.
  • Model-7 (full configuration) achieves higher Dice values with reduced variance and earlier convergence.The improvement is evidenced by an earlier knee epoch and a stable plateau phase.
  • Representation-level stabilization and attention-guided modulation improve regional consistency during training.
  • The full model’s smoother convergence is consistent with the joint configuration achieving the best Dice/HD95 and lowest run-to-run variance.
Loading 2608.15537v1…