Optical Technique, Volume. 50, Issue 1, 112(2024)

AMD subtype classification technique using Self-attention mechanism

YANG Wenyi, CHEN Minghui*, WU Yuquan, QIN Kaibo, and YANG Zhengqi
Author Affiliations
  • [in Chinese]
  • show less

    At present, the retinal optical coherence tomography (OCT) image classification method based on convolutional neural network (CNN) has the problem of unclear identification of small-scale lesion areas, which leads to the difficulty in diagnosing the dry and wet aspects of age-related macular degeneration (AMD), and judging the activity of choroidal neovascularization (CNV), but correct judgment of lesion type is crucial for ophthalmologists to formulate treatment plans. Therefore, a CNN model MobileX-ViT based on the self-attention mechanism is proposed, which combines the traditional convolution layers and self-attention module, and simultaneously extracts the feature information of the shallow network and obtains the global information of the image to improve the performance of the model. Experiments have proved that compared with the classic CNN classification models Inception-V3, ResNet-50, VGG-16 and MobileNeXt, the classification accuracy of the proposed model is increased by 5.6%, 5.3%, 4.5% and 2.8% respectively. The effectiveness of the model is proved, and it provides a new method to solve the problem of unclear identification of small-scale lesion areas in the current classification of retinal OCT images.

    Tools

    Get Citation

    Copy Citation Text

    YANG Wenyi, CHEN Minghui, WU Yuquan, QIN Kaibo, YANG Zhengqi. AMD subtype classification technique using Self-attention mechanism[J]. Optical Technique, 2024, 50(1): 112

    Download Citation

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

    Received: May. 5, 2023

    Accepted: --

    Published Online: Jun. 28, 2024

    The Author Email: Minghui CHEN (cmhui.43@163.com)

    DOI:

    Topics