基于注意力的深度多实例学习
基于注意力的深度多实例学习可以应用于广泛的医学成像应用。 在项目“ ”@ ,我在 ICML 2018 论文“Attention-based Deep Multiple Instance Learning”( )中撰写了Keras版本这个 repo 为 Keras 用户分享解决方案。
可以在找到官方 Pytorch 实现。 我使用Tensorflow后端建造它与Keras。 我编写了论文中描述的注意力层,并在结肠图像中进行了 10 倍交叉验证的实验。 我得到了论文中描述的非常接近的平均准确率,可视化结果如下所示。 部分代码来自 。
在训练模型时,我们只使用图像级标签(0 或 1 以查看它是否是癌症图像)。 注意层可以通过仅呈现积极补丁的一小部分子集来提供对决策的解释。
我的实施结果
数据集
结肠癌数据集
已处理的补丁
我把我处理的数据放在这里,你也可以
1