Akhtar Khayyam, Yaseen Muhammad Usman, Imran Muhammad, Khattak Sohaib Bin Altaf, M Nasralla Moustafa
COMSATS University Islamabad, Islamabad, Pakistan.
Prince Sultan University, Riyadh, Saudi Arabia.
PeerJ Comput Sci. 2024 Jun 19;10:e2051. doi: 10.7717/peerj-cs.2051. eCollection 2024.
The convergence of smart technologies and predictive modelling in prisons presents an exciting opportunity to revolutionize the monitoring of inmate behaviour, allowing for the early detection of signs of distress and the effective mitigation of suicide risks. While machine learning algorithms have been extensively employed in predicting suicidal behaviour, a critical aspect that has often been overlooked is the interoperability of these models. Most of the work done on model interpretations for suicide predictions often limits itself to feature reduction and highlighting important contributing features only. To address this research gap, we used Anchor explanations for creating human-readable statements based on simple rules, which, to our knowledge, have never been used before for suicide prediction models. We also overcome the limitation of anchor explanations, which create weak rules on high-dimensionality datasets, by first reducing data features with the help of SHapley Additive exPlanations (SHAP). We further reduce data features through anchor interpretations for the final ensemble model of XGBoost and random forest. Our results indicate significant improvement when compared with state-of-the-art models, having an accuracy and precision of 98.6% and 98.9%, respectively. The F1-score for the best suicide ideation model appeared to be 96.7%.
智能技术与监狱中的预测模型相结合,为彻底改变对囚犯行为的监测提供了一个令人兴奋的机会,能够实现对痛苦迹象的早期检测并有效降低自杀风险。虽然机器学习算法已被广泛用于预测自杀行为,但这些模型的互操作性这一关键方面却常常被忽视。大多数关于自杀预测模型解释的工作往往仅限于特征约简和仅突出重要的促成特征。为了填补这一研究空白,我们使用了基于简单规则创建人类可读语句的锚定解释,据我们所知,这从未被用于自杀预测模型。我们还克服了锚定解释在高维数据集上创建弱规则的局限性,方法是首先借助夏普利值附加解释(SHAP)减少数据特征。我们通过对XGBoost和随机森林的最终集成模型进行锚定解释,进一步减少数据特征。与现有最佳模型相比,我们的结果显示出显著改进,准确率和精确率分别为98.6%和98.9%。最佳自杀意念模型的F1分数似乎为96.7%。
JMIR Public Health Surveill. 2023-2-20
Int J Med Inform. 2022-12
Int J Environ Res Public Health. 2022-3-2
Diagnostics (Basel). 2023-3-13
Artif Intell Med. 2022-10