A geometry-aware multi-coordinate transformation fusion network for optic disc and cup segmentation

Yajie Yang,Gang Yang,Yanni Wang,Xinyue Liu,Jianchun Zhao,Dayong Ding
DOI: https://doi.org/10.1007/s10489-024-05507-w
IF: 5.3
2024-05-24
Applied Intelligence
Abstract:The determination of potential glaucoma cases often relies on the vertical cup-to-disc ratio, which is directly derived from segmentation results. This study introduces an approach that leverages the geometric attributes of the optic disc (OD) and optic cup (OC) by employing a sector-based segmentation method to generate accurate segmentation results. This technique treats the OD and OC as approximate ellipses, partitioned into sectors. We propose a deep learning architecture named Sector Association and Multi-Coordinate Transformation Fusion (SAMF) network to address the problems that Convolutional Neural Networks struggle to learn long-range features. Meanwhile, our network possesses the capability to reduce the discontinuity of the boundaries of the OD and OC when inputs are images transformed into polar coordinates. The proposed SAMF network mainly consists of a Sector Association Layer and a Multi-Coordinate Transformation Fusion module. The Sector Association Layer models the interrelationships among sectors, especially the first and the last sectors, which are not contiguous in the polar transformed image. While the Multi-Coordinate Transformation Fusion module considers complementary representations in different coordinate systems, and it should be noted that the Multi-Coordinate Transformation Fusion component for various segmentation tasks. We perform experiments on the public datasets of REFUGE, Drishti-GS1 and a private dataset. Experimental results indicate that our SAMF network provides a significant boost to the performance of neural networks. The proposed method exhibits remarkable advancements in glaucoma screening through improved segmentation outcomes. The key code can be accessed at https://github.com/JieGenius/SAMFNetwork.
computer science, artificial intelligence
What problem does this paper attempt to address?