arXiv:2606.22780cs.LGcs.AI2026-06中稿 · presenting at the …

用智能算法提升吸毒人群心理健康预测准确率与可解释性。

Explainable AI for Mental Health Prediction in Drug-Affected Populations with Dragonfly Algorithm and GAN Oversampling

论文配图:Explainable AI for Mental Health Prediction in Drug-Affected Populations with Dragonfly Algorithm and GAN Oversampling
图 1 · 摘自论文原文
  • 融合特征选择、GAN增样本与蜻蜓算法优化,解决数据不平衡与高维问题。
  • 模型准确率达94.17%,加权F1-score达93.80%,优于传统方法。
  • 通过SHAP分析揭示睡眠质量等关键因素,适合临床辅助决策使用。

药物滥用人群的心理健康问题日益严重,尤其在早期检测困难的地区。现有研究普遍忽视对吸毒人群的AI心理评估,且存在类别不平衡处理差、模型可解释性低及超参数优化不足等问题。本文提出一种可解释的多分类心理健康预测框架,基于多维度吸毒者数据集,结合混合PCA-信息增益特征选择、GAN增样本与蜻蜓算法(DA)优化的XGBoost模型。实验表明,该框架能有效处理高维分类数据,缓解类别不平衡,并通过智能调参提升预测性能。最优模型准确率达94.17%,加权F1-score为93.80%,显著优于基线模型。特征分析显示,行为、生活方式及健康因素(尤其是睡眠质量、身体状况与情绪调节)是核心预测因子,而人口学特征影响较小。基于SHAP的可解释性分析提供实例级透明洞察,增强临床信任。结果表明该框架具备开发实用心理健康预警工具的潜力,有助于早期干预与治疗优化。

原文摘要 · Abstract (English)

Mental illnesses among drug users are an increasing international issue, particularly in regions where early detection cannot be easily undertaken. The current literature tends to ignore the use of AI-based mental health analysis in drug users, and low quality of the class imbalance treatment, low interpretability, and optimal hyperparameter optimization can lower predictive quality and clinical utility. This study present a detailed, explainable machine learning (ML) model of multiclass mental health prediction, using a multidimensional data set of drug-affected persons. We combine hybrid PCA-Information Gain (PCA-IG) feature selection, Generative Adversarial Network (GAN)-based oversampling, and Dragonfly Algorithm (DA)-optimized XGBoost to address some of the limitations of existing methods. The suggested framework is effective to work with high-dimensional categorical data, address the issue of class imbalance, and improve predictive performance due to intelligent hyperparameter tuning. The experimental findings show that the XGBoost model optimized using the DA, in combination with GAN-based oversampling, has an accuracy of 94.17% and a weighted F1-score of 93.80%, which is better than the traditional and baseline models. The behavioral, lifestyle, and health factors, particularly sleep quality, physical health, and emotional regulation, are strongly predictive of mental health, with demographic factors having little impact, as seen through feature analysis. SHAP-based explainable AI provides easy-to-understand, instance-level information, enhancing interpretability and trust in models to be used in clinical settings. The results indicate that this framework has the potential to generate valid mental health forecasting tools, which would facilitate early intervention and enhance the treatment of drug-influenced people.

心理健康可解释AIGAN优化算法

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。