Generalizable Machine Learning in Neuroscience using Graph Neural Networks

by   Paul Y. Wang, et al.

Although a number of studies have explored deep learning in neuroscience, the application of these algorithms to neural systems on a microscopic scale, i.e. parameters relevant to lower scales of organization, remains relatively novel. Motivated by advances in whole-brain imaging, we examined the performance of deep learning models on microscopic neural dynamics and resulting emergent behaviors using calcium imaging data from the nematode C. elegans. We show that neural networks perform remarkably well on both neuron-level dynamics prediction, and behavioral state classification. In addition, we compared the performance of structure agnostic neural networks and graph neural networks to investigate if graph structure can be exploited as a favorable inductive bias. To perform this experiment, we designed a graph neural network which explicitly infers relations between neurons from neural activity and leverages the inferred graph structure during computations. In our experiments, we found that graph neural networks generally outperformed structure agnostic models and excel in generalization on unseen organisms, implying a potential path to generalizable machine learning in neuroscience.



page 1

page 2


Revisiting Graph Neural Networks: All We Have is Low-Pass Filters

Graph neural networks have become one of the most important techniques t...

Graph Neural Networks in Network Neuroscience

Noninvasive medical neuroimaging has yielded many discoveries about the ...

GOPHER: Categorical probabilistic forecasting with graph structure via local continuous-time dynamics

We consider the problem of probabilistic forecasting over categories wit...

Machine Learning for Scent: Learning Generalizable Perceptual Representations of Small Molecules

Predicting the relationship between a molecule's structure and its odor ...

Thermodynamics-informed graph neural networks

In this paper we present a deep learning method to predict the temporal ...

Using Graph Neural Networks to model the performance of Deep Neural Networks

With the unprecedented proliferation of machine learning software, there...

Scaling up graph homomorphism for classification via sampling

Feature generation is an open topic of investigation in graph machine le...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.