A neural network aided classification method for human eye diseases

Jiannan Zhou; Yuanying Wang; Wei Liu

doi:10.1117/12.2675128

25 May 2023 A neural network aided classification method for human eye diseases

Jiannan Zhou, Yuanying Wang, Wei Liu

Proceedings Volume 12636, Third International Conference on Machine Learning and Computer Application (ICMLCA 2022); 126360J (2023) https://doi.org/10.1117/12.2675128
Event: Third International Conference on Machine Learning and Computer Application (ICMLCA 2022), 2022, Shenyang, China

Abstract

Patients with eye diseases may have a variety of pathological features. If a certain pathological feature is ignored during diagnosis, the overall treatment effect may be seriously affected. To solve this problem, a new neural network model is proposed, in which all the possible pathological features in an OCT image are trained together as the target of multi task classification. The model uses an improved residual neural network to extract the features of OCT images, and adds a spatial attention module in the residual module to facilitate the model to capture the location of lesions more accurately. MMoE model is used to classify the extracted features in multiple tasks, making full use of the potential relationship between each task to help classification, so as to improve the stability of training. In addition, the patient's personal information features and image features are fused with attention and embedded in the top of the model to improve the classification accuracy. The model is compared with VGG16, ResNet10 general single task model and other multi task learning models on 2729 human eye OCT images containing four pathological features. The experimental results show that the training time of the model is 1/4 of that of the single task model, and the average classification accuracy of the four tasks is improved by 4%.

Citation Download Citation

Jiannan Zhou, Yuanying Wang, and Wei Liu "A neural network aided classification method for human eye diseases", Proc. SPIE 12636, Third International Conference on Machine Learning and Computer Application (ICMLCA 2022), 126360J (25 May 2023); https://doi.org/10.1117/12.2675128

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available