Discrete Wavelet Transform for Generative Adversarial Network to Identify Drivers Using Gyroscope and Accelerometer Sensors.

Paper

Rouhollah Ahmadian¹ , Mehdi Ghatee¹, Johan Wahlström^2
1Amirkabir University of Technology, ²University of Exeter

Abstract

Driver identification is a central research area in intelligent transportation systems, with applications in commercial freight transport and usage-based insurance. One way to perform the identification is to use smartphones as the main sensor devices. After extracting features from smartphone-embedded sensors, various machine learning methods can be used to identify the driver. However, the accuracy often degrades as the number of drivers increases. This paper uses a Generative Adversarial Network (GAN) for data augmentation to obtain a driver identification algorithm that maintains excellent performance also when the number of drivers increases. Since GAN diversifies the drivers’ data, it makes it possible to apply the identification algorithm on a larger number of drivers without overfitting. Although GANs are commonly used in image processing for image augmentation, their use for driving signal augmentation is novel. However, GAN’s training on raw driving signals diverges. This challenge is solved by getting the Discrete Wavelet Transform (DWT) on driving signals before feeding to GAN. Our experiments prove the usefulness of GAN model for generating driving signals emanating from DWT on smartphones’ accelerometer and gyroscope signals. After collecting the augmented data, their histograms along the overlapped windows are fed to machine learning methods covered by a Stacked Generalization Method (SGM). The presented hybrid GAN-SGM approach identifies drivers with 97% accuracy, 98% precision, 97% recall, and 97% F1-measure that outperforms standard machine learning methods that utilize features extracted by the statistical, spectral, and temporal approaches.

Pre-Requisits

• To implement the algorithms, we used Scikit-learn version 0.22, Tensorflow version 1.13.2, and Keras version 2.3.0.
• A model selection for hyperparameters tuning is needed to achieve the best accuracy.

Dataset

In this project, to evaluate the methods, the driving data of this dataset has been used.

Citation

If you use this code for your research, please cite our paper.

@article{ahmadian2022discrete,
  title={Discrete wavelet transform for generative adversarial network to identify drivers using gyroscope and accelerometer sensors},
  author={Ahmadian, Rouhollah and Ghatee, Mehdi and Wahlstr{\"o}m, Johan},
  journal={IEEE Sensors Journal},
  volume={22},
  number={7},
  pages={6879--6886},
  year={2022},
  publisher={IEEE}
}