联营
计算机科学
分割
频道(广播)
人工智能
特征(语言学)
代表(政治)
模式识别(心理学)
机器学习
电信
政治学
语言学
政治
哲学
法学
作者
Chi Zhang,Jingben Lu,Liu Yang,Chunguo Li
标识
DOI:10.1016/j.compbiomed.2021.104875
摘要
Channel attention, a channel-wise method often used in computer vision tasks, including liver tumor segmentation tasks, is able to model the channel relationship to augment the representation ability of feature maps. Channel attention could adaptively generate channel-wise responses using global pooling, which aggregates spatial information roughly. Actually, global pooling may introduce the loss of fine information, which is vital for segmentation tasks. Hence, we rethink the problem and propose the channel attention with adaptive global pooling(short for CAAGP), which preserves spatial and fine-grained information for liver tumor segmentation tasks when channel attention is generated. The model consists of three main parts, including improved self-attention, adaptive global pooling and responses generation modules. Self-attention achieves excellent performance in the computing of the spatial attention, while introducing serious calculation and memory burdens. In order to remedy these burdens, we improve self-attention and consider aggregating spatial information from x and y directions respectively. Extensive experiments have been conducted to verify the effectiveness of our proposed method. Our CAAGP outperforms other attention mechanisms significantly in liver tumor segmentation, especially for tumors with small size.
科研通智能强力驱动
Strongly Powered by AbleSci AI