Rethinking Multi-Instance Learning Through Graph-Driven Fusion: A Dual-Path Approach to Adaptive Representation
Yu-Xuan Zhang, Zhengchun Zhou, Weisha Liu, Mingxing Zhang
Abstract
Multi-instance learning (MIL) has become a powerful paradigm for weakly supervised learning tasks, where each sample is a bag of unlabeled instances with only the bag-level label. While graph-based MIL methods enhance bag topological structure modeling, they often suffer from high computation costs and limited representation due to rigid graph construction and insufficient integration of intra-bag semantics. To address these challenges, we propose GDF-MIL, a novel graph-driven MIL framework, which introduces a dual-path feature fusion mechanism to adaptively balance topological structure modeling and semantic feature preservation. First, the adaptive bag mapping module (ABMM) performs soft clustering to extract compact and informative representations. Subsequently, a dynamic graph structure learning (DGSL) component efficiently learns sparse topological structures via weighted connectivity, aggregating them into a comprehensive graph-level representation. Finally, to balance fast graph construction and bag-level knowledge, dual-path feature fusion (DPFF) employs a dual-path gating mechanism to integrate both types of features, which are then passed to the classifier for bag label prediction. Extensive experiments on twenty-four datasets across four domains show that GDF-MIL significantly outperforms eighteen state-of-the-art methods on the majority of datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0c0356f9-fdd4-4717-9be9-aba1e5abe5eeBuilds on11
- TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image ClassificationZhuchen Shao, Hao Bian, Yang Chen, Yifeng Wang et al.NeurIPS 2021 · 1,163 citations
- Multi-Instance Causal Representation Learning for Instance Label Prediction and Out-of-Distribution GeneralizationWeijia Zhang, Xuanhui Zhang, Hanwen Deng, Min-Ling ZhangNeurIPS 2022 · 32 citations
- CAMIL: Context-Aware Multiple Instance Learning for Cancer Detection and Subtyping in Whole Slide ImagesOlga Fourkioti, Matt De Vries, Chris BakalICLR 2024 · 25 citations
- Bag Graph: Multiple Instance Learning Using Bayesian Graph Neural NetworksSoumyasundar Pal, Antonios Valkanas, Florence Regol, Mark CoatesAAAI 2022 · 24 citations
- Agent Aggregator with Mask Denoise Mechanism for Histopathology Whole Slide Image AnalysisXitong Ling, Minxi Ouyang, Yizhi Wang, Xinrui Chen et al.ACM MM 2024 · 18 citations
Related papers
- Dual Graph Disambiguation for Multi-Instance Partial-Label LearningZhen Zhu, Kai Tang, Songhe Feng, Yixuan Tang et al.AAAI 2026
- Predicting Lymph Node Metastasis Using Histopathological Images Based on Multiple Instance Learning With Deep Graph ConvolutionYu Zhao, Fan Yang, Yuqi Fang, Hailing Liu et al.CVPR 2020
- Robust Self-Supervised Multi-Instance Learning with Structure AwarenessYejiang Wang, Yuhai Zhao, Zhengkui Wang, Meixia WangAAAI 2023 · 6 citations
- Multi-graph Multi-label Learning with Dual-granularity LabelingYuhai Zhao, Yejiang Wang, Zhengkui Wang, Chengqi ZhangKDD 2021 · 12 citations
- Loss-Based Attention for Deep Multiple Instance LearningXiaoshuang Shi, Fuyong Xing, Yuanpu Xie, Zizhao Zhang et al.AAAI 2020 · 123 citations
