Atlsmms Network: an Attentional-Bilstm Based Multi-Model Prediction for Smartphone Visual Fatigue

Yunyang Shi,Yan Tu,Lili Wang,Nianfang Zhu
DOI: https://doi.org/10.1016/j.displa.2024.102754
IF: 3.074
2024-01-01
Displays
Abstract:Researching and predicting visual fatigue is beneficial to human eye health while using electronic displays. In recent years, EEG-based prediction models have been trending. However, there is room for more accurate and stable performance in classification. In this study, an AtLSMMs network is proposed, with a multi-model design. In the network, display spectral information and temporal sequences of multi-dimensional signal features are fused with reduced dimension, to classify the three visual fatigue states. To optimize performance, in the single model, a multi-layer structure with bidirectional LSTM and self-attention is designed. To ensemble multi-model outputs, three variants are obtained by adding a dropOut layer, data enhancement of Gaussian noise, and class weight assignment, separately. As a result, the performance of the AtLSMMs network on data from 10 participants improved: 96.56% accuracy, 97.11% weighted precision, 96.56% weighted recall, and 96.83% F1-score on average. To verify the performance on different data, a validation experiment was conducted, and the classification results remained nice (75.20% accuracy, 72.12% weighted precision, 78.47% weighted recall, and 72.12% F1-score).
What problem does this paper attempt to address?