Rapidly and accurately estimating brain strain and strain rate across head impact types with transfer learning and data fusion

Xianghao Zhan,Yuzhe Liu,Nicholas J. Cecchi,Olivier Gevaert,Michael M. Zeineh,Gerald A. Grant,David B. Camarillo

DOI: https://doi.org/10.1109/TBME.2024.3354192

2021-08-31

Abstract:Brain strain and strain rate are effective in predicting traumatic brain injury (TBI) caused by head impacts. However, state-of-the-art finite element modeling (FEM) demands considerable computational time in the computation, limiting its application in real-time TBI risk monitoring. To accelerate, machine learning head models (MLHMs) were developed, and the model accuracy was found to decrease when the training/test datasets were from different head impacts types. However, the size of dataset for specific impact types may not be enough for model training. To address the computational cost of FEM, the limited strain rate prediction, and the generalizability of MLHMs to on-field datasets, we propose data fusion and transfer learning to develop a series of MLHMs to predict the maximum principal strain (MPS) and maximum principal strain rate (MPSR). We trained and tested the MLHMs on 13,623 head impacts from simulations, American football, mixed martial arts, car crash, and compared against the models trained on only simulations or only on-field impacts. The MLHMs developed with transfer learning are significantly more accurate in estimating MPS and MPSR than other models, with a mean absolute error (MAE) smaller than 0.03 in predicting MPS and smaller than 7 (1/s) in predicting MPSR on all impact datasets. The MLHMs can be applied to various head impact types for rapidly and accurately calculating brain strain and strain rate. Besides the clinical applications in real-time brain strain and strain rate monitoring, this model helps researchers estimate the brain strain and strain rate caused by head impacts more efficiently than FEM.

Machine Learning,Medical Physics,Quantitative Methods,Tissues and Organs

What problem does this paper attempt to address?

The problems that this paper attempts to solve mainly focus on the following aspects: 1. **High computational cost of finite element modeling (FEM)**: The existing FEM methods require a large amount of computational time when calculating brain strain and strain rate, which limits its application in real - time traumatic brain injury (TBI) risk monitoring. 2. **Insufficient generalization ability of machine - learning head models (MLHM)**: The existing MLHM has low prediction accuracy on different types of head - impact datasets, especially performing poorly on small - scale actual on - site impact datasets. This is because there are differences in the data distributions of different types of head - impact data, and the existing MLHM is difficult to adapt to these differences. 3. **Lack of the ability to predict brain strain and strain rate simultaneously**: The existing MLHM mainly focuses on the prediction of brain strain while ignoring the prediction of strain rate. However, the strain rate is also important for predicting traumatic axonal injury (TAI), so it is necessary to develop a model that can predict both simultaneously. To address the above problems, the paper proposes two strategies to optimize MLHM: - **Data Fusion**: Combine simulation data and actual on - site impact data to improve the generalization ability and prediction accuracy of the model. - **Transfer Learning**: Fine - tune the pre - trained MLHM on the actual on - site impact data to make it better adapt to specific types of actual on - site impact data. Through these strategies, the paper aims to develop a series of MLHM that can quickly and accurately predict the maximum principal strain (MPS) and the maximum principal strain rate (MPSR) under different types of head impacts, thereby providing more effective tools for clinical applications and research.

Rapidly and accurately estimating brain strain and strain rate across head impact types with transfer learning and data fusion

Brain Deformation Estimation With Transfer Learning for Head Impact Datasets Across Impact Types

Evaluating Head Impact Intensities and Accelerations Using Flexible Wearable Sensors for Traumatic Brain Injury Assessment

Deep Learning Head Model for Real-time Estimation of Entire Brain Deformation in Concussion

Adaptive Machine Learning Head Model Across Different Head Impact Types Using Unsupervised Domain Adaptation and Generative Adversarial Networks

Instantaneous Whole-Brain Strain Estimation in Dynamic Head Impact

Toward more accurate and generalizable brain deformation estimators for traumatic brain injury detection with unsupervised domain adaptation

Multi-Directional Dynamic Model for Traumatic Brain Injury Detection

Identification of head impact locations, speeds, and force based on head kinematics

Data-driven decomposition of brain dynamics with principal component analysis in different types of head impacts

Machine-learning-based head impact subtyping based on the spectral densities of the measurable head kinematics

Enhancing Hospital Course and Outcome Prediction in Patients with Traumatic Brain Injury: A Machine Learning Study.

Evaluation of Axonal Strain as a Predictor for Mild Traumatic Brain Injuries Using Finite Element Modeling

Kinematics clustering enables head impact subtyping for better traumatic brain injury prediction

Human Brain Modeling with Its Anatomical Structure and Realistic Material Properties for Brain Injury Prediction

A QCT-Based Nonsegmentation Finite Element Head Model for Studying Traumatic Brain Injury

Brain Injury Prediction for Vulnerable Road Users in Vehicle Accidents Using Mathematical Models

Low-rank representation of head impact kinematics: A data-driven emulator

Analysis of finite element models for head injury investigation: reconstruction of four real-world impacts

Predictive Factors of Kinematics in Traumatic Brain Injury from Head Impacts Based on Statistical Interpretation

A Finite Element Model of Cerebral Vascular Injury for Predicting Microbleeds Location