Dual-Level Hypergraph Generation for Addressing Feature Scarcity in Whole-Slide Image Classification
Shuilian Yao, Qi Jia, Yu Liu, Pengshuo Zhang, Lili Sun, Weimin Wang, Yanmei Zhu, Bo Zhang, Xin Fan
Abstract
Lymph node metastasis diagnosis in pathological images is a highly challenging four-class classification task, comprising macrometastasis, micrometastasis, isolated tumor cells (ITC), and negative lesions. Unlike conventional classification settings, this four-class scenario simultaneously suffers from inter-class and intra-slide scarcity of minority information. Existing approaches based on CNNs or GNNs primarily emphasize node-level feature learning, making it difficult to capture high-order feature interactions and topological dependencies among cells, while also overlooking the representational insufficiency induced by class scarcity. To address these challenges, we propose a dual-level generative framework that integrates class-prompt priors with high-order structural modeling to enhance the representation capacity of minority classes. At the hypergraph level, we develop a prompt-guided hierarchical hypergraph variational autoencoder capable of generating diverse and topologically consistent hypergraph representations for minority classes. At the hypernode level, we introduce an anchordiffusion mixup strategy to enrich the minority node features of high-attention positive anchor nodes. Extensive experiments on the four-class NIMM dataset, as well as TCGA datasets, demonstrate that the proposed framework effectively alleviates feature scarcity and significantly boosts the classification performance of minority classes. The code is
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 47e81a3e-2b6f-4cbb-a72a-b73188fd6bbbBuilds on14
- Conditional Prompt Learning for Vision-Language ModelsKaiyang Zhou, Jingkang Yang, Chen Change Loy, Ziwei LiuCVPR 2022 · 1,438 citations
- 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
- The Rise of AI Language Pathologists: Exploring Two-level Prompt Learning for Few-shot Weakly-supervised Whole Slide Image ClassificationLinhao Qu, Xiaoyuan Luo, Kexue Fu, Manning Wang et al.NeurIPS 2023 · 75 citations
- ViLa-MIL: Dual-scale Vision-Language Multiple Instance Learning for Whole Slide Image ClassificationJiangbo Shi, Chen Li, Tieliang Gong, Yefeng Zheng et al.CVPR 2024 · 38 citations
- Leveraging Tumor Heterogeneity: Heterogeneous Graph Representation Learning for Cancer Survival Prediction in Whole Slide ImagesJunxian Wu, Xinyi Ke, Xiaoming Jiang, Huanwen Wu et al.NeurIPS 2024 · 12 citations
Related papers
- Robust Multimodal Survival Prediction with Conditional Latent Differentiation Variational AutoEncoderJunjie Zhou, Jiao Tang, Yingli Zuo, Peng Wan et al.CVPR 2025
- MAPLE: Multi-scale Attribute-enhanced Prompt Learning for Few-shot Whole Slide Image ClassificationJunjie Zhou, Wei Shao, Yagao Yue, Wei Mu et al.NeurIPS 2025 · 1 citation
- 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
- Topology-Guided Multi-Class Cell Context Generation for Digital PathologyShahira Abousamra, Rajarsi Gupta, Tahsin M. Kurç, Dimitris Samaras et al.CVPR 2023
- GraphSHA: Synthesizing Harder Samples for Class-Imbalanced Node ClassificationWen-Zhi Li, Chang-Dong Wang, Hui Xiong, Jian-Huang LaiKDD 2023 · 37 citations
