Comparative Analysis of Machine Learning Models for Driver Classification Using Data from Microelectromechanical System Sensors
Abstract
Comparative Analysis of Machine Learning Models for Driver Classification Using Data from Microelectromechanical System Sensors
Incoming article date: 20.12.2024This study presents a comparative analysis of machine learning models used for driver classification based on microelectromechanical system (MEMS) sensor data. The research utilizes the “UAH-DriveSet” open dataset, which includes over 500 minutes of driving data with annotations for aggressive driving events, such as sudden braking, sharp turns, and rapid acceleration. The models evaluated in this study include gradient boosting algorithms, a recurrent neural network and a convolutional neural network. Special attention is given to the impact of data segmentation parameters, specifically window size and overlap, on classification performance using the sliding window method. The effectiveness of each model was assessed based on classification metrics such as accuracy, precision, and F1 score. The results show that gradient boosting “LightGBM” outperforms the other models in terms of accuracy and F1 score, while long short-term memory model demonstrates good performance with time-series data but requires larger datasets for better generalization. Convolutional neural network, while effective for identifying short-term patterns, faced difficulties with class imbalances. This research provides valuable insights into selecting the most appropriate machine learning models for driver behavior classification and offers directions for future work in developing intelligent systems using MEMS sensor data.
Keywords: driver behavior analysis, microelectromechanical system sensors, machine learning, aggressive driving, gradient boosting, recurrent neural networks, convolutional neural networks, sliding window, driver classification