AttenNet: Deep Attention Based Retinal Disease Classification in OCT Images

Jun Wu, Yao Zhang, Jie Wang, Jianchun Zhao, Dayong Ding, Ningjiang Chen, Lingling Wang, Xuan Chen, Chunhui Jiang, Xuan Zou, Xing Liu, Hui Xiao, Yuan Tian, Zongjiang Shang, Kaiwei Wang, Xirong Li, Gang Yang, Jianping Fan

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

19 Scopus citations

Abstract

An optical coherence tomography (OCT) image is becoming the standard imaging modality in diagnosing retinal diseases and the assessment of their progression. However, the manual evaluation of the volumetric scan is time consuming, expensive and the signs of the early disease are easy to miss. In this paper, we mainly present an attention-based deep learning method for the retinal disease classification in OCT images, which can assist the large-scale screening or the diagnosis recommendation for an ophthalmologist. First, according to the unique characteristic of a retinal OCT image, we design a customized pre-processing method to improve image quality. Second, in order to guide the network optimization more effectively, a specially designed attention model, which pays more attention to critical regions containing pathological anomalies, is integrated into a typical deep learning network. We evaluate our proposed method on two data sets, and the results consistently show that it outperforms the state-of-the-art methods. We report an overall four-class accuracy of 97.4%, a two-class sensitivity of 100.0%, and a two-class specificity of 100.0% on a public data set shared by Zhang et al. with 1,000 testing B-scans in four disease classes. Compared to their work, our method improves the numbers by 0.8%, 2.2%, and 2.6% respectively.

Original languageEnglish
Title of host publicationMultiMedia Modeling - 26th International Conference, MMM 2020, Proceedings
EditorsYong Man Ro, Junmo Kim, Jung-Woo Choi, Wen-Huang Cheng, Wei-Ta Chu, Peng Cui, Min-Chun Hu, Wesley De Neve
PublisherSpringer
Pages565-576
Number of pages12
ISBN (Print)9783030377335
DOIs
StatePublished - 2020
Externally publishedYes
Event26th International Conference on MultiMedia Modeling, MMM 2020 - Daejeon, Korea, Republic of
Duration: 5 Jan 20208 Jan 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11962 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference26th International Conference on MultiMedia Modeling, MMM 2020
Country/TerritoryKorea, Republic of
CityDaejeon
Period5/01/208/01/20

Keywords

  • Attention model
  • Deep learning
  • Optical Coherence Tomography (OCT)
  • Retinal disease classification

Fingerprint

Dive into the research topics of 'AttenNet: Deep Attention Based Retinal Disease Classification in OCT Images'. Together they form a unique fingerprint.

Cite this