用多尺度图文对齐提升病理图像少样本分类效果
Few-Shot Learning from Gigapixel Images via Hierarchical Vision-Language Alignment and Modeling
- 构建跨尺度图文节点图,捕捉组织结构层级关系
- 在16样本设置下,宏平均F1最高提升4.1%
- 适合医疗影像少样本学习研究者参考
视觉-语言模型(VLM)近期被整合进多实例学习(MIL)框架,以应对全切片图像(WSIs)的少样本弱监督分类挑战。现有方法常存在两大局限:(1)同一尺度内模态间交互建模不足(如5x与20x);(2)同尺度下视觉与文本模态对齐不够。为此,我们提出HiVE-MIL,一个分层视觉-语言框架,构建统一图结构:(1)粗粒度(5x)与细粒度(20x)视觉/文本节点间的父子连接,以捕捉层级关系;(2)同尺度下视觉与文本节点间的异构内部边。为增强语义一致性,引入两阶段文本引导动态过滤机制,剔除相关性弱的图像块-文本对,并设计分层对比损失,实现跨尺度语义对齐。在TCGA乳腺、肺及肾癌数据集上的实验表明,HiVE-MIL持续优于传统MIL及近期VLM-based MIL方法,在16样本设置下宏平均F1最高提升4.1%。结果证明联合建模层级结构与多模态对齐对高效可扩展地利用有限病理数据具有价值。代码已公开于https://github.com/bryanwong17/HiVE-MIL。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have recently been integrated into multiple instance learning (MIL) frameworks to address the challenge of few-shot, weakly supervised classification of whole slide images (WSIs). A key trend involves leveraging multi-scale information to better represent hierarchical tissue structures. However, existing methods often face two key limitations: (1) insufficient modeling of interactions within the same modalities across scales (e.g., 5x and 20x) and (2) inadequate alignment between visual and textual modalities on the same scale. To address these gaps, we propose HiVE-MIL, a hierarchical vision-language framework that constructs a unified graph consisting of (1) parent-child links between coarse (5x) and fine (20x) visual/textual nodes to capture hierarchical relationships, and (2) heterogeneous intra-scale edges linking visual and textual nodes on the same scale. To further enhance semantic consistency, HiVE-MIL incorporates a two-stage, text-guided dynamic filtering mechanism that removes weakly correlated patch-text pairs, and introduces a hierarchical contrastive loss to align textual semantics across scales. Extensive experiments on TCGA breast, lung, and kidney cancer datasets demonstrate that HiVE-MIL consistently outperforms both traditional MIL and recent VLM-based MIL approaches, achieving gains of up to 4.1% in macro F1 under 16-shot settings. Our results demonstrate the value of jointly modeling hierarchical structure and multimodal alignment for efficient and scalable learning from limited pathology data. The code is available at https://github.com/bryanwong17/HiVE-MIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。