Zoom in to where it matters: a hierarchical graph based model for mammogram analysis
arXiv:1912.07517
Abstract
In clinical practice, human radiologists actually review medical images with high resolution monitors and zoom into region of interests (ROIs) for a close-up examination. Inspired by this observation, we propose a hierarchical graph neural network to detect abnormal lesions from medical images by automatically zooming into ROIs. We focus on mammogram analysis for breast cancer diagnosis for this study. Our proposed network consist of two graph attention networks performing two tasks: (1) node classification to predict whether to zoom into next level; (2) graph classification to classify whether a mammogram is normal/benign or malignant. The model is trained and evaluated on INbreast dataset and we obtain comparable AUC with state-of-the-art methods.
References in corpus (4)
- Semi-Supervised Classification with Graph Convolutional Networks
- Disease Prediction using Graph Convolutional Networks: Application to Autism Spectrum Disorder and Alzheimer's Disease
- FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling
- Deep Vessel Segmentation By Learning Graphical Connectivity
Cited by in corpus (5)
- Graph-Based Deep Learning for Medical Diagnosis and Analysis: Past, Present and Future
- Multimodal Data Integration for Oncology in the Era of Deep Neural Networks: A Review
- RA-GCN: Graph Convolutional Network for Disease Prediction Problems with Imbalanced Data
- GKD: Semi-supervised Graph Knowledge Distillation for Graph-Independent Inference
- U-GAT: Multimodal Graph Attention Network for COVID-19 Outcome Prediction