Deep co-supervision and attention fusion strategy for automatic COVID-19 lung infection segmentation on CT images

作者:

Highlights:

• An ESM is presented to highlight low-level boundary features, and the edge supervised information is incorporated into the initial stage of down-sampling.

• An ASSM is proposed to enhance high-level semantics from feature maps with different scales, and the mask supervised information is introduced into the later stage of down-sampling.

• An AFM is developed to fuse various scale feature maps from the up-sampling stage. An attention mechanism is utilized to reduce the semantic gaps between high-level and low-level feature maps, so as to strengthen and supplement the lost detailed information in high-level representations.

• A joint loss function is constructed by combining the edge supervised loss, auxiliary semantic supervised loss and fusion loss, thereby achieving a deep collaborative supervision on edges and semantics.

摘要

•An ESM is presented to highlight low-level boundary features, and the edge supervised information is incorporated into the initial stage of down-sampling.•An ASSM is proposed to enhance high-level semantics from feature maps with different scales, and the mask supervised information is introduced into the later stage of down-sampling.•An AFM is developed to fuse various scale feature maps from the up-sampling stage. An attention mechanism is utilized to reduce the semantic gaps between high-level and low-level feature maps, so as to strengthen and supplement the lost detailed information in high-level representations.•A joint loss function is constructed by combining the edge supervised loss, auxiliary semantic supervised loss and fusion loss, thereby achieving a deep collaborative supervision on edges and semantics.

论文关键词:Semantic segmentation,Multi-scale features,Attention mechanism,Feature fusion,COVID-19

论文评审过程:Received 5 June 2021, Revised 20 September 2021, Accepted 22 November 2021, Available online 25 November 2021, Version of Record 3 December 2021.

论文官网地址:https://doi.org/10.1016/j.patcog.2021.108452