arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1802.04712cs.LGstat.ML
收藏

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

Attention-based Deep Multiple Instance Learning

University of Amsterdam(阿姆斯特丹大学)

Maximilian Ilse, Jakub M. Tomczak, Max Welling

摘要:

多实例学习(MIL)是监督学习的一种变体,其中单个类标签被分配给一组实例。在本文中,我们国家的MIL问题学习的Bernoulli分布的袋标签的袋标签的概率是完全参数化的神经网络。此外,我们提出了一个基于神经网络的置换不变聚合算子,对应于注意力机制。值得注意的是,所提出的基于注意力的运营商的应用程序提供洞察每个实例的袋标签的贡献。我们的经验表明,我们的方法在基准MIL数据集上实现了与最佳MIL方法相当的性能,并且在基于MNIST的MIL数据集和两个真实组织病理学数据集上优于其他方法,而不会牺牲可解释性。

英文摘要:

Multiple instance learning (MIL) is a variation of supervised learning where a single class label is assigned to a bag of instances. In this paper, we state the MIL problem as learning the Bernoulli distribution of the bag label where the bag label probability is fully parameterized by neural networks. Furthermore, we propose a neural network-based permutation-invariant aggregation operator that corresponds to the attention mechanism. Notably, an application of the proposed attention-based operator provides insight into the contribution of each instance to the bag label. We show empirically that our approach achieves comparable performance to the best MIL methods on benchmark MIL datasets and it outperforms other methods on a MNIST-based MIL dataset and two real-life histopathology datasets without sacrificing interpretability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑