Design of Autoencoder Algorithm for Compression of Lightweight EEG Signals Based on 2-D Rhythm Feature Maps

Abstract

In recent years, with the development of brain science, the value of electroencephalography (EEG) data has become prominent. However, due to the characteristics of real-time transmission and large amounts of data, there is an urgent need for efficient and lightweight EEG compression algorithms. The existing EEG compression methods have many shortcomings, such as limited compression ratio (CR), poor reconstruction signal quality, and too large model scale, which cannot meet the developmental needs of portable wearable EEG detection devices. In this letter, a method of EEG signal compression based on 2-D rhythm feature maps is proposed. Through discrete wavelet transformation (DWT) extraction of the signal rhythm characteristics, the signal is compressed and reconstructed using encoding and reconstruction channels based on an autoencoder network. At the output end of the encoding channel, entropy coding is carried out to further compress the data volume. Through the discussion of several coding algorithms, JPEG2000 is selected as the local optimal coding algorithm. In addition, based on the idea of grouping convolution and void convolution kernel, a lightweight structure is designed to simplify the process of the proposed network and greatly reduce the number of model parameters. Experiments show that, compared with other similar algorithms, the percentage-root-mean-square distortion and mean squared error (MSE) of the proposed algorithm are 14.76% and 2.95%, respectively, at a relatively high CR (CR is about 16). And only 87.9-k parameters are used, which is more suitable for embedded scenarios and wearable devices.