Source-linked AI summary
Graph-Based Recognition of Simulated Train-Driver States From Facial and Upper-Body Keypoints
Olivia Nocentini, Marta Lagomarsino, Gokhan Solak, Younggeol Cho, Qiyi Tong, Sara Zeynalpour, Marta Lorenzini, Alessandro Ledda, Arash Ajoudani
TL;DR
Railway fatigue remains a safety concern, while conventional dead-man switches provide basic alertness checks. The paper uses a single front-facing RGB stream and a graph neural network to classify simulated driver states. The combined facial-and-skeletal representation achieves the reported strongest three-class result in light conditions, alongside high binary alert/not-alert accuracy.
Problem
Railway fatigue is a safety concern, and dead-man switches provide basic checks through periodic pedal presses that can burden drivers.
Source: §I · p. 1Method
A single-stream vision framework uses facial and upper-body landmarks in a graph neural network to classify alert, not alert, and emergency-like states.
Source: §I · p. 3, 1) Network architecture · p. 4Results
81% three-class test accuracy in light condition for the whole-body graph, which also achieved 99% accuracy in light condition for alert/not alert classification.
Source: §IV · p. 6, §VI · p. 9Takeaways & Limitations
The study supports whole-body visual features for simulated train-driver state recognition in a passive, non-contact, non-wearable configuration.
Source: §I · p. 3, §VI · p. 9Both not-alert and emergency classes were simulated in a controlled laboratory setup, so the framework is a preliminary state-recognition study.
Source: §V · p. 7
Abstract
from arXiv · showhide
Driver fatigue poses a significant challenge to railway safety, with traditional systems like the dead-man switch offering limited and basic alertness checks. This study presents a vision-based monitoring system that relies solely on a single front-facing RGB camera and a graph neural network to classify simulated train-driver states into alert, not-alert, and an emergency class comprising acted emergency-like behaviours. To optimize input representations for the model, an ablation study was performed, comparing three feature configurations: skeletal-only, facial-only, and a combination of both. Experimental results show that combining facial and skeletal features yields the highest accuracy (81%) for the three-class model under the light condition, outperforming models that use only facial or skeletal features. Furthermore, the combination of facial and skeletal features achieves 99% accuracy in the alert/not alert classification in light condition. Additionally, we introduced a controlled RGB video dataset containing alert, not alert, and acted emergency-like behaviours recorded under three illumination conditions. These contributions represent a step toward passive and non-contact train-driver state recognition based on facial and upper-body dynamics.
I. INTRODUCTION
Railway fatigue is a safety concern, and the paper identifies limited railway-specific approaches for recognizing driver states from non-contact visual observations. It proposes a vision-only graph framework that combines facial and upper-body dynamics to distinguish alertness states and acted emergency-like behaviours.
- Long shifts, repetitive duties, and nighttime operation raise fatigue concerns, while dead-man switches use periodic pedal checks that burden drivers.The switch can trigger an alarm and may halt the train if the pedal is not pressed within a specified time.Source: §I · p. 1
- Vehicle-based alertness assessment is constrained by intermittent train-control interactions, while physiological sensing can involve wearability, noise, and cost limitations.Source: §I · p. 2
- Prior vision-based fatigue work often emphasizes facial features, while train operators’ wider, periodically varying movements differ from road-driver patterns.Source: §I · p. 2
- The proposed single-stream framework avoids wearables, train telemetry, and verbal confirmation, learning facial and upper-body landmark dynamics in one graph rather than aggregating handcrafted cue signs.The reduced sensing burden may facilitate cabin integration while limiting interference with normal activities.Source: §I · p. 3
- The paper identifies limited railway-specific visual state recognition, especially for distinguishing reduced alertness from acted emergency-like behaviours, and notes illumination variation as a landmark-estimation challenge.Source: §I · p. 3
- The study formulates alert, not alert, and acted emergency-like behaviour as three classes and introduces a controlled dataset with three illumination settings for feature ablation.The ablation compares facial-only, skeletal-only, and combined representations; the emergency class includes behaviours such as pretending to faint or clutching the chest.Source: §I · p. 3
II. METHODOLOGY
The methodology uses RGB video to build facial-and-skeletal graphs for DGNN-based driver-state classification. It also evaluates simulated states across lighting conditions and compares whole-body features with facial-only and skeletal-only inputs.
- RGB images are converted to facial and skeletal keypoints, assembled into a whole-body graph, and processed by a custom DGNN to classify three driver states.Source: §II · p. 3, Figure 1 · p. 4
- The preliminary RGB video dataset covers light, dark, and sudden light/dark conditions and supports ablation and keypoint-occlusion analysis.Source: §II · p. 3
- The study asks whether the graph model detects simulated alertness and emergency states and whether whole-body movements improve three-class performance over facial-only or skeletal-only keypoints.Source: §II · pp. 3–4
A. ESTIMATION OF FACIAL AND SKELETAL KEYPOINTS
The input representation combines upper-body skeleton landmarks with dense facial landmarks, then forms normalized graphs from their coordinates. A masking strategy addresses keypoints that are not consistently detected.
- The method uses Mediapipe’s first 17 upper-body keypoints and adds 478 Face Mesh landmarks to capture facial contours, eyebrows, eye openness, and mouth movements.The upper skeleton is selected because drivers are usually seated; the chosen detectors are described as stable for RGB images.Source: A. ESTIMATION OF FACIAL AND SKELETAL… · p. 4
- A masking strategy handles keypoints that were not consistently detected during extraction from RGB video.Source: A. ESTIMATION OF FACIAL AND SKELETAL… · p. 4
- Facial and upper-body 2D coordinates become graph nodes, bones define edges, and coordinates are normalized relative to the nose across the three input configurations.Source: A. ESTIMATION OF FACIAL AND SKELETAL… · p. 4
1) Network architecture
The customized DGNN combines directed graph processing with temporal convolutions to model landmark interactions over time. Its architecture and training settings are adapted to the dataset, with validation performance guiding hyperparameter tuning.
1) Network architecture
- The customized DGNN uses two graph-temporal convolution blocks, pairing directed graph relational learning with bidirectional temporal convolutions over node and edge features.The architecture processes incoming and outgoing graph connections, then applies temporal convolution along the time dimension.Source: 1) Network architecture · p. 4
- The compact model adds dropout, uses LeakyReLU and label smoothing, then pools learned node and edge features for fully connected output predictions.The authors describe these changes as adaptations for the relatively small dataset and noisy keypoint detections.Source: 1) Network architecture · p. 5
2) Network hyperparameters
- Training used Adam, a learning rate of 0.001, 50 epochs, batch size 512, label smoothing 0.1, and a 30-frame lookback, tuned using validation performance.Participants were seated at a desk and followed on-screen instructions for lever, button, and side-panel tasks while a front-facing RGB camera monitored them.Source: 2) Network hyperparameters · p. 5, Figure 3 · p. 5
C. RELEVANCE OF FACIAL AND SKELETAL KEYPOINTS
The section compares facial, skeletal, and combined keypoint inputs for the DGNN, then uses occlusion sensitivity to assess the importance of individual keypoints.
1) Ablation study
- Three DGNN variants use whole-body, facial-only, or skeletal-only keypoints to isolate the contributions of facial and skeletal information.Whole-body combines facial and skeletal keypoints; the study examines whether integration enhances performance.Source: 1) Ablation study · p. 5
2) Occlusion Sensitivity Analysis
- Occlusion Sensitivity Analysis hides one or more keypoints and assesses changes in predictions or performance to identify important inputs.A substantial degradation after occlusion indicates that the keypoint is particularly important to the network’s decisions.Source: 2) Occlusion Sensitivity Analysis · p. 5
III. EXPERIMENTAL CAMPAIGN
The experimental campaign collected RGB video from participants seated before a front-facing camera and created a controlled dataset of three simulated driver states under three lighting conditions.
A. SETUP
- Participants sat before a computer monitor and a front-facing Realsense D435i camera during data collection.The camera captured RGB video at 30 fps and 640×480 resolution.Source: A. SETUP · pp. 5–6
B. DATASET
- The dataset included 25 participants, divided into training, validation, and test sets of 15, 5, and 5 subjects.Participants were 14 males and 11 females aged 23–36; 12 wore glasses and 13 did not.Source: B. DATASET · p. 6
- Each participant recorded nine 2.5-minute videos covering alert, not-alert, and acted emergency-like states under light, dark, and sudden light/dark conditions.The sudden light/dark condition switched the room lights every 15 seconds; emergency behaviours were acted proxies, not clinically validated medical events.Source: B. DATASET · p. 6
- Alert participants followed train-driving prompts, not-alert participants kept their eyes closed and pretended to sleep, and emergency participants acted behaviours such as fainting or clutching their chest.Some participants covered parts of their faces during not-alert and emergency recordings, introducing self-occlusions.Source: B. DATASET · p. 6
1) Accuracy
The section defines accuracy as the proportion of correct predictions and describes confusion matrices used to evaluate driver-state classifications across input types and lighting conditions.
1) Accuracy
- Accuracy is the proportion of correct predictions among all samples and measures agreement with ground-truth classes.The section notes that accuracy ranges from 0 to 100% and is particularly useful for balanced classification tasks.Source: 1) Accuracy · p. 6
2) Confusion Matrices
- Confusion matrices compare predictions with actual labels for whole-body keypoints across lighting conditions and three- and two-class tasks.The analysis also includes facial-only and skeletal-only scenarios.Source: 2) Confusion Matrices · p. 6
IV. RESULTS
The combined whole-body model performs best among the three-class input configurations, while alert/not-alert classification reaches higher accuracy. Results also vary with lighting, and the confusion-matrix analysis identifies emergency-to-not-alert misclassifications.
IV. RESULTS
- 99% test accuracy was achieved for alert/not-alert classification; adding emergency reduced accuracy, while whole-body outperformed facial-only and skeletal-only models.The reported metrics were computed on five test subjects.Source: §IV · p. 6
- Performance was highest in full light, followed by sudden light/dark conditions across all scenarios.Source: §IV · p. 6
- The three-class whole-body confusion-matrix analysis found that emergency states were sometimes misclassified as not alert.Source: §IV · p. 6, Figure 4 · p. 7
- The keypoint-importance plot represents each keypoint as a bar whose height indicates its significance in distinguishing the three driver states.Source: §IV · p. 7
V. DISCUSSION
The whole-body model performs strongly on binary alertness classification but has more difficulty distinguishing the acted emergency class, especially under challenging illumination. The discussion also identifies limits in the study’s simulated scenarios and graph-topology analysis.
V. DISCUSSION
- 78% overall three-class accuracy was achieved with whole-body keypoints, while skeletal-only or facial-only features were insufficient for appropriate three-class classification, particularly under varying lighting.The study describes whole-body analysis as highly effective for binary driver-alertness classification.Source: §V · p. 7
- Globally, 90.7% of alert and 87.6% of not-alert samples were correctly classified, versus 54.1% of emergency samples, with 34.1% of emergency samples misclassified as not alert.The authors link the emergency/not-alert confusion to similar postures in the two acted conditions.Source: §V · p. 7
- The framework is preliminary evidence from acted scenarios, not a clinically validated detector of real fatigue or medical emergencies; real-world validation and a larger dataset remain future work.Planned dataset expansion includes occlusions and real-world artifacts such as camera vibration.Source: §V · pp. 7, 9
- In light, alert and not-alert recognition reached 94.1% and 93.4%, respectively, while emergency recognition reached 56.2%; in dark, emergency recognition fell to 50.6%.The authors attribute weaker dark-condition performance to RGB-camera and keypoint-estimation limits in low light.Source: §V · p. 8
- Under sudden light/dark changes, 91.8% of alert samples were correctly classified, but 17.1% of not-alert samples and 32.3% of emergency samples were confused across those classes.The authors suggest abrupt illumination changes may add variability to extracted features and hinder that distinction.Source: §V · p. 8
- Occlusion analysis ranked skeletal keypoints, especially the right wrist and both shoulders, among the most influential features, alongside mouth and left-wrist keypoints.Figure 5 presents the overall keypoint ranking; the authors interpret the result as posture cues complemented by facial information.Source: §V · p. 8, Figure 5 · p. 9
- The analysis ranks individual nodes but does not compare alternative graph topologies, root definitions, or connectivity patterns; topology comparison is identified as future work.The authors note that node importance does not isolate the contribution of edge configurations.Source: §V · pp. 8–9
- The study does not make a direct quantitative comparison with prior approaches because datasets, acquisition protocols, class definitions, and evaluation procedures differ.The authors also note that prior studies often use binary classification and more stable illumination settings.Source: §V · p. 9
VI. CONCLUSIONS
The conclusion presents a vision-based monitoring system using a customised DGNN and whole-body movement to classify alert, not-alert, and simulated emergency states. It reports its best performance with a whole-body graph and introduces a controlled RGB dataset.
VI. CONCLUSIONS
- The system uses a customised Directed-Graph Neural Network and whole-body movement to classify alert, not-alert, and simulated emergency conditions.Source: §VI · p. 9
- 81% test accuracy for three-class classification and 99% accuracy for two-class classification were achieved by the whole-body graph in light condition, which had the highest performance among tested configurations.Source: §VI · p. 9
- The study introduces a controlled RGB dataset of alert, not-alert, and acted emergency-like behaviours; planned work includes dataset expansion and thermal or near-infrared sensing for low-light robustness.Source: §VI · p. 9