Rethinking Multi-Instance Learning Through Graph-Driven Fusion: A Dual-Path Approach to Adaptive Representation
Yu-Xuan Zhang, Zhengchun Zhou, Weisha Liu, Mingxing Zhang
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper11
- TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image ClassificationZhuchen Shao, Hao Bian, Yang Chen, Yifeng Wang 等NeurIPS 2021 · 被引用 1,163 次
- Multi-Instance Causal Representation Learning for Instance Label Prediction and Out-of-Distribution GeneralizationWeijia Zhang, Xuanhui Zhang, Hanwen Deng, Min-Ling ZhangNeurIPS 2022 · 被引用 32 次
- CAMIL: Context-Aware Multiple Instance Learning for Cancer Detection and Subtyping in Whole Slide ImagesOlga Fourkioti, Matt De Vries, Chris BakalICLR 2024 · 被引用 25 次
- Bag Graph: Multiple Instance Learning Using Bayesian Graph Neural NetworksSoumyasundar Pal, Antonios Valkanas, Florence Regol, Mark CoatesAAAI 2022 · 被引用 24 次
- Agent Aggregator with Mask Denoise Mechanism for Histopathology Whole Slide Image AnalysisXitong Ling, Minxi Ouyang, Yizhi Wang, Xinrui Chen 等ACM MM 2024 · 被引用 18 次
相关 Paper
- Dual Graph Disambiguation for Multi-Instance Partial-Label LearningZhen Zhu, Kai Tang, Songhe Feng, Yixuan Tang 等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 等CVPR 2020
- Robust Self-Supervised Multi-Instance Learning with Structure AwarenessYejiang Wang, Yuhai Zhao, Zhengkui Wang, Meixia WangAAAI 2023 · 被引用 6 次
- Multi-graph Multi-label Learning with Dual-granularity LabelingYuhai Zhao, Yejiang Wang, Zhengkui Wang, Chengqi ZhangKDD 2021 · 被引用 12 次
- Loss-Based Attention for Deep Multiple Instance LearningXiaoshuang Shi, Fuyong Xing, Yuanpu Xie, Zizhao Zhang 等AAAI 2020 · 被引用 123 次
