用图神经网络增强视觉变换器,实现病理切片自动分类与描述生成。
GNN-ViTCap: GNN-Enhanced Multiple Instance Learning with Vision Transformers for Whole Slide Image Classification and Captioning
- 通过聚类和注意力机制筛选关键图像块,减少冗余信息。
- 在BreakHis和PatchGastric数据集上分类F1达0.934,生成描述的BLEU-4为0.811。
- 适合医学图像分析、智能病理诊断方向的研究者参考。
组织病理学显微图像的评估对癌症精准诊断与治疗至关重要。全切片图像(WSI)分类与描述生成已成为辅助病理学的重要任务。然而,显微WSI存在冗余图像块及块位置未知等问题,且自动生成病理描述仍具挑战性。为此,本文提出GNN-ViTCap框架,用于从组织病理图像中进行分类与描述生成。首先,视觉特征提取器生成图像块嵌入;随后,通过深度嵌入聚类动态去除冗余块,并利用标量点积注意力机制选取代表性块;构建基于相似度矩阵最近邻连接的图结构,使用图神经网络捕捉局部与全局上下文信息;最后,将聚合后的图像嵌入通过线性层投影至语言模型输入空间,结合描述标记微调大语言模型。在BreakHis与PatchGastric数据集上的实验表明,该方法分类F1得分为0.934,AUC为0.963,描述生成的BLEU-4为0.811,METEOR为0.569,优于现有方法,为显微病理诊断提供可靠高效的解决方案。
原文摘要 · Abstract (English)
Microscopic assessment of histopathology images is vital for accurate cancer diagnosis and treatment. Whole Slide Image (WSI) classification and captioning have become crucial tasks in computer-aided pathology. However, microscopic WSI face challenges such as redundant patches and unknown patch positions due to subjective pathologist captures. Moreover, generating automatic pathology captions remains a significant challenge. To address these issues, we introduce a novel GNN-ViTCap framework for classification and caption generation from histopathological microscopic images. First, a visual feature extractor generates patch embeddings. Redundant patches are then removed by dynamically clustering these embeddings using deep embedded clustering and selecting representative patches via a scalar dot attention mechanism. We build a graph by connecting each node to its nearest neighbors in the similarity matrix and apply a graph neural network to capture both local and global context. The aggregated image embeddings are projected into the language model's input space through a linear layer and combined with caption tokens to fine-tune a large language model. We validate our method on the BreakHis and PatchGastric datasets. GNN-ViTCap achieves an F1 score of 0.934 and an AUC of 0.963 for classification, along with a BLEU-4 score of 0.811 and a METEOR score of 0.569 for captioning. Experimental results demonstrate that GNN-ViTCap outperforms state of the art approaches, offering a reliable and efficient solution for microscopy based patient diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。