A Synergic Neural Network For Medical Image Classification Based On Attention Mechanism

被引:1
|
作者
Wang Shanshan [1 ]
Zhang Tao [2 ]
Li Fei [2 ]
Ruan ZhenPing [2 ]
Yang Zhen [2 ]
Zhan Shu [1 ]
Zhang ZhiQiang [2 ]
机构
[1] Hefei Univ Technol, Sch Comp Sci & Informat Engn, Hefei, Anhui, Peoples R China
[2] Anhui Med Univ, Hosp 2, Hefei, Anhui, Peoples R China
关键词
Medical Image Classification; Attention Mechanism; Synergic Network;
D O I
10.1109/CACML55074.2022.00022
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
For current research in using the deep learning method analyzing and processing medical image in CT or MRI format, identifying the category is challenging because the pathology of the medical images can only be distinguished by tiny difference since the medical image of human organ or tissue is highly homogeneous. The purpose of this paper is to design a convolutional neural network model to extract medical image features and establish a model to classify the medical images. We designed a synergic network based on attention mechanism, capturing the intra-class differences and similarities between classes. The whole pipeline consists of three parts: the input layer, attention based convolutional block and synergic network. By combining these three components, we trained an end-to-end model to classify the medical image. Moreover, through a carefully designed attention model, the network can adaptively pay attention to the region of interest of medical images, avoiding using expensive annotations like bounding boxes or part information labeling. We compare our model on the ADNI dataset(part of) and MRNet dataset with the current SOTA method; the results show our method achieves SOTA performance.
引用
收藏
页码:82 / 87
页数:6
相关论文
共 50 条
  • [1] Hyperspectral Image Classification Based on Convolution Neural Network with Attention Mechanism
    Chen Wenhao
    Jing, He
    Gang, Liu
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2022, 59 (18)
  • [2] A Neural Network Based Text Classification with Attention Mechanism
    Lu SiChen
    [J]. PROCEEDINGS OF 2019 IEEE 7TH INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE AND NETWORK TECHNOLOGY (ICCSNT 2019), 2019, : 333 - 338
  • [3] An Evolutionary Attention-Based Network for Medical Image Classification
    Zhu, Hengde
    Wang, Jian
    Wang, Shui-Hua
    Raman, Rajeev
    Gorriz, Juan M.
    Zhang, Yu-Dong
    [J]. INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2023, 33 (03)
  • [4] A Visual Attention Based Convolutional Neural Network for Image Classification
    Chen, Yaran
    Zhao, Dongbin
    Lv, Le
    Li, Chengdong
    [J]. PROCEEDINGS OF THE 2016 12TH WORLD CONGRESS ON INTELLIGENT CONTROL AND AUTOMATION (WCICA), 2016, : 764 - 769
  • [5] Hyperspectral Image Classification of Brain-Inspired Spiking Neural Network Based on Attention Mechanism
    Liu, Yang
    Cao, Kejing
    Wang, Ruiyi
    Tian, Meng
    Xie, Yi
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [6] A Convolutional Neural Network for Brain CT Image Classification Based on Residual Hybrid Attention Mechanism
    Qiao, Si-Bo
    Pang, Shan-Chen
    Wang, Min
    Zhai, Xue
    Yu, Shi-Hang
    Ding, Tong
    [J]. Tien Tzu Hsueh Pao/Acta Electronica Sinica, 2021, 49 (05): : 984 - 991
  • [7] A Hyperspectral Image Classification Method Based on the Nonlocal Attention Mechanism of a Multiscale Convolutional Neural Network
    Li, Mingtian
    Lu, Yu
    Cao, Shixian
    Wang, Xinyu
    Xie, Shanjuan
    [J]. SENSORS, 2023, 23 (06)
  • [8] Rapid Image Classification is Realized by Spiking Neural Network based on Attention Mechanism and Parallel Neurons
    Qu, Haicheng
    Wang, Wanzhong
    [J]. PROCEEDINGS OF THE 5TH INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE IN ELECTRONICS ENGINEERING, AIEE 2024, 2024, : 35 - 41
  • [9] Neural architecture search based on dual attention mechanism for image classification
    Jin, Cong
    Huang, Jinjie
    Wei, Tianshu
    Chen, Yuanjian
    [J]. MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (02) : 2691 - 2715
  • [10] Hyperspectral Image Classification Based on Dilated Convolutional Attention Neural Network
    Zhang Xiangdong
    Wang Tengjun
    Zhu Shaojun
    Yang Yun
    [J]. ACTA OPTICA SINICA, 2021, 41 (03)