Integrating Explainable Machine Learning and Mixed-Integer Optimization for Personalized Sleep Quality Intervention

TL;DR

This paper presents a personalized framework combining interpretable machine learning and mixed-integer optimization to generate actionable sleep quality interventions based on survey data.

Contribution

It introduces a novel integrated predictive-prescriptive approach that translates sleep risk predictions into personalized behavioral recommendations using SHAP explanations and optimization.

Findings

Achieved high predictive accuracy with F1-score of 0.9544

Generated concise, high-impact behavioral change recommendations

Revealed a trade-off between intervention impact and effort

Abstract

Sleep quality is influenced by a complex interplay of behavioral, environmental, and psychosocial factors, yet most computational studies focus mainly on predictive risk identification rather than actionable intervention design. Although machine learning models can accurately predict subjective sleep outcomes, they rarely translate predictive insights into practical intervention strategies. To address this gap, we propose a personalized predictive-prescriptive framework that integrates interpretable machine learning with mixed-integer optimization. A supervised classifier trained on survey data predicts sleep quality, while SHAP-based feature attribution quantifies the influence of modifiable factors. These importance measures are incorporated into a mixed-integer optimization model that identifies minimal and feasible behavioral adjustments, while modelling resistance to change through a penalty mechanism. The framework achieves strong predictive performance, with a test F1-score of 0.9544 and an accuracy of 0.9366. Sensitivity and Pareto analyses reveal a clear trade-off between expected improvement and intervention intensity, with diminishing returns as additional changes are introduced. At the individual level, the model generates concise recommendations, often suggesting one or two high-impact behavioral adjustments and sometimes recommending no change when expected gains are minimal. By integrating prediction, explanation, and constrained optimization, this framework demonstrates how data-driven insights can be translated into structured and personalized decision support for sleep improvement.