Accurate use of label dependency in multi-label text classification through the lens of causality

被引:3
|
作者
Fan, Caoyun [1 ]
Chen, Wenqing [2 ]
Tian, Jidong [1 ]
Li, Yitian [1 ]
He, Hao [1 ]
Jin, Yaohui [1 ]
机构
[1] Shanghai Jiao Tong Univ, AI Inst, MoE Key Lab Artificial Intelligence, Shanghai, Peoples R China
[2] Sun Yat Sen Univ, Sch Software Engn, Guangzhou, Peoples R China
关键词
Multi-label text classification; Label dependency; Correlation shortcut; Counterfactual de-bias;
D O I
10.1007/s10489-023-04623-3
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Multi-Label Text Classifiction (MLTC) aims to assign the most relevant labels to each given text. Existing methods demonstrate that label dependency can help to improve the model's performance. However, the introduction of label dependency may cause the model to suffer from unwanted prediction bias. In this study, we attribute the bias to the model's misuse of label dependency, i.e., the model tends to utilize the correlation shortcut in label dependency rather than fusing text information and label dependency for prediction. Motivated by causal inference, we propose a CounterFactual Text Classifier (CFTC) to eliminate the correlation bias, and make causality-based predictions. Specifically, our CFTC first adopts the predict-then-modify backbone to extract precise label information embedded in label dependency, then blocks the correlation shortcut through the counterfactual de-bias technique with the help of the human causal graph. Experimental results on three datasets demonstrate that our CFTC significantly outperforms the baselines and effectively eliminates the correlation bias in datasets.
引用
收藏
页码:21841 / 21857
页数:17
相关论文
共 50 条
  • [1] Accurate use of label dependency in multi-label text classification through the lens of causality
    Caoyun Fan
    Wenqing Chen
    Jidong Tian
    Yitian Li
    Hao He
    Yaohui Jin
    Applied Intelligence, 2023, 53 : 21841 - 21857
  • [2] Label prompt for multi-label text classification
    Song, Rui
    Liu, Zelong
    Chen, Xingbing
    An, Haining
    Zhang, Zhiqi
    Wang, Xiaoguang
    Xu, Hao
    APPLIED INTELLIGENCE, 2023, 53 (08) : 8761 - 8775
  • [3] Label prompt for multi-label text classification
    Rui Song
    Zelong Liu
    Xingbing Chen
    Haining An
    Zhiqi Zhang
    Xiaoguang Wang
    Hao Xu
    Applied Intelligence, 2023, 53 : 8761 - 8775
  • [4] Online multi-label dependency topic models for text classification
    Sophie Burkhardt
    Stefan Kramer
    Machine Learning, 2018, 107 : 859 - 886
  • [5] Online multi-label dependency topic models for text classification
    Burkhardt, Sophie
    Kramer, Stefan
    MACHINE LEARNING, 2018, 107 (05) : 859 - 886
  • [6] Exploiting Label Dependency and Feature Similarity for Multi-Label Classification
    Nedungadi, Prema
    Haripriya, H.
    2014 INTERNATIONAL CONFERENCE ON ADVANCES IN COMPUTING, COMMUNICATIONS AND INFORMATICS (ICACCI), 2014, : 2196 - 2200
  • [7] A Simple Approach to Incorporate Label Dependency in Multi-label Classification
    Cherman, Everton Alvares
    Metz, Jean
    Monard, Maria Carolina
    ADVANCES IN SOFT COMPUTING - MICAI 2010, PT II, 2010, 6438 : 33 - 43
  • [8] LABEL-AWARE TEXT REPRESENTATION FOR MULTI-LABEL TEXT CLASSIFICATION
    Guo, Hao
    Li, Xiangyang
    Zhang, Lei
    Liu, Jia
    Chen, Wei
    2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 7728 - 7732
  • [9] Multi-Label Text Classification Based on DistilBERT and Label Correlation
    Wang, Xuyang
    Geng, Liuqing
    Zhang, Xin
    Computer Engineering and Applications, 2024, 60 (23) : 168 - 175
  • [10] Multi-label Classification of Legal Text with Fusion of Label Relations
    Song Z.
    Li Y.
    Li D.
    Wang S.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2022, 35 (02): : 185 - 192