Related Experiment Video
Updated: Jun 25, 2026

Artificial Intelligence-Based System for Detecting Attention Levels in Students
Published on: December 15, 2023
Prediction of college student psychological state based on deep learning framework combining the improved Whale
1Shandong Institute of Petroleum and Chemical Technology, Dongying, 257100, Shandong, China.
Abstract:
In intelligent education, accurate prediction of college students' psychological states is crucial for teaching quality improvement and mental health interventions. Psychological state data with temporal dependency and nonlinearity limits traditional machine learning, while Long Short-Term Memory (LSTM) is sensitive to hyperparameters and prone to premature convergence. To solve these problems, this paper proposes a CNWOA-LSTM framework combining an improved Whale Optimization Algorithm (WOA) with LSTM: chaotic initialization is introduced to enhance population diversity and global search ability, and a niche operator is added to alleviate premature convergence; the improved WOA optimizes key LSTM hyperparameters (hidden units, learning rate, dropout rate, batch size). Experiments are conducted on a public kaggle dataset of college students' behavioral features, learning interaction data and psychological assessment indicators. CNWOA-LSTM is compared with traditional machine learning models (SVM, Random Forest, GBDT), classical deep learning models (CNN, Transformer, GRU, BiLSTM), and meta-heuristic optimized LSTM models (DE-LSTM, HHO-LSTM, MPA-LSTM, AOA-LSTM, RSA-LSTM). Results show CNWOA-LSTM achieves 93.64% accuracy, 9.33% and 6.63% higher than original LSTM and standard WOA-LSTM respectively, and 5.21%-7.34% higher than other meta-heuristic optimized LSTM models, verifying its effectiveness for psychological state prediction.