A Selective RAG-Enhanced Hybrid ML-LLM Framework for Efficient and Explainable Fatigue Prediction Using Wearable Sensor Data

Fatigue is a multifactorial phenomenon affecting both physical and psychological performance, particularly in high-stress occupations. Although wearable sensors enable continuous monitoring, conventional machine-learning (ML) models can produce unstable, weakly calibrated, and opaque predictions in real-world settings. To improve reliability and interpretability, we developed a selective Retrieval-Augmented Generation (RAG)-enhanced hybrid ML-LLM framework that integrates the efficiency of ML with the reasoning capability of large language models (LLMs). Using wearable and ecological momentary assessment data from 297 emergency responders (9543 seven-day windows), logistic regression, XGBoost, and LSTM models were trained to classify fatigue levels dichotomized by the median of daily tiredness scores. The LLM was selectively activated only for borderline ML outputs (0.45 ≤ <i>p</i> ≤ 0.55), using symbolic rules and retrieved analog examples. In the uncertainty region, performance improved from 0.556/0.684/0.635/0.659 to 0.617/0.703/0.748/0.725 (accuracy/precision/recall/F1). On the full test set, performance similarly improved from 0.707/0.739/0.918/0.819 to 0.718/0.741/0.937/0.827, with gains confirmed by McNemar's paired comparison test (<i>p</i> < 0.05). SHAP-based ML interpretation and LLM reasoning analyses independently identified short-term sleep duration and heart-rate variability as dominant predictors, providing transparent explanations for model behavior. This framework enhances classification robustness, interpretability, and efficiency, offering a scalable solution for real-world fatigue monitoring.

Authors

Institutions

Publication Details

Journal
Bioengineering
Published
2026-01-03
DOI
https://doi.org/10.3390/bioengineering13010058
Primary Topic
Sleep and Work-Related Fatigue
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A Selective RAG-Enhanced Hybrid ML-LLM Framework for Efficient and Explainable Fatigue Prediction Using Wearable Sensor Data

Hyungjun Seo, Taeyoung Lee, Sujung Yoon, Hwamin Lee et al.
Bioengineering
Sleep and Work-Related Fatigue
article

A Selective RAG-Enhanced Hybrid ML-LLM Framework for Efficient and Explainable Fatigue Prediction Using Wearable Sensor Data

Hyungjun Seo, Taeyoung Lee, Sujung Yoon, Hwamin Lee, Soonho Ha
article en

Abstract

Fatigue is a multifactorial phenomenon affecting both physical and psychological performance, particularly in high-stress occupations. Although wearable sensors enable continuous monitoring, conventional machine-learning (ML) models can produce unstable, weakly calibrated, and opaque predictions in real-world settings. To improve reliability and interpretability, we developed a selective Retrieval-Augmented Generation (RAG)-enhanced hybrid ML-LLM framework that integrates the efficiency of ML with the reasoning capability of large language models (LLMs). Using wearable and ecological momentary assessment data from 297 emergency responders (9543 seven-day windows), logistic regression, XGBoost, and LSTM models were trained to classify fatigue levels dichotomized by the median of daily tiredness scores. The LLM was selectively activated only for borderline ML outputs (0.45 ≤ <i>p</i> ≤ 0.55), using symbolic rules and retrieved analog examples. In the uncertainty region, performance improved from 0.556/0.684/0.635/0.659 to 0.617/0.703/0.748/0.725 (accuracy/precision/recall/F1). On the full test set, performance similarly improved from 0.707/0.739/0.918/0.819 to 0.718/0.741/0.937/0.827, with gains confirmed by McNemar's paired comparison test (<i>p</i> < 0.05). SHAP-based ML interpretation and LLM reasoning analyses independently identified short-term sleep duration and heart-rate variability as dominant predictors, providing transparent explanations for model behavior. This framework enhances classification robustness, interpretability, and efficiency, offering a scalable solution for real-world fatigue monitoring.

BioengineeringVol. 13(1)
Ewha Womans University (KR), Korea University (KR)
National Research Foundation, Ewha Womans University, Ministry of Science and ICT, South Korea
Openalex Percentile: Top 97%
Sleep and Work-Related Fatigue
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.