Research map / Graph learning / Graph self-supervised learning
A Simple Graph Contrastive Learning Framework for Short Text Classification
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 SimSTC, a graph contrastive learning framework for short text classification that builds word, POS, and entity component graphs and contrasts their resulting text embeddings directly, without data augmentation.
Problem
Short text classification suffers from semantic sparsity and label scarcity; existing graph+contrastive STC models rely on explicit data augmentation that corrupts semantics and only contrast two views.
Approach
Build three text-related component graphs (word, POS, entity) with GCNs; the per-view text embeddings form natural contrastive views, so multi-view contrastive learning is applied across all three views without any data augmentation, with a theoretical mutual-information lower bound justification.
Key results
- SimSTC reaches 73.51% ACC on Twitter, 66.52% on MR, 85.14% on Snippets and 84.07% on StackOverflow, best among non-LLM baselines (GIFT is second-best).
- On Snippets and StackOverflow SimSTC outperforms fine-tuned GPT-3.5, Bloom-7.1B, Llama2-7B and Llama3-8B; LLMs are better on Twitter and MR (e.g., Llama3-8B 87.65% on Twitter).
- Ablation shows each added contrastive view improves accuracy over the no-contrast baseline; using all three views gives the best results, and SimSTC trains about 1.5-4.7% of the parameters of the 7-8B LLMs.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| add the Word-POS contrastive view (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 72.59; 61.94 -> 64.18; 79.27 -> 83.63; 80.29 -> 82.26 | Helped |
| add the POS-Entity contrastive view (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 71.92; 61.94 -> 62.99; 79.27 -> 82.96; 80.29 -> 82.15 | Helped |
| add the Word-Entity contrastive view (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 72.56; 61.94 -> 63.95; 79.27 -> 83.66; 80.29 -> 82.20 | Helped |
| add the Word-POS and POS-Entity contrastive views (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 72.90; 61.94 -> 65.20; 79.27 -> 84.52; 80.29 -> 82.36 | Helped |
| add the Word-POS and Word-Entity contrastive views (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 73.15; 61.94 -> 65.76; 79.27 -> 84.66; 80.29 -> 82.55 | Helped |
| add the POS-Entity and Word-Entity contrastive views (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 73.29; 61.94 -> 65.29; 79.27 -> 84.79; 80.29 -> 82.49 | Helped |
| add all three contrastive views (ablation) | SimSTC without contrastive views · Twitter, MR, Snippets, StackOverflow | ACC: 70.60 -> 73.51; 61.94 -> 66.52; 79.27 -> 85.14; 80.29 -> 84.07 | Helped |
| use SimSTC instead of GIFT | Short-text classification baselines · Twitter, MR, Snippets, StackOverflow | ACC: 73.16 -> 73.51; 65.21 -> 66.52; 83.73 -> 85.14; 83.07 -> 84.07 | Helped |
| use SimSTC instead of Llama3-8B | Short-text classification · Snippets, StackOverflow | ACC: 75.25 -> 85.14; 82.99 -> 84.07 | Helped |
| use SimSTC instead of Llama3-8B | Short-text classification · Twitter, MR | ACC: 87.65 -> 73.51; 87.36 -> 66.52 | Hurt |
Surprising findings
- Fine-tuned LLMs (Llama3-8B, Llama2-7B, Bloom-7.1B) substantially outperform the proposed model on Twitter and MR binary sentiment tasks.
Limitations
- LLM baselines are fine-tuned with LoRA and 4-bit quantization, and the GPT-3.5 comparison relies on a closed API, so the comparison may not be fully controlled.
- Mutual-information theorem is loosely stated (N described as number of datasets, though it is number of texts), and the bound depends on sufficient optimization of the contrastive loss.
- Method relies on external NELL knowledge base and POS/entity tagging pipelines, and does not clearly beat LLMs on binary sentiment datasets.
Experimental setup
Datasets: Twitter MR Snippets StackOverflow NELL knowledge base
Benchmarks: Twitter MR Snippets StackOverflow
Metrics: accuracy macro-F1
Baselines: CNN LSTM BERT RoBERTa TextING DADGNN TextGCN SHINE GIFT GPT-3.5 Bloom-7.1B Llama2-7B
Builds on: GCN TextGCN contrastive learning NELL knowledge base
Closest papers by meaning
- Improved Graph Contrastive Learning for Short Text Classification (AAAI 2024 · similarity 0.94)
- Contrastive Learning with Simplicial Convolutional Networks for Short-Text Classification (ICML 2025 · similarity 0.92)
- Boosting Short Text Classification with Multi-Source Information Exploration and Dual-Level Contrastive Learning (AAAI 2025 · similarity 0.91)
- ReGCL: Rethinking Message Passing in Graph Contrastive Learning (AAAI 2024 · similarity 0.86)
- Perfect Alignment May be Poisonous to Graph Contrastive Learning (ICML 2024 · similarity 0.85)
- Weakly-Supervised Contrastive Learning for Imprecise Class Labels (ICML 2025 · similarity 0.85)
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.
