An Adversarial Variational Autoencoder Approach Toward Transfer Learning for mTBI Identification

Subject variability in mild traumatic brain injury (mTBI) data has often been an obstacle in accurate injury-related feature extraction and biomarker identification for successful mTBI diagnosis. To address this issue, we propose an adversarial variational autoencoder model as a novel regularization...

Full description

Saved in:
Bibliographic Details
Published in:International IEEE/EMBS Conference on Neural Engineering (Online) pp. 408 - 411
Main Authors: Salsabilian, Shiva, Najafizadeh, Laleh
Format: Conference Proceeding
Language:English
Published: IEEE 04.05.2021
Subjects:
Biological system modeling
Brain modeling
Calcium
Feature extraction
Neural activity
Neural engineering
Transfer learning
ISSN:1948-3554
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Subject variability in mild traumatic brain injury (mTBI) data has often been an obstacle in accurate injury-related feature extraction and biomarker identification for successful mTBI diagnosis. To address this issue, we propose an adversarial variational autoencoder model as a novel regularization approach to extract subject-invariant representations for transfer learning in mTBI identification. The proposed method consists of a variational autoencoder with an attached adversarial network. The autoencoder attempts to learn the latent space mappings from neural activity, while the adversary network is used in a discriminative setting to detach the subject individuality from the representations. The trained encoder is then transferred to extract the representations from new subject's data. Several classifiers are utilized to classify the extracted representations into two categories of normal and mTBI. To evaluate the performance of the proposed method, recorded cortical activity of GCaMP6s transgenic calcium reporter mice before and after inducing an injury is used. Experimental results on cross-subject transfer learning exhibit the efficiency of the proposed framework by achieving 89.7% classification accuracy, suggesting the feasibility of the proposed method in learning invariant representations in mTBI data.
ISSN:1948-3554
DOI:10.1109/NER49283.2021.9441372