Gastric polyp detection module based on improved attentional feature fusion

BioMedical Engineering OnLine

Research

Changyin Sun¹  Mingchao Liao²  Yun Xie²  Yao Yu² 

[1] School of Artificial Intelligence, Anhui University, Anhui, China;School of Intelligence Science and technology, University of Science and Technology Beijing, Beijing, China;

关键词: Gastric cancer;    Deep learning;    Polyp detection;    Attention module;    Feature fusion;   

DOI  :  10.1186/s12938-023-01130-x

 received in 2022-12-31, accepted in 2023-06-26,  发布年份 2023

来源: Springer

【 摘 要 】

Gastric cancer is a deadly disease and gastric polyps are at high risk of becoming cancerous. Therefore, the timely detection of gastric polyp is of great importance which can reduce the incidence of gastric cancer effectively. At present, the object detection method based on deep learning is widely used in medical images. However, as the contrast between the background and the polyps is not strong in gastroscopic image, it is difficult to distinguish various sizes of polyps from the background. In this paper, to improve the detection performance metrics of endoscopic gastric polyps, we propose an improved attentional feature fusion module. First, in order to enhance the contrast between the background and the polyps, we propose an attention module that enables the network to make full use of the target location information, it can suppress the interference of the background information and highlight the effective features. Therefore, on the basis of accurate positioning, it can focus on detecting whether the current location is the gastric polyp or background. Then, it is combined with our feature fusion module to form a new attentional feature fusion model that can mitigate the effects caused by semantic differences in the processing of feature fusion, using multi-scale fusion information to obtain more accurate attention weights and improve the detection performance of polyps of different sizes. In this work, we conduct experiments on our own dataset of gastric polyps. Experimental results show that the proposed attentional feature fusion module is better than the common feature fusion module and can improve the situation where polyps are missed or misdetected.

【 授权许可】

CC BY   
© The Author(s) 2023

【 图 表 】

40798_2023_622_Article_IEq1.gif

Fig. 2

41512_2023_153_Article_IEq116.gif

Fig. 2

Fig. 1

Fig. 3

Fig. 3

Fig. 4

Fig. 1

12938_2023_1137_Article_IEq32.gif

40517_2023_267_Article_IEq6.gif

40517_2023_267_Article_IEq5.gif

40517_2023_267_Article_IEq4.gif

40517_2023_267_Article_IEq3.gif

Fig. 2

Fig. 2

Fig. 3

Fig. 5

Fig. 8

Fig. 1

41512_2023_153_Article_IEq101.gif

Fig. 2

Fig. 4

Fig. 6

Fig. 3

Fig. 1

Fig. 1

13570_2023_282_Article_IEq9.gif