2501.02447

Total: 1

#1 MedSegDiffNCA: Diffusion Models With Neural Cellular Automata for Skin Lesion Segmentation [PDF1] [Copy] [Kimi] [REL]

Authors: Avni Mittal, John Kalkhof, Anirban Mukhopadhyay, Arnav Bhavsar

Denoising Diffusion Models (DDMs) are widely used for high-quality image generation and medical image segmentation but often rely on Unet-based architectures, leading to high computational overhead, especially with high-resolution images. This work proposes three NCA-based improvements for diffusion-based medical image segmentation. First, Multi-MedSegDiffNCA uses a multilevel NCA framework to refine rough noise estimates generated by lower level NCA models. Second, CBAM-MedSegDiffNCA incorporates channel and spatial attention for improved segmentation. Third, MultiCBAM-MedSegDiffNCA combines these methods with a new RGB channel loss for semantic guidance. Evaluations on Lesion segmentation show that MultiCBAM-MedSegDiffNCA matches Unet-based model performance with dice score of 87.84% while using 60-110 times fewer parameters, offering a more efficient solution for low resource medical settings.

Subjects: Computer Vision and Pattern Recognition , Machine Learning , Image and Video Processing

Publish: 2025-01-05 05:54:27 UTC