Journal of Innovative Optical Health Sciences, Volume. 15, Issue 2, 2250009(2022)

Computer-aided diagnosis of retinopathy based on vision transformer

[in Chinese]1... [in Chinese]1, [in Chinese]1, [in Chinese]2, [in Chinese]2, [in Chinese]1,* and [in Chinese]1 |Show fewer author(s)
Author Affiliations
  • 1Shanghai Institute of Technology, 100 Haiquan Road, Shanghai 201418, China
  • 2School of Ophthalmology and Optometry, Wenzhou Medical University, Xueyuan Road 270, Wenzhou, Zhejiang 325027, China
  • show less

    Age-related Macular Degeneration (AMD) and Diabetic Macular Edema (DME) are two common retinal diseases for elder people that may ultimately cause irreversible blindness. Timely and accurate diagnosis is essential for the treatment of these diseases. In recent years, computer-aided diagnosis (CAD) has been deeply investigated and effectively used for rapid and early diagnosis. In this paper, we proposed a method of CAD using vision transformer to analyze optical coherence tomography (OCT) images and to automatically discriminate AMD, DME, and normal eyes. A classification accuracy of 99.69% was achieved. After the model pruning, the recognition time reached 0.010 s and the classification accuracy did not drop. Compared with the Convolutional Neural Network (CNN) image classification models (VGG16, Resnet50, Densenet121, and E±cientNet), vision transformer after pruning exhibited better recognition ability. Results show that vision transformer is an improved alternative to diagnose retinal diseases more accurately.

    Tools

    Get Citation

    Copy Citation Text

    [in Chinese], [in Chinese], [in Chinese], [in Chinese], [in Chinese], [in Chinese], [in Chinese]. Computer-aided diagnosis of retinopathy based on vision transformer[J]. Journal of Innovative Optical Health Sciences, 2022, 15(2): 2250009

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Aug. 10, 2021

    Accepted: Nov. 22, 2021

    Published Online: Feb. 28, 2022

    The Author Email: (jiangwenping@sit.edu.cn)

    DOI:10.1142/s1793545822500092

    Topics