本文已被:浏览 556次 下载 1037次
Received:July 29, 2024 Revised:August 20, 2024
Received:July 29, 2024 Revised:August 20, 2024
中文摘要: 针对当下层级文本分类模型尚未充分利用层级实例的标签信息以及缺乏对类别分布不平衡的处理这两方面问题, 本文提出一种面向标签共现和长尾分布的层级文本分类方法(hierarchical text classification for label co-occurrence and long-tail distribution, LC-LTD), 对基于共享标签的文本全局语义和面向长尾分布的平衡损失函数进行研究. 首先, 设计一种基于共享标签的对比学习目标, 使具有更多共享标签的文本表示在特征空间中的语义距离更近, 引导模型生成具有判别性的语义表征; 其次, 引入分布平衡损失函数替换二进制交叉熵损失, 缓解层级分类固有的长尾分布问题, 提高模型的泛化能力. 在WOS、BGC两个公开数据集上将LC-LTD与当前多个主流模型进行比较, 结果表明所提方法具有更好的分类性能, 更适合处理层级文本分类任务.
Abstract:There are two problems in existing hierarchical text classification model: underutilization of the label information across hierarchical instances, and lack of handling unbalanced label distribution. To solve these problems, this study proposes a hierarchical text classification method for label co-occurrence and long-tail distribution (LC-LTD) to study the global semantic of text based on shared labels and balanced loss function for long-tail distribution. First, a contrastive learning objective based on shared labels is devised to narrow the semantic distance between text representations with more shared labels in feature space and to guide the model to generate discriminative semantic representations. Second, the distribution balanced loss function is introduced to replace binary cross-entropy loss to alleviate the long-tail distribution problem inherent in hierarchical classification, improving the generalization ability of the model. LC-LTD is compared with various mainstream models on WOS and BGC public datasets, and the results show that the proposed method achieves better classification performance and is more suitable for hierarchical text classification.
keywords: hierarchical text classification (HTC) label co-occurrence long-tail distribution (LTD) contrastive learning balanced loss
文章编号: 中图分类号: 文献标志码:
基金项目:
引用文本:
智媛,雷海卫,张斌龙.面向标签共现和长尾分布的层级文本分类.计算机系统应用,2025,34(2):174-182
ZHI Yuan,LEI Hai-Wei,ZHANG Bin-Long.Hierarchical Text Classification for Label Co-occurrence and Long-tail Distribution.COMPUTER SYSTEMS APPLICATIONS,2025,34(2):174-182
智媛,雷海卫,张斌龙.面向标签共现和长尾分布的层级文本分类.计算机系统应用,2025,34(2):174-182
ZHI Yuan,LEI Hai-Wei,ZHANG Bin-Long.Hierarchical Text Classification for Label Co-occurrence and Long-tail Distribution.COMPUTER SYSTEMS APPLICATIONS,2025,34(2):174-182

