PuSH - Publikationsserver des Helmholtz Zentrums München

Soberanis-Mukul, R.D.* ; Navab, N.* ; Albarqouni, S.

An uncertainty-driven GCN refinement strategy for organ segmentation.

MELBA 1, Special Issue: Medical Imaging with Deep Learning (MIDL), 1-27 (2020)
Preprint

Organ segmentation in CT volumes is an important pre-processing step in many computerassisted intervention and diagnosis methods. In recent years, convolutional neural networkshave dominated the state of the art in this task. However, since this problem presents achallenging environment due to high variability in the organ’s shape and similarity betweentissues, the generation of false negative and false positive regions in the output segmentationis a common issue. Recent works have shown that the uncertainty analysis of the modelcan provide us with useful information about potential errors in the segmentation. In thiscontext, we proposed a segmentation refinement method based on uncertainty analysisand graph convolutional networks. We employ the uncertainty levels of the convolutionalnetwork in a particular input volume to formulate a semi-supervised graph learning problemthat is solved by training a graph convolutional network. To test our method we refine theinitial output of a 2D U-Net. We validate our framework with the NIH pancreas datasetand the spleen dataset of the medical segmentation decathlon. We show that our methodoutperforms the state-of-the-art CRF refinement method by improving the dice score by1% for the pancreas and 2% for spleen, with respect to the original U-Net’s prediction.Finally, we perform a sensitivity analysis on the parameters of our proposal and discussthe applicability to other CNN architectures, the results, and current limitations of themodel for future work in this research direction. For reproducibility purposes, we make ourcode publicly available athttps://github.com/rodsom22/gcn_refinement.

Impact Factor
Scopus SNIP
0.000
0.000
Tags
Anmerkungen
Besondere Publikation
Auf Hompepage verbergern

Zusatzinfos bearbeiten
Eigene Tags bearbeiten
Privat
Eigene Anmerkung bearbeiten
Privat
Auf Publikationslisten für
Homepage nicht anzeigen
Als besondere Publikation
markieren
Publikationstyp Artikel: Journalartikel
Dokumenttyp Wissenschaftlicher Artikel
Schlagwörter Organ segmentation refinement, Uncertainty Quantification, Graph Convo-lutional Networks, Semi-Supervised Learning
Sprache englisch
Veröffentlichungsjahr 2020
HGF-Berichtsjahr 2020
ISSN (print) / ISBN 2766-905X
e-ISSN 2766-905X
Quellenangaben Band: 1, Heft: 1, Seiten: 1-27, Artikelnummer: , Supplement: Special Issue: Medical Imaging with Deep Learning (MIDL)
Begutachtungsstatus Peer reviewed
Institut(e) Helmholtz AI - HMGU (HAI - HMGU)
POF Topic(s) 30205 - Bioengineering and Digital Health
Forschungsfeld(er) Enabling and Novel Technologies
PSP-Element(e) G-530005-001
Erfassungsdatum 2021-03-16