Counting Network for Learning from Majority Label
Abstract: The paper proposes a novel problem in multi-class Multiple-Instance Learning (MIL) called Learning from the Majority Label (LML). In LML, the majority class of instances in a bag is assigned as the bag's label. LML aims to classify instances using bag-level majority classes. This problem is valuable in various applications. Existing MIL methods are unsuitable for LML due to aggregating confidences, which may lead to inconsistency between the bag-level label and the label obtained by counting the number of instances for each class. This may lead to incorrect instance-level classification. We propose a counting network trained to produce the bag-level majority labels estimated by counting the number of instances for each class. This led to the consistency of the majority class between the network outputs and one obtained by counting the number of instances. Experimental results show that our counting network outperforms conventional MIL methods on four datasets The code is publicly available at https://github.com/Shiku-Kaito/Counting-Network-for-Learning-from-Majority-Label.
- “Is object localization for free? - weakly-supervised learning with convolutional neural networks,” in Computer Vision and Pattern Recognition, 2015, pp. 685–694.
- “Seed, expand and constrain: Three principles for weakly-supervised image segmentation,” in European Conference on Computer Vision. Springer, 2016.
- “Weakly-supervised semantic segmentation network with deep seeded region growing,” in Conference on Computer Vision and Pattern Recognition, 2018, pp. 7014–7023.
- “Self-supervised equivariant attention mechanism for weakly supervised semantic segmentation,” in Computer Vision and Pattern Recognition, 2020.
- “Transmil: Transformer based correlated multiple instance learning for whole slide image classification,” Neural Information Processing Systems, vol. 34, pp. 2136–2147, 2021.
- “Additive mil: Intrinsic interpretability for pathology,” in arXiv preprint arXiv:2206.01794, 2022.
- “Transformer based multiple instance learning for weakly supervised histopathology image segmentation,” in Medical Image Computing and Computer-Assisted Intervention. Springer, 2022, pp. 160–170.
- “Multi instance neural networks,” in International Conference on Machine Learning workshop, 2000, pp. 53–60.
- “Revisiting multiple instance neural networks,” Pattern Recognition, vol. 74, pp. 15–24, 2018.
- “Deep multiple instance learning for digital histopathology,” in Handbook of Medical Image Computing and Computer Assisted Intervention. 2020, pp. 521–546, Academic Press.
- P. O. Pinheiro and R. Collobert, “From image-level to pixel-level labeling with convolutional networks,” in Computer Vision and Pattern Recognition, 2015, pp. 1713–1721.
- “Attention-based deep multiple instance learning,” in International Conference on Machine Learning. PMLR, 2018, pp. 2127–2136.
- “Kernel self-attention for weakly-supervised image classification using deep multiple instance learning,” in Winter Conference on Applications of Computer Vision, 2021, pp. 1721–1730.
- “Loss-based attention for deep multiple instance learning,” in Association for the Advancement of Artificial Intelligence, 2020, vol. 34, pp. 5742–5749.
- “Multi-instance multi-label learning,” Artificial Intelligence, vol. 176, no. 1, pp. 2291–2320, 2012.
- “Miml-fcn+: Multi-instance multi-label learning via fully convolutional networks with privileged information,” in Computer Vision and Pattern Recognition, 2017, pp. 1577–1585.
- Ji Feng and Zhi-Hua Zhou, “Deep miml network,” in Association for the Advancement of Artificial Intelligence, 2017, vol. 31.
- “Learning multiple layers of features from tiny images,” 2009.
- “Reading digits in natural images with unsupervised feature learning,” 2011.
- “Medmnist v2-a large-scale lightweight benchmark for 2d and 3d biomedical image classification,” Scientific Data, vol. 10, no. 1, pp. 41, 2023.
- “Pytorch: An imperative style, high-performance deep learning library,” in Neural Information Processing Systems, 2019.
- “Deep residual learning for image recognition,” in Computer Vision and Pattern Recognition, 2016, pp. 770–778.
- Kingma Diederik P and Ba Jimmy, “Adam: A method for stochastic optimization,” in arXiv, 2014.
Paper Prompts
Sign up for free to create and run prompts on this paper.