English

Causal Feature Selection with Dimension Reduction for Interpretable Text Classification

Machine Learning 2020-10-12 v1 Computation and Language Information Retrieval

Abstract

Text features that are correlated with class labels, but do not directly cause them, are sometimesuseful for prediction, but they may not be insightful. As an alternative to traditional correlation-basedfeature selection, causal inference could reveal more principled, meaningful relationships betweentext features and labels. To help researchers gain insight into text data, e.g. for social scienceapplications, in this paper we investigate a class of matching-based causal inference methods fortext feature selection. Features used in document classification are often high dimensional, howeverexisting causal feature selection methods use Propensity Score Matching (PSM) which is known to beless effective in high-dimensional spaces. We propose a new causal feature selection framework thatcombines dimension reduction with causal inference to improve text feature selection. Experiments onboth synthetic and real-world data demonstrate the promise of our methods in improving classificationand enhancing interpretability.

Keywords

Cite

@article{arxiv.2010.04609,
  title  = {Causal Feature Selection with Dimension Reduction for Interpretable Text Classification},
  author = {Guohou Shan and James Foulds and Shimei Pan},
  journal= {arXiv preprint arXiv:2010.04609},
  year   = {2020}
}

Comments

11 pages, 3 pages

R2 v1 2026-06-23T19:12:41.220Z