FGAM: A pluggable light-weight attention module for medical image segmentation

Zhongxi Qiu, Yan Hu, Jiayi Zhang, Xiaoshan Chen, Jiang Liu

Research output: Journal PublicationArticlepeer-review

Abstract

Medical image segmentation is fundamental for computer-aided diagnosis or surgery. Various attention modules are proposed to improve segmentation results, which exist some limitations for medical image segmentation, such as large computations, weak framework applicability, etc. To solve the problems, we propose a new attention module named FGAM, short for Feature Guided Attention Module, which is a simple but pluggable and effective module for medical image segmentation. The FGAM tries to dig out the feature representation ability in the encoder and decoder features. Specifically, the decoder shallow layer always contains abundant information, which is taken as a queryable feature dictionary in the FGAM. The module contains a parameter-free activator and can be deleted after various encoder-decoder networks’ training. The efficacy of the FGAM is proved on various encoder-decoder models based on five datasets, including four publicly available datasets and one in-house dataset.

Original languageEnglish
Article number105628
JournalComputers in Biology and Medicine
Volume146
DOIs
Publication statusPublished - Jul 2022
Externally publishedYes

Keywords

  • Attention mechanism
  • Encoder-decoder network
  • Medical image segmentation

ASJC Scopus subject areas

  • Computer Science Applications
  • Health Informatics

Fingerprint

Dive into the research topics of 'FGAM: A pluggable light-weight attention module for medical image segmentation'. Together they form a unique fingerprint.

Cite this