Potential of Attention Mechanism for Classification of Optical Coherence Tomography Images.

Zhihua Shang,Zilong Fu,Chuanbin Liu,Hongtao Xie,Yongdong Zhang
DOI: https://doi.org/10.1109/vcip.2018.8698622
2018-01-01
Abstract:Deep neural network (DNN) can extract high-dimensional feature of images for computer vision tasks including Optical Coherence Tomography (OCT) images classification. However, OCT images are usually processed by DNN just like natural images, thus the performance of DNN is not satisfactory. We present an end-to-end DNN targeting OCT images classification. Considering the characteristic of OCT images, we introduce attention mechanism into classifier to extract more specific feature of OCT images. Our network demonstrates its capacity to enhance the features that represent the disease region. Our method achieves the state-of-the-art performance with average accuracy of 99.5% and F1-score of 0.995 on the OCT images dataset.
What problem does this paper attempt to address?