Light-weight Retinal Layer Segmentation with Global Reasoning

  • 2024-04-25 06:42:41
  • Xiang He, Weiye Song, Yiming Wang, Fabio Poiesi, Ji Yi, Manishi Desai, Quanqing Xu, Kongzheng Yang, Yi Wan
  • 0

Abstract

Automatic retinal layer segmentation with medical images, such as opticalcoherence tomography (OCT) images, serves as an important tool for diagnosingophthalmic diseases. However, it is challenging to achieve accuratesegmentation due to low contrast and blood flow noises presented in the images.In addition, the algorithm should be light-weight to be deployed for practicalclinical applications. Therefore, it is desired to design a light-weightnetwork with high performance for retinal layer segmentation. In this paper, wepropose LightReSeg for retinal layer segmentation which can be applied to OCTimages. Specifically, our approach follows an encoder-decoder structure, wherethe encoder part employs multi-scale feature extraction and a Transformer blockfor fully exploiting the semantic information of feature maps at all scales andmaking the features have better global reasoning capabilities, while thedecoder part, we design a multi-scale asymmetric attention (MAA) module forpreserving the semantic information at each encoder scale. The experiments showthat our approach achieves a better segmentation performance compared to thecurrent state-of-the-art method TransUnet with 105.7M parameters on both ourcollected dataset and two other public datasets, with only 3.3M parameters.

 

Quick Read (beta)

loading the full paper ...