Hybrid Approach for Enhancing Lesion Segmentation in Fundus Images

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Health & Medical Research, Medical Devices & Health Technology · Depth: Expert, extended

Summary

A novel hybrid model has been developed to enhance the segmentation of choroidal nevi (CN) lesions in fundus images, addressing challenges like indistinct borders, noise sensitivity, and dependency on large, high-quality datasets. This approach combines traditional mathematical/clustering segmentation methods, specifically SLIC, with insights from a U-Net convolutional neural network. The hybrid model achieved a Dice coefficient of 89.7% and an IoU of 80.01% on 1024x1024 fundus images, significantly outperforming the Attention U-Net model, which scored 51.3% and 34.2% respectively. It also demonstrated superior generalizability on external datasets and operates efficiently on standard CPUs, making it suitable for resource-constrained clinical environments. This work is part of a broader effort to create a decision support system for early CN diagnosis and monitoring.

Key takeaway

For AI Scientists and Computer Vision Engineers developing medical image diagnostic tools, this hybrid segmentation approach offers a compelling solution for choroidal nevus detection. Your teams should consider integrating traditional segmentation methods with lightweight U-Net models to achieve high accuracy on large, high-resolution images while significantly reducing GPU dependency and improving generalizability. This strategy can lead to more practical, deployable solutions for clinical settings, especially where hardware resources are limited.

Key insights

A hybrid model combining U-Net and SLIC significantly improves choroidal nevus segmentation accuracy and generalizability on fundus images.

Principles

Method

The proposed method trains a small-sized U-Net (128x128) to generate parameters for a traditional SLIC segmentation model, which then segments full-size (1024x1024 or 3900x3900) images, automating parameter selection and improving efficiency.

In practice

Topics

Best for: AI Scientist, Research Scientist, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.