atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Depression Detection via Capsule Networks with Contrastive Learning

AAAI 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Dense features and foundation priors · 11 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes DeCapsNet, a hierarchical capsule network with user- and post-level contrastive learning that detects depression from social media posts via PHQ-9 symptom descriptions, trained end-to-end.

Problem

Depression detection from online user posts that is both interpretable and accurate, avoiding black-box models and the two-stage pipelines that depend on costly symptom annotations.

Approach

Extract nine PHQ-9 symptom capsules from risky posts selected by Sentence-BERT similarity to symptom descriptions, route them via dynamic routing into two depression capsules, and train end-to-end with max-margin loss plus user-level and post-level supervised contrastive losses and an orthogonal attention constraint.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add the symptom-diversity loss Lsym (ablation)DeCapsNet · eRisk2018, RSDD, and TRTF1: 0.78 ± 0.07 -> 0.79 ± 0.02; 0.62 ± 0.04 -> 0.64 ± 0.01; 0.75 ± 0.01 -> 0.77 ± 0.01Helped
add user-level contrastive loss Luc (ablation)DeCapsNet · eRisk2018, RSDD, and TRTF1: 0.74 ± 0.02 -> 0.79 ± 0.02; 0.58 ± 0.01 -> 0.64 ± 0.01; 0.72 ± 0.00 -> 0.77 ± 0.01Helped
add post-level contrastive loss Lpc (ablation)DeCapsNet · eRisk2018, RSDD, and TRTF1: 0.76 ± 0.02 -> 0.79 ± 0.02; 0.59 ± 0.01 -> 0.64 ± 0.01; 0.73 ± 0.00 -> 0.77 ± 0.01Helped

Limitations

Experimental setup

Datasets: eRisk2018 RSDD TRT

Benchmarks: eRisk2018 RSDD TRT

Metrics: F1 AUC

Baselines: BERT-CNN Pattern (threshold) Pattern (CNN) PHQ9 (scores) PHQ9 (vectors) PHQ9plus DeCapsNet w/o Lsym DeCapsNet w/o Luc DeCapsNet w/o Lpc

Builds on: Capsule networks (Sabour et al. 2017) Supervised contrastive learning (Khosla et al. 2020) PHQ-9 based depression detection (Nguyen et al. 2022) Sentence-BERT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy