| Literature DB >> 33920805 |
Navjodh Singh Dhillon1, Agustinus Sutandi1, Manoj Vishwanath2, Miranda M Lim3,4, Hung Cao2,5, Dong Si1.
Abstract
Traumatic Brain Injury (TBI) is a common cause of death and disability. However, existing tools for TBI diagnosis are either subjective or require extensive clinical setup and expertise. The increasing affordability and reduction in the size of relatively high-performance computing systems combined with promising results from TBI related machine learning research make it possible to create compact and portable systems for early detection of TBI. This work describes a Raspberry Pi based portable, real-time data acquisition, and automated processing system that uses machine learning to efficiently identify TBI and automatically score sleep stages from a single-channel Electroencephalogram (EEG) signal. We discuss the design, implementation, and verification of the system that can digitize the EEG signal using an Analog to Digital Converter (ADC) and perform real-time signal classification to detect the presence of mild TBI (mTBI). We utilize Convolutional Neural Networks (CNN) and XGBoost based predictive models to evaluate the performance and demonstrate the versatility of the system to operate with multiple types of predictive models. We achieve a peak classification accuracy of more than 90% with a classification time of less than 1 s across 16-64 s epochs for TBI vs. control conditions. This work can enable the development of systems suitable for field use without requiring specialized medical equipment for early TBI detection applications and TBI research. Further, this work opens avenues to implement connected, real-time TBI related health and wellness monitoring systems.Entities:
Keywords: electroencephalogram (EEG); machine learning (ML); raspberry pi (RPI); traumatic brain injury (TBI)
Mesh:
Year: 2021 PMID: 33920805 PMCID: PMC8071098 DOI: 10.3390/s21082779
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Figure 1Deployment system hardware setup.
Figure 2Proposed system architecture.
Figure 3Live classification display for the RPi based system (64 s epoch, batch size: 100).
System performance comparison of RPi with HPC with 4 classes using XGBoost.
| Device | PC | RPi | PC | RPi | PC | RPi |
|---|---|---|---|---|---|---|
| Epoch | 16 | 16 | 32 | 32 | 64 | 64 |
| Accuracy | 0.982 | 0.982 | 0.974 | 0.974 | 0.968 | 0.968 |
| Sham Wake | ||||||
| Precision | 0.972 | 0.972 | 0.981 | 0.981 | 0.981 | 0.981 |
| Recall | 0.986 | 0.986 | 0.986 | 0.986 | 0.982 | 0.982 |
| Sham Sleep | ||||||
| Precision | 0.973 | 0.973 | 0.951 | 0.951 | 0.937 | 0.937 |
| Recall | 0.951 | 0.951 | 0.945 | 0.945 | 0.948 | 0.948 |
| mTBI Wake | ||||||
| Precision | 0.989 | 0.989 | 0.961 | 0.961 | 0.951 | 0.951 |
| Recall | 0.990 | 0.990 | 0.961 | 0.961 | 0.934 | 0.934 |
| mTBI Sleep | ||||||
| Precision | 0.998 | 0.998 | 0.997 | 0.997 | 0.989 | 0.989 |
| Recall | 0.997 | 0.997 | 0.995 | 0.995 | 0.990 | 0.990 |
Figure 4Variation of epoch processing time with the number of epochs on RPi (64 s epoch).
Figure 5Variation of accuracy with epoch size.
Figure 6Variation of classification time with epoch length.
Epoch collection and processing time (64 s epoch).
| Number of Epochs | Epoch Collection Time (s) | Epoch Processing Time (s) | Processing Time as % of Collection Time |
|---|---|---|---|
| 1 | 64 | 0.02 | 0.03% |
| 10 | 640 | 0.08 | 0.01% |
| 100 | 6400 | 0.6 | 0.01% |