返回
Improving text classification via computing category correlation matrix from text graph
DOI:10.1016/j.csl.2024.101688.png)
摘要
En 中文
In text classification task, models have shown remarkable accuracy across various datasets. However, confusion often arises when certain categories within the dataset are too similar, causing misclassification of certain samples. This paper proposes an improved method for this problem, through the creation of a three-layer text graph for the corpus, which is used to calculate the Category Correlation Matrix (CCM). Additionally, this paper introduces categoryadaptive contrastive learning for text embedding from the encoder, enhancing the model's ability to distinguish between samples in confusable categories that are easily confused. Soft labels are generated using this matrix to guide the classifier, preventing the model from becoming overconfident with one-hot vectors. The efficacy of this approach was demonstrated through experimental evaluations on three text encoders and six different datasets.
Keyword:
Natural language processing
Text classification
Contrastive learning
Soft label learning
期刊
C
IF:
3.4
论文数:
1.5K
被引数:
2.6K
机构
引用论文
Machine learning for email spam filtering: review, approaches and open research problems
HELIYON
IF3.6
Deep learning, graph-based text representation and classification: a survey, perspectives and challenges深度学习,基于图的文本表示和分类: 综述,观点和挑战
The Potential of Smart Factories and Innovative Industry 4.0 Technologies—A Case Study of Different-Sized Companies in the Furniture Industry in Central Europe
Forests
IF0
A dynamic programming approach to missing data estimation using neural networks使用神经网络进行缺失数据估计的动态规划方法
Cobalt and Copper Composite Oxides as Efficient Catalysts for Preferential Oxidation of CO in H2-Rich Stream钴和铜复合氧化物作为H2-Rich流中CO优先氧化的有效催化剂

