Ensemble transformer with post-hoc explanations for depression emotion and severity detection

集成Transformer模型结合事后解释,用于检测抑郁情绪和严重程度

阅读:1

Abstract

This study presents an ensemble transformer framework for detecting depression-related emotions and classifying their severity in social media text. It addresses the need for scalable and trustworthy AI solutions in mental health by integrating four transformer models. The DepTformer-XAI-SV model uses a weighted soft-voting mechanism based on validation macro-F1 scores to improve accuracy and incorporates LIME to highlight key linguistic features associated with depression. The framework is evaluated on two benchmark datasets: DepressionEmo, with eight emotion classes, and the merged depression severity detection (MDSD), with four severity levels, both sourced from social media. To address class imbalance, we use class-weighted cross-entropy, stratified k-fold splits, and minority-aware sampling. Results show that the model surpasses individual transformer models and traditional methods, achieving macro-F1 scores of 80.44% for DepressionEmo and 79.88% for MDSD, significantly improving minority class detection. Lastly, a web application has been developed for interactive and interpretable inference.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。