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

Jun-Ding Sun,Chong Yao,Jie Liu,Weifan Liu,Ze-Kuan Yu
DOI: https://doi.org/10.1109/lsp.2022.3151549
2022-01-01
IEEE Signal Processing Letters
Abstract: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-U2Net 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 U2-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.
What problem does this paper attempt to address?