Attention Network with Forced Recall Feature for Image Segmentation
Author:
Affiliation:

Funding:

National Key Research and Development Program of China(2018YFB1105600);National Natural Science Foundation of China(61671440)

Ethical statement:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
    Abstract:

    To solve the problem of serious imbalance between the foreground and background in medical images and small objects segmentation, we propose an attention network based on Gaussian image pyramid to fuse spatial information and abstract information in the feature decoding stage. In addition, a feature recaller is designed to force the encoder to avoid missing features of the region of interest. Finally, a hybrid loss function composed of classification accuracy and global overlapping terms is employed to deal with the serious imbalance between the foreground and background. The proposed method was validated on a knee articular cartilage dataset and the COVOID-19 chest CT dataset where the foreground proportions are 2.08% and 10.73%, respectively. The proposed method achieves the highest Dice coefficients on both datasets as compared with U-Net and its state-of-the-art variants, which are 0.884±0.032 and 0.831±0.072, respectively.

    Reference
    Related
    Cited by
Get Citation

WEI Jianhua, LI Jiaying, HUANG Chengjian, et al. Attention Network with Forced Recall Feature for Image Segmentation[J]. Journal of Integration Technology,2020,9(6):59-70

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: November 24,2020
  • Published: