arXiv:2603.14075cs.LG2026-03

通过残差加权与对比学习提升心理状态分类准确率

Enhancing Mental Health Classification with Layer-Attentive Residuals and Contrastive Feature Learning

  • 利用层注意力残差聚合融合多层语义特征
  • 在SWMH数据集上达74.36%准确率,优于专业模型
  • 改善特征空间结构,提升可解释性与区分度

心理状态分类面临多重挑战:症状重叠、情境依赖性强。尽管微调Transformer提升了性能,但标准交叉熵训练易导致特征空间纠缠,未能充分利用模型信息。本文提出新框架,通过两种方法改进表示能力:一是层注意力残差聚合,对各层残差连接进行加权融合,保留高层语义;二是监督对比学习,采用温度缩放与渐进权重策略,增强混淆类别间的几何间距,减少类间重叠。该方法在SWMH基准上取得74.36%的准确率,优于领域专用模型MentalBERT(高出2.2%-3.25%)和MentalRoBERTa(召回率高出2.41点)。结果表明,精心设计的表示几何与层感知残差整合可超越领域自适应预训练,同时提升可解释性。

原文摘要 · Abstract (English)

The classification of mental health is challenging for a variety of reasons. For one, there is overlap between the mental health issues. In addition, the signs of mental health issues depend on the context of the situation, making classification difficult. Although fine-tuning transformers has improved the performance for mental health classification, standard cross-entropy training tends to create entangled feature spaces and fails to utilize all the information the transformers contain. We present a new framework that focuses on representations to improve mental health classification. This is done using two methods. First, \textbf{layer-attentive residual aggregation} which works on residual connections to to weigh and fuse representations from all transformer layers while maintaining high-level semantics. Second, \textbf{supervised contrastive feature learning} uses temperature-scaled supervised contrastive learning with progressive weighting to increase the geometric margin between confusable mental health problems and decrease class overlap by restructuring the feature space. With a score of \textbf{74.36\%}, the proposed method is the best performing on the SWMH benchmark and outperforms models that are domain-specialized, such as \textit{MentalBERT} and \textit{MentalRoBERTa} by margins of (3.25\% - 2.2\%) and 2.41 recall points over the highest achieving model. These findings show that domain-adaptive pretraining for mental health text classification can be surpassed by carefully designed representation geometry and layer-aware residual integration, which also provide enhanced interpretability through learnt layer importance.

心理健康特征学习对比学习Transformer

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