Genetic U-Net

Genetic U-Net applies an improved genetic algorithm to automatically optimize U-Net–style convolutional neural network architectures for accurate and parameter-efficient retinal vessel segmentation from fundus images.


Key Features:

  • Automated Architecture Design: Uses an improved genetic algorithm to explore a condensed, flexible search space based on a U-shaped encoder-decoder (U-Net) structure to generate CNN architectures.
  • Parameter Efficiency: Produces architectures with substantially fewer parameters, using less than 1% of the parameter count of the original U-Net and fewer than other state-of-the-art models.
  • Enhanced Performance: Identifies effective operations and network patterns via exhaustive search within the defined architecture space to improve retinal vessel segmentation accuracy.
  • Reduced Overfitting Risk: Lowers model complexity and parameter count to mitigate overfitting and improve robustness across diverse datasets.

Scientific Applications:

  • Retinal vessel segmentation: Generates accurate vessel segmentations from fundus images for quantitative vascular analysis.
  • Ophthalmic disease diagnosis and monitoring: Supports diagnosis and monitoring of ocular diseases such as diabetic retinopathy and glaucoma through vessel-based image biomarkers.
  • Clinical research and diagnostics: Enables parameter-efficient models suitable for medical imaging research and diagnostic workflows.

Methodology:

Search space definition based on a U-shaped encoder-decoder framework; improved genetic algorithm optimization to iteratively explore and select candidate architectures; performance evaluation of resulting architectures on retinal vessel segmentation tasks with focus on accuracy and parameter efficiency.

Topics

Details

License:
MIT
Cost:
Free of charge
Tool Type:
command-line tool
Operating Systems:
Mac, Linux, Windows
Programming Languages:
Python
Added:
3/1/2022
Last Updated:
3/1/2022

Operations

Publications

Wei J, Zhu G, Fan Z, Liu J, Rong Y, Mo J, Li W, Chen X. Genetic U-Net: Automatically Designed Deep Networks for Retinal Vessel Segmentation Using a Genetic Algorithm. IEEE Transactions on Medical Imaging. 2022;41(2):292-307. doi:10.1109/tmi.2021.3111679. PMID:34506278.

PMID: 34506278
Funding: - Special Fund of Science and Technology Innovation Strategy of Guangdong Province: 2019A050520001 - International Cooperation Base of Guangdong Province: 2019A050519008 - State Key Lab of Digital Manufacturing Equipment & Technology: DMETKF2019020