A 3D CNN-based Multi-task Learning for Cataract screening and left and right eye classification on 3D AS-OCT images

Xiao，Zunjie1; Zhang，Xiaoqing1; Higashita，Risa2; Chen，Wan3; Yuan，Jin3; Liu，Jiang1

10.1145/3484377.3484378

2021-08-13

ACM International Conference Proceeding Series

1-7

Cataract is the leading cause for visual impairment and blindness. Cataract screening can effectively improve the recovery rate of cataract, and the left and right eye classification is a significant step in cataract screening. Anterior segment optical coherence tomography (AS-OCT) is a non-contact, high-resolution ophthalmic imaging technique, which can quickly obtain pathological information of cataract and left and right eye position information through three-dimensional (3D) imaging. In order to improve the efficiency of cataract screening, we propose a multi-task three-dimensional convolutional neural network (MT-CNN) for automatic cataract detection and left and right eye classification simultaneously based on the 3D AS-OCT images. The MT-CNN is designed based on the hard sharing mechanism, achieving better performance with fewer parameters than single-task learning. The results on an AS-OCT image dataset show that the 3D CNN model obtains better classification performance than the 2D CNN model. Compared with the single-task 3D CNN model, MT-CNN achieves higher accuracy under the premise of greatly parameters reduction and computational complexity reduction.

AS-OCT Cataract screening Left and right eye classification Multi-task learning Three-dimensional convolutional neural network

第一

英语

EI

20220411489768

3D modeling ; Classification (of information) ; Convolution ; Image classification ; Image enhancement ; Learning systems ; Optical tomography

Information Theory and Signal Processing:716.1 ; Data Processing and Image Processing:723.2 ; Optical Devices and Systems:741.3 ; Information Sources and Analysis:903.1

2-s2.0-85123024753

Scopus

1.Department of Computer Science and Engineering,Southern University of Science and Technology,China
2.Tomey Corporation,Japan and Department of Computer Science and Engineering,Southern University of Science and Technology,China
3.Zhongshan Ophthalmic Center,Sun Yat-sen University,China

Xiao，Zunjie,Zhang，Xiaoqing,Higashita，Risa,et al. A 3D CNN-based Multi-task Learning for Cataract screening and left and right eye classification on 3D AS-OCT images[C],2021:1-7.