构建肾小球免疫微环境图神经网络,提升肾病病理分类精度
HIEGNet: A Heterogenous Graph Neural Network Including the Immune Environment in Glomeruli Classification
- 基于图像分析自动构建包含肾小球与免疫细胞的异构图
- 在移植肾活检数据上准确率显著优于基线模型
- 适合病理学研究与数字病理辅助诊断应用
图神经网络(GNN)在组织病理学中表现优异,但在肾小球健康状态分类这一关键任务上仍缺乏深入探索。该任务在图结构构建方面面临挑战,即如何识别节点、边及有效特征。本文提出一套融合传统与机器学习计算机视觉技术的流程,用于构建异构图。进一步设计新型异构GNN架构HIEGNet,整合肾小球及其周围免疫细胞信息,从而考虑其免疫微环境。HIEGNet在来自肾移植患者的全切片图像数据集上训练与测试,实验表明其性能超越多个基线模型,且在患者间泛化能力最强。代码已公开于https://github.com/nklsKrmnn/HIEGNet.git。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have recently been found to excel in histopathology. However, an important histopathological task, where GNNs have not been extensively explored, is the classification of glomeruli health as an important indicator in nephropathology. This task presents unique difficulties, particularly for the graph construction, i.e., the identification of nodes, edges, and informative features. In this work, we propose a pipeline composed of different traditional and machine learning-based computer vision techniques to identify nodes, edges, and their corresponding features to form a heterogeneous graph. We then proceed to propose a novel heterogeneous GNN architecture for glomeruli classification, called HIEGNet, that integrates both glomeruli and their surrounding immune cells. Hence, HIEGNet is able to consider the immune environment of each glomerulus in its classification. Our HIEGNet was trained and tested on a dataset of Whole Slide Images from kidney transplant patients. Experimental results demonstrate that HIEGNet outperforms several baseline models and generalises best between patients among all baseline models. Our implementation is publicly available at https://github.com/nklsKrmnn/HIEGNet.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。