arXiv:2604.15998cs.CL2026-04

通过对比学习提升细粒度分类能力,解决少样本层级文本分类难题

SCHK-HTC: Sibling Contrastive Learning with Hierarchical Knowledge-Aware Prompt Tuning for Hierarchical Text Classification

论文配图:SCHK-HTC: Sibling Contrastive Learning with Hierarchical Knowledge-Aware Prompt Tuning for Hierarchical Text Classification
图 1 · 摘自论文原文
  • 引入层级知识提取与兄弟类对比学习机制
  • 在三个数据集上超越现有最佳方法,显著提升分类准确率
  • 适合需要细粒度语义区分的少样本文本分类任务

少样本层级文本分类(few-shot HTC)是在数据稀缺条件下将文本映射到预定义树状标签层次结构的挑战性任务。现有方法虽利用标签层次结构保持父子预测一致性,但难以区分语义相近的兄弟类别,因缺乏领域知识。本文提出SCHK-HTC:一种基于层级知识感知提示调优的兄弟对比学习方法。该方法通过两个核心模块——层级知识提取模块与兄弟对比学习机制——增强模型在深层级别对兄弟类别细微差别的感知能力,而非仅依赖层级规则。该设计使模型在各层级编码更具区分性的特征,从而提升易混淆类别的可分性。实验在三个基准数据集上验证了其优越性能,多数情况下超越现有最先进方法。代码已开源。

原文摘要 · Abstract (English)

Few-shot Hierarchical Text Classification (few-shot HTC) is a challenging task that involves mapping texts to a predefined tree-structured label hierarchy under data-scarce conditions. While current approaches utilize structural constraints from the label hierarchy to maintain parent-child prediction consistency, they face a critical bottleneck, the difficulty in distinguishing semantically similar sibling classes due to insufficient domain knowledge. We introduce an innovative method named Sibling Contrastive Learning with Hierarchical Knowledge-aware Prompt Tuning for few-shot HTC tasks (SCHK-HTC). Our work enhances the model's perception of subtle differences between sibling classes at deeper levels, rather than just enforcing hierarchical rules. Specifically, we propose a novel framework featuring two core components: a hierarchical knowledge extraction module and a sibling contrastive learning mechanism. This design guides model to encode discriminative features at each hierarchy level, thus improving the separability of confusable classes. Our approach achieves superior performance across three benchmark datasets, surpassing existing state-of-the-art methods in most cases. Our code is available at https://github.com/happywinder/SCHK-HTC.

层级分类对比学习少样本

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