Dynamic Graph Representation with Knowledge-aware Attention for Histopathology Whole Slide Image Analysis

Jiawen Li,Yuxuan Chen,Hongbo Chu,Sun Qiehe,Tian Guan,Anjia Han,Yonghong He
DOI: https://doi.org/10.1109/cvpr52733.2024.01076
2024-01-01
Computer Vision and Pattern Recognition
Abstract:Histopathological whole slide images (WSIs) classification has become afoundation task in medical microscopic imaging processing. Prevailingapproaches involve learning WSIs as instance-bag representations, emphasizingsignificant instances but struggling to capture the interactions betweeninstances. Additionally, conventional graph representation methods utilizeexplicit spatial positions to construct topological structures but restrict theflexible interaction capabilities between instances at arbitrary locations,particularly when spatially distant. In response, we propose a novel dynamicgraph representation algorithm that conceptualizes WSIs as a form of theknowledge graph structure. Specifically, we dynamically construct neighbors anddirected edge embeddings based on the head and tail relationships betweeninstances. Then, we devise a knowledge-aware attention mechanism that canupdate the head node features by learning the joint attention score of eachneighbor and edge. Finally, we obtain a graph-level embedding through theglobal pooling process of the updated head, serving as an implicitrepresentation for the WSI classification. Our end-to-end graph representationlearning approach has outperformed the state-of-the-art WSI analysis methods onthree TCGA benchmark datasets and in-house test sets. Our code is available athttps://github.com/WonderLandxD/WiKG.
What problem does this paper attempt to address?