A Light Causal Feature Selection Approach to High-Dimensional Data
Zhaolong Ling,Ying Li,Yiwen Zhang,Kui Yu,Peng Zhou,Bo Li,Xindong Wu
DOI: https://doi.org/10.1109/tkde.2022.3218786
IF: 9.235
2022-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Causal feature selection has received increasing attention in recent years. However, the state-of-the-art causal feature selection algorithms use the conditional independence tests, which require enumerating conditioning sets, leading to an exponential increase in computational complexity along with an increase in feature space. To address this problem, in this paper, we theoretically analyze the unique performance of causal features in mutual information, and propose a novel Causal Feature Selection algorithm using Mutual Information, called CFS-MI. Specifically, CFS-MI separately instantiates the pairwise comparison of mutual information in two stages to reduce computational complexity, and thus improves the efficiency on high-dimensional data. Extensive experiments on 5 benchmark Bayesian networks and 16 real-world datasets validate that CFS-MI has comparable accuracy compared to 7 state-of-the-art causal feature selection algorithms, while presenting more superior computational efficiency.
computer science, information systems, artificial intelligence,engineering, electrical & electronic