Data Compression for an Edge-based Machine Learning Model in IoT Devices

Abstract

In this project, an IoT device, comprising an AD8232 ECG sensor kit and ESP32 microcontroller, was designed and developed to collect ECG data. Compression using a wavelet transform was then applied on the collected data prior to transmission giving a compression ratio of 12.8. The transmitted data was then rebuilt on the edge and processed using a deep learning sequential model. The model had training and validation accuracies of 97.1% and 96.5% respectively. In the aforementioned approach, we considered the context of driving behavior monitoring in where ECG signals data are collected from the driver using the developed IoT device and sent to an edge node for stress level detection. The results showed that the amount of transmitted data was reduced and this did not affect the driver stress level prediction accuracy.