Maekawa Eduardo, Jensen Esben, van de Ven Pepijn, Mathiasen Kim
Department of Electronic and Computer Engineering, University of Limerick, Limerick, Ireland.
Health Research Institute, University of Limerick, Limerick, Ireland.
Front Psychiatry. 2024 Sep 3;15:1422587. doi: 10.3389/fpsyt.2024.1422587. eCollection 2024.
This study proposes a Bayesian network model to aid mental health specialists making data-driven decisions on suitable treatments. The aim is to create a probabilistic machine learning model to assist psychologists in selecting the most suitable treatment for individuals for four potential mental disorders: Depression, Panic Disorder, Social Phobia, or Specific Phobia.
This study utilized a dataset from 1,094 individuals in Denmark containing socio-demographic details and mental health information. A Bayesian network was initially employed in a purely data-driven approach and was later refined with expert knowledge, referred to as a hybrid model. The model outputted probabilities for each disorder, with the highest probability indicating the most suitable disorder for treatment.
By incorporating expert knowledge, the model demonstrated enhanced performance compared to a strictly data-driven approach. Specifically, it achieved an AUC score of 0.85 vs 0.80 on the test data. Furthermore, we evaluated some cases where the predictions of the model did not match the actual treatment. The symptom questionnaires indicated that these participants likely had comorbid disorders, with the actual treatment being proposed by the model with the second highest probability.
In 90.1% of cases, the hybrid model ranked the actual disorder treated as either the highest (67.3%) or second-highest (22.8%) on the test data. This emphasizes that instead of suggesting a single disorder to be treated, the model can offer the probabilities for multiple disorders. This allows individuals seeking treatment or their therapists to incorporate this information as an additional data-driven factor when collectively deciding on which treatment to prioritize.
本研究提出了一种贝叶斯网络模型,以帮助心理健康专家基于数据做出关于合适治疗方案的决策。目的是创建一个概率机器学习模型,以协助心理学家为患有四种潜在精神障碍(抑郁症、惊恐障碍、社交恐惧症或特定恐惧症)的个体选择最合适的治疗方法。
本研究使用了来自丹麦1094名个体的数据集,其中包含社会人口统计学细节和心理健康信息。贝叶斯网络最初采用纯数据驱动的方法,后来结合专家知识进行了优化,称为混合模型。该模型输出每种障碍的概率,概率最高的表示最适合治疗的障碍。
通过纳入专家知识,该模型与严格的数据驱动方法相比表现出更高的性能。具体而言,在测试数据上,其AUC分数为0.85,而严格数据驱动方法的AUC分数为0.80。此外,我们评估了一些模型预测与实际治疗不匹配的案例。症状问卷表明,这些参与者可能患有共病,实际治疗方案由概率第二高的模型提出。
在90.1%的案例中,混合模型在测试数据上将实际治疗的障碍列为最高(67.3%)或第二高(22.8%)。这强调了该模型不是建议单一的待治疗障碍,而是可以提供多种障碍的概率。这使得寻求治疗的个体或其治疗师在共同决定优先考虑哪种治疗时,可以将此信息作为另一个数据驱动因素纳入考虑。