An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation

06/05/2019
by   Roger D. Soberanis-Mukul, et al.
0

Organ segmentation is an important pre-processing step in many computer assisted intervention and computer assisted diagnosis methods. In recent years, CNNs have dominated the state of the art in this task. Organ segmentation scenarios present a challenging environment for these methods due to high variability in shape, similarity with background, etc. This leads to the generation of false negative and false positive regions in the output segmentation. In this context, the uncertainty analysis of the model can provide us with useful information about potentially misclassified elements. In this work we propose a method based on uncertainty analysis and graph convolutional networks as a post-processing step for segmentation. For this, we employ the uncertainty levels of the CNN to formulate a semi-supervised graph learning problem that is solved by training a GCN on the low uncertainty elements. Finally, we evaluate the full graph on the trained GCN to get the refined segmentation. We compare our framework with CRF on a graph-like data representation as refinement strategy.

READ FULL TEXT
research
02/24/2023

3D PETCT Tumor Lesion Segmentation via GCN Refinement

Whole-body PET/CT scan is an important tool for diagnosing various malig...
research
04/22/2018

Semi-supervised User Geolocation via Graph Convolutional Networks

Social media user geolocation is vital to many applications such as even...
research
08/06/2021

Uncertainty-Based Dynamic Graph Neighborhoods For Medical Segmentation

In recent years, deep learning based methods have shown success in essen...
research
09/11/2021

Border-SegGCN: Improving Semantic Segmentation by Refining the Border Outline using Graph Convolutional Network

We present Border-SegGCN, a novel architecture to improve semantic segme...
research
08/06/2019

OD-GCN object detection by knowledge graph with GCN

Classical object detection frameworks lack of utilizing objects' surroun...
research
06/02/2023

Graph Sparsification for GCN Towards Optimal Crop Yield Predictions

In agronomics, predicting crop yield at a per field/county granularity i...
research
11/27/2019

PointRGCN: Graph Convolution Networks for 3D Vehicles Detection Refinement

In autonomous driving pipelines, perception modules provide a visual und...

Please sign up or login with your details

Forgot password? Click here to reset