Graph neural networks

1 minute read

Published:

Graph neural network (GNN) is used for unstructured data. I created a PPT for the general introduction on GNN.

Blogs to introduce GNN in general

  • https://distill.pub/2021/gnn-intro/

GNN on medical images

We could regard each pixel (or voxel for 3D images) as a node in a graph, and then convert a image data to a graph data. Such conversion is memory efficient to save data and train networks for sparse data (like vessels).

Python packages for GNN

torch_geometric is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs). All the related documentation including public datasets, dataloaders, convolution layers could be found here and here.

I strongly recommend to learn the package from the official tutorials

Different GNN designs and their implementations could be found here

some tips

  • The number of graph convolution layers is normally between 2 and 4. Deeper networks do not lead to better performance.
  • Data augmentation is time-consuming for big graph data.
  • GNN is super faster than CNN (about 10 times faster).
  • You could read some surveys or overview papers to have a complete view about GNN.

Leave a Comment