Generalizable Machine Learning in Neuroscience Using Graph Neural Networks

Wang, Paul Y. and Sapra, Sandalika and George, Vivek Kurien and Silva, Gabriel A. (2021) Generalizable Machine Learning in Neuroscience Using Graph Neural Networks. Frontiers in Artificial Intelligence, 4. ISSN 2624-8212

[thumbnail of pubmed-zip/versions/4/package-entries/frai-04-618372.pdf] Text
pubmed-zip/versions/4/package-entries/frai-04-618372.pdf - Published Version

Download (2MB)

Abstract

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. As one of the only species for which neuron-level dynamics can be recorded, C. elegans serves as the ideal organism for designing and testing models bridging recent advances in deep learning and established concepts in neuroscience. 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 favourable 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.

Item Type: Article
Subjects: Oalibrary Press > Multidisciplinary
Depositing User: Managing Editor
Date Deposited: 02 Jan 2023 11:47
Last Modified: 23 Mar 2024 04:18
URI: http://asian.go4publish.com/id/eprint/1083

Actions (login required)

View Item
View Item