Assessing the relevance of mental health factors in fibromyalgia severity: A data-driven case study using explainable AI

Abstract

Background and objective: Fibromyalgia is a chronic disease that causes pain and affects patients' quality of life. Current treatments focus on pharmacological therapies for pain reduction. However, patients' psychological well-being is also affected, with depression and pain catastrophizing being common. This research addresses the clinicians' need to assess the influence of mental health factors on FM severity compared to pain factors.

Methods: A co-development study between FM clinicians and data scientists analyzed data from 166 FM-diagnosed patients to assess the influence of mental health factors on FM severity in comparison to pain factors. The study used the Polysymptomatic Distress Scale (PDS) and Fibromyalgia Impact Questionnaire (FIQ) as FM severity indicators and collected 15 variables including regarding demographics, pain intensity perceived, and mental health factors. The team used an author's developed framework to identify the optimal FM severity classifier and explainability by selecting a number of features that lead to obtaining the best classification result. Machine learning classifiers employed in the framework were: decision trees, logistic regression, support vector machines, random forests, AdaBoost, extra trees, and RUSBoost. Explainability analyses were conducted using the following explainable AI techniques: SHapley Additive exPlanations (SHAP), Partial Dependence Plot (PDP), and Mean Decrease Impurity (MDI).

Results: A balanced random forest with 6 features achieved the best performance with PDS (AUC_ROC, mean = 0.81, std = 0.07). Being FIQ the target variable, due to the imbalance in FM severity levels, a binary and a multiclass classification approaches were considered achieving the optimal performance, respectively, a logistic regression classifier (AUC_ROC, mean = 0.83, std = 0.08) with 6 selected features, and a random forest (AUC_ROC, mean = 0.91, std = 0.04) with 8 selected features. Next, the explainability analysis determined mental health factors were found to be more relevant than pain perceived factors for FM severity.

Conclusions: This study's findings, validated by clinicians, are potentially aligned with FM international guidelines that promote non-pharmacological interventions such as promoting mental well-being of FM patients.

Keywords: Classification machine learning; Explainable AI; Fibromyalgia severity; Mental health; Pain.