Graph Neural Network Based Multi-Instance Learning with Graph Structure Learning

Fan Liu,Weidong Liu
DOI: https://doi.org/10.1109/icaibd62003.2024.10604509
2024-01-01
Abstract:Multi-instance learning can make use of the existing weak labeled data to avoid a lot of manual labeling work costs in supervised learning, and has been widely used in text classification, computer-aided diagnosis, and other fields. However, in recent graph neural network based multi-instance learning methods, the quality of the graph is not high, without considering the specific requirements of downstream tasks while constructing the graph. In addition, the network used now is not enough to fully obtain the information of the bags, which limits the performance of the model. In this paper, we propose a graph neural network based multi-instance learning with graph structure learning method, which explicitly introduces the graph structure learning module to improve the quality of the constructed graph, and uses the graph neural network method based on multi-channel and decoupled transformation and propagation to update the node embedding. Experiments on five benchmark datasets demonstrate that our method achieves improvement compared with baselines.
What problem does this paper attempt to address?