A joint 3D UNet-Graph Neural Network-based method for Airway Segmentation from chest CTs
Publikation: Bidrag til bog/antologi/rapport › Konferencebidrag i proceedings › Forskning › fagfællebedømt
Standard
A joint 3D UNet-Graph Neural Network-based method for Airway Segmentation from chest CTs. / Juarez, Antonio Garcia-Uceda; Selvan, Raghavendra; Saghir, Zaigham; Bruijne, Marleen de.
Machine Learning in Medical Imaging: 10th International Workshop, MLMI 2019, Held in Conjunction with MICCAI 2019, Shenzhen, China, October 13, 2019, Proceedings. Springer, 2019. s. 583-591 (Lecture Notes in Computer Science, Bind LNCS 11861).Publikation: Bidrag til bog/antologi/rapport › Konferencebidrag i proceedings › Forskning › fagfællebedømt
Harvard
APA
Vancouver
Author
Bibtex
}
RIS
TY - GEN
T1 - A joint 3D UNet-Graph Neural Network-based method for Airway Segmentation from chest CTs
AU - Juarez, Antonio Garcia-Uceda
AU - Selvan, Raghavendra
AU - Saghir, Zaigham
AU - Bruijne, Marleen de
PY - 2019/8/22
Y1 - 2019/8/22
N2 - We present an end-to-end deep learning segmentation method by combining a 3D UNet architecture with a graph neural network (GNN) model. In this approach, the convolutional layers at the deepest level of the UNet are replaced by a GNN-based module with a series of graph convolutions. The dense feature maps at this level are transformed into a graph input to the GNN module. The incorporation of graph convolutions in the UNet provides nodes in the graph with information that is based on node connectivity, in addition to the local features learnt through the downsampled paths. This information can help improve segmentation decisions. By stacking several graph convolution layers, the nodes can access higher order neighbourhood information without substantial increase in computational expense. We propose two types of node connectivity in the graph adjacency: i) one predefined and based on a regular node neighbourhood, and ii) one dynamically computed during training and using the nearest neighbour nodes in the feature space. We have applied this method to the task of segmenting the airway tree from chest CT scans. Experiments have been performed on 32 CTs from the Danish Lung Cancer Screening Trial dataset. We evaluate the performance of the UNet-GNN models with two types of graph adjacency and compare it with the baseline UNet.
AB - We present an end-to-end deep learning segmentation method by combining a 3D UNet architecture with a graph neural network (GNN) model. In this approach, the convolutional layers at the deepest level of the UNet are replaced by a GNN-based module with a series of graph convolutions. The dense feature maps at this level are transformed into a graph input to the GNN module. The incorporation of graph convolutions in the UNet provides nodes in the graph with information that is based on node connectivity, in addition to the local features learnt through the downsampled paths. This information can help improve segmentation decisions. By stacking several graph convolution layers, the nodes can access higher order neighbourhood information without substantial increase in computational expense. We propose two types of node connectivity in the graph adjacency: i) one predefined and based on a regular node neighbourhood, and ii) one dynamically computed during training and using the nearest neighbour nodes in the feature space. We have applied this method to the task of segmenting the airway tree from chest CT scans. Experiments have been performed on 32 CTs from the Danish Lung Cancer Screening Trial dataset. We evaluate the performance of the UNet-GNN models with two types of graph adjacency and compare it with the baseline UNet.
KW - eess.IV
KW - cs.CV
U2 - 10.1007/978-3-030-32692-0_67
DO - 10.1007/978-3-030-32692-0_67
M3 - Article in proceedings
SN - 978-3-030-32691-3
T3 - Lecture Notes in Computer Science
SP - 583
EP - 591
BT - Machine Learning in Medical Imaging
PB - Springer
T2 - 10th International Workshop on Machine Learning in Medical Imaging
Y2 - 13 October 2019 through 13 October 2019
ER -
ID: 229018707