Apr, 2024

基于多头注意力的深度多示例学习

TL;DRMAD-MIL 是一个基于多头注意力机制的深度多实例学习模型,针对数字病理学中弱监督的整张切片图像分类而设计。在模型复杂度简化的同时,MAD-MIL 能够与先进模型如 CLAM 和 DS-MIL 取得竞争性结果,并在 MNIST-BAGS 和公共数据集(包括 TUPAC16、TCGA BRCA、TCGA LUNG 和 TCGA KIDNEY)上对 ABMIL 进行持续超越。该模型展示了切片表示中的信息多样性、可解释性和效率的提升。结合较少的可训练参数和较低的计算复杂度,这使得该模型成为自动化病理工作流程的有希望的解决方案。我们的代码可在此 https URL 找到。