计算机工程

作者：刘波, 李小霞, 秦佳敏, 周颖玥

单位：1. 西南科技大学 信息工程学院, 四川 绵阳 6210002. 特殊环境机器人技术四川省重点实验室, 四川 绵阳 6210003. 四川绵阳四0四医院 消化内科, 四川 绵阳 621000

关键词：块级自注意力,块相关机制,相对位置偏移,块相关矩阵,多类食管癌前病变

基金：国家自然科学基金(62071399); 四川省科技计划(2021YFG0383)

食管内镜图像多类癌前病变区域的类间特征差异小且个体差异大，难以实现高精度分割，使用自注意力机制可提取远距离依赖信息以获取判别性特征，但是计算开销大。为此，提出一种全局相关块级自注意力（GC-BLSA）方法，用于食管癌前病变区域分割。利用块级自注意力对骨干网络的输出特征进行分块，并在多个特征块上引入自注意力机制，降低网络参数量和计算量，采用块相关机制建模每个特征块和整个特征图之间的关系，解决每个特征块在单独使用自注意力时无法提取与全局相关的远距离依赖信息的问题。在此基础上，在块级自注意力模块中引入相对位置偏移对位置信息进行补充，有效提高网络分割精度。实验结果表明，在四分类食管癌数据集上该方法的分割指标mIoU和F1值分别为50.213%和63.786%，相比传统自注意力Non-local模块分别提高3.744和4.297个百分点，参数量和计算量分别下降26.38%和10.62%。

来源：2023年第7期

《计算机工程》期刊编辑部

查看计算机工程杂志2023年第7期