Causal Feature Selection via Orthogonal Search

• Main Authors: Soleymani, Ashkan, Raj, Anant, Bauer, Stefan, Schölkopf, Bernhard, Besserve, Michel
• Format: Journal Article
• Language: English
• Published: 06.07.2020
• Subjects: Computer Science - Learning; Mathematics - Statistics Theory; Statistics - Machine Learning; Statistics - Theory
• DOI: 10.48550/arxiv.2007.02938

The problem of inferring the direct causal parents of a response variable among a large set of explanatory variables is of high practical importance in many disciplines. However, established approaches often scale at least exponentially with the number of explanatory variables, are difficult to extend to nonlinear relationships, and are difficult to extend to cyclic data. Inspired by {\em Debiased} machine learning methods, we study a one-vs.-the-rest feature selection approach to discover the direct causal parent of the response. We propose an algorithm that works for purely observational data while also offering theoretical guarantees, including the case of partially nonlinear relationships possibly under the presence of cycles. As it requires only one estimation for each variable, our approach is applicable even to large graphs. We demonstrate significant improvements compared to established approaches.