GNAS-U2Net: A New Optic Cup and Optic Disc Segmentation Architecture With Genetic Neural Architecture Search

Neural architecture search (NAS) has made incredible progress in medical image segmentation tasks, due to its automatic design of the model. However, the search spaces studied in many existing studies are based on U-Net and its variants, which limits the potential of neural architecture search in modeling better architectures. In this study, we propose a new NAS architecture named GNAS-U(2)Net for the joint segmentation of optic cup and optic disc. This architecture is the first application of NAS in a two-level nested U-shaped structure. The best performance achieved by the joint segmentation model designed by NAS on the REFUGE dataset has an average DICE of 92.88%. Compared to U-2-Net and other related work, the model has better performance and uses only 34.79M parameters. We then verify the generalization of the model on two datasets, namely the Drishti-GS dataset and the GAMMA dataset, for which we obtain an average DICE of 92.32% and 92.11% respectively.

Automated summary

In this study, a new NAS architecture named GNAS-U(2)Net is proposed for medical image segmentation, achieving significant performance improvement. Compared to common U-Net and its variants, GNAS-U(2)Net shows better performance with fewer parameters.