科研之友
微信
新浪微博
Facebook
分享链接
摘要
近年来，有很多基于深度学习的分类模型被提出。由于组织病理图像的尺寸极大，现有方法一般将其切割为很多等尺寸的小图像块（切块），再构建分类模型。模型首先提取各切块的特征，然后采用池化等方法将各切块特征融合为整个病理组织图像的特征表示并对图像进行分类。其中切块特征的提取与后续的特征融合与分类过程互相独立，导致模型无法根据后续的分类结果反馈自适应学习切块特征，因此无法确保得到最有利于图像分类的特征。为解决上述问题，本文提出一种基于切块打分模型的端到端注意力池化的病理图像分类方法。首先基于多示例学习方法构造一个切块打分模型对每个切块进行打分，根据得分选择部分切块；然后再采用注意力池化机制融合被选的部分切块的特征构建分类模型。同时，根据病理图像特点，提出在注意力池化机制中采用自定义的平方平均函数进行权重归一化，使阳性病理图像中得分高的切块获得更多关注，让分类结果具有更好的可解释性。在公开的CAMELYON16和BACH数据集上分类的F1分数分别达到了0.644和0.593，结果表明深度学习模型中采用本文提出的端到端注意力池化方法比采用其他池化方法可达到更优的分类性能，表明本文提出方法在病理图像分类应用中的有效性。
- 单位
