Amortization Transformer for Brain Effective Connectivity Estimation from fMRI Data

NeurostimML: a machine learning model for predicting neurostimulation-induced tissue damage

Objective. The safe delivery of electrical current to neural tissue depends on many factors, yet previous methods for predicting tissue damage rely on only a few stimulation parameters. Here, we report the development of a machine learning approach that could lead to a more reliable method for predicting electrical stimulation-induced tissue damage by incorporating additional stimulation parameters.Approach. A literature search was conducted to build an initial database of tissue response information after electrical stimulation, categorized as either damaging or non-damaging. Subsequently, we used ordinal encoding and random forest for feature selection, and investigated four machine learning models for classification: Logistic Regression, K-nearest Neighbor, Random Forest, and Multilayer Perceptron. Finally, we compared the results of these models against the accuracy of the Shannon equation.Main Results. We compiled a database with 387 unique stimulation parameter combinations collected from 58 independent studies conducted over a period of 47 years, with 195 (51%) categorized as non-damaging and 190 (49%) categorized as damaging. The features selected for building our model with a Random Forest algorithm were: waveform shape, geometric surface area, pulse width, frequency, pulse amplitude, charge per phase, charge density, current density, duty cycle, daily stimulation duration, daily number of pulses delivered, and daily accumulated charge. The Shannon equation yielded an accuracy of 63.9% using akvalue of 1.79. In contrast, the Random Forest algorithm was able to robustly predict whether a set of stimulation parameters was classified as damaging or non-damaging with an accuracy of 88.3%.Significance. This novel Random Forest model can facilitate more informed decision making in the selection of neuromodulation parameters for both research studies and clinical practice. This study represents the first approach to use machine learning in the prediction of stimulation-induced neural tissue damage, and lays the groundwork for neurostimulation driven by machine learning models.

MetaCAE: Causal autoencoder with meta-knowledge transfer for brain effective connectivity estimation

Using machine learning methods to estimate brain effective connectivity networks from functional magnetic resonance imaging (fMRI) data has gradually become one of the hot subjects in the fields of neuroscience. In particular, the encoder-decoder based methods can effectively extract the connections in fMRI time series, which have achieved promising performance. However, these methods generally use Granger causality model, which may identify false directions due to the non-stationary characteristic of fMRI data. Additionally, fMRI datasets have limited sample sizes, which significantly constrains the development of these methods. In this paper, we propose a novel brain effective connectivity estimation method based on causal autoencoder with meta-knowledge transfer, called MetaCAE. The proposed approach employs a causal autoencoder (CAE) to extract causal dependencies from non-stationary fMRI time series, and leverages meta-knowledge transfer to improve the estimation accuracy on small-sample data. More specifically, MetaCAE first employs a temporal convolutional encoder to extract non-stationary temporal information from fMRI time series. Then it uses a structural equation model-based decoder to decode causal relationships between brain regions. Finally, it utilizes a model-agnostic meta-learning method to learn the meta-knowledge of the shared brain effective connectivity among different subjects, and transfers the meta-knowledge to the CAE to enhance its estimation ability on small-sample fMRI data. Comprehensive experiments on both simulated and real-world data demonstrate the efficacy of MetaCAE in estimating brain effective connectivity.

NeurostimML: A machine learning model for predicting neurostimulation-induced tissue damage

Objective

The safe delivery of electrical current to neural tissue depends on many factors, yet previous methods for predicting tissue damage rely on only a few stimulation parameters. Here, we report the development of a machine learning approach that could lead to a more reliable method for predicting electrical stimulation-induced tissue damage by incorporating additional stimulation parameters.

Approach

A literature search was conducted to build an initial database of tissue response information after electrical stimulation, categorized as either damaging or non-damaging. Subsequently, we used ordinal encoding and random forest for feature selection, and investigated four machine learning models for classification: Logistic Regression, K-nearest Neighbor, Random Forest, and Multilayer Perceptron. Finally, we compared the results of these models against the accuracy of the Shannon equation.

Main Results

We compiled a database with 387 unique stimulation parameter combinations collected from 58 independent studies conducted over a period of 47 years, with 195 (51%) categorized as non-damaging and 190 (49%) categorized as damaging. The features selected for building our model with a Random Forest algorithm were: waveform shape, geometric surface area, pulse width, frequency, pulse amplitude, charge per phase, charge density, current density, duty cycle, daily stimulation duration, daily number of pulses delivered, and daily accumulated charge. The Shannon equation yielded an accuracy of 63.9% using a k value of 1.79. In contrast, the Random Forest algorithm was able to robustly predict whether a set of stimulation parameters was classified as damaging or non-damaging with an accuracy of 88.3%.

Significance

This novel Random Forest model can facilitate more informed decision making in the selection of neuromodulation parameters for both research studies and clinical practice. This study represents the first approach to use machine learning in the prediction of stimulation-induced neural tissue damage, and lays the groundwork for neurostimulation driven by machine learning models.

Learning Causal Biological Networks with Parallel Ant Colony Optimization Algorithm

A wealth of causal relationships exists in biological systems, both causal brain networks and causal protein signaling networks are very classical causal biological networks (CBNs). Learning CBNs from biological signal data reliably is a critical problem today. However, most of the existing methods are not excellent enough in terms of accuracy and time performance, and tend to fall into local optima because they do not take full advantage of global information. In this paper, we propose a parallel ant colony optimization algorithm to learn causal biological networks from biological signal data, called PACO. Specifically, PACO first maps the construction of CBNs to ants, then searches for CBNs in parallel by simulating multiple groups of ants foraging, and finally obtains the optimal CBN through pheromone fusion and CBNs fusion between different ant colonies. Extensive experimental results on simulation data sets as well as two real-world data sets, the fMRI signal data set and the Single-cell data set, show that PACO can accurately and efficiently learn CBNs from biological signal data.