提出可解释的细粒度多标签文本分类框架,提升模型透明度与准确率。
ProtoSiTex: Learning Semi-Interpretable Prototypes for Multi-label Text Classification
- 分两阶段训练:先无监督发现语义一致的原型,再有监督映射标签。
- 在酒店评论数据集上达到最优性能,解释与人工标注高度一致。
- 适合需要透明决策过程的多标签文本应用,如舆情分析、智能客服。
数字平台用户生成文本的快速增长加剧了对可解释细粒度文本分类模型的需求。现有基于原型的方法通常仅在句子或文档层面提供粗粒度解释,且难以处理真实文本的多标签特性。本文提出ProtoSiTex,一种面向细粒度多标签文本分类的半可解释框架。该框架采用双阶段交替训练策略:第一阶段为无监督原型发现,学习语义连贯且多样化的原型;第二阶段为有监督分类,将原型映射至类别标签。通过层次化损失函数,在子句、句子和文档层级间保持一致性,增强可解释性与对齐度。与以往方法不同,ProtoSiTex利用自适应原型和多头注意力捕捉重叠与冲突语义。我们还构建了一个在子句级别标注多标签的酒店评论基准数据集。在该数据集及两个公开基准(二分类与多分类)上的实验表明,ProtoSiTex在性能上达到当前最优,同时提供忠实且符合人类认知的解释,证明其是半可解释多标签文本分类的可靠解决方案。
原文摘要 · Abstract (English)
The rapid growth of user-generated text across digital platforms has intensified the need for interpretable models capable of fine-grained text classification and explanation. Existing prototype-based models offer intuitive explanations but typically operate at coarse granularity (sentence or document level) and fail to address the multi-label nature of real-world text classification. We propose ProtoSiTex, a semi-interpretable framework designed for fine-grained multi-label text classification. ProtoSiTex employs a dual-phase alternate training strategy: an unsupervised prototype discovery phase that learns semantically coherent and diverse prototypes, and a supervised classification phase that maps these prototypes to class labels. A hierarchical loss function enforces consistency across subsentence, sentence, and document levels, enhancing interpretability and alignment. Unlike prior approaches, ProtoSiTex captures overlapping and conflicting semantics using adaptive prototypes and multi-head attention. We also introduce a benchmark dataset of hotel reviews annotated at the subsentence level with multiple labels. Experiments on this dataset and two public benchmarks (binary and multi-class) show that ProtoSiTex achieves state-of-the-art performance while delivering faithful, human-aligned explanations, establishing it as a robust solution for semi-interpretable multi-label text classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。