Hybrid Approach for Enhancing Lesion Segmentation in Fundus Images
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
- Smaller image sizes improve U-Net segmentation accuracy.
- Hybrid models can reduce dataset dependency and computational cost.
- Traditional segmentation methods benefit from automated parameter selection.
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
- Use 128x128 image sizes for U-Net training to optimize accuracy.
- Integrate traditional segmentation with ML for resource-efficient inference.
- Automate SLIC parameter selection using U-Net outputs.
Topics
- Choroidal Nevi Segmentation
- Hybrid Segmentation Model
- U-Net Architecture
- SLIC Algorithm
- Fundus Image Analysis
Best for: AI Scientist, Research Scientist, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.