Search papers, labs, and topics across Lattice.
This paper introduces BLITZ, a novel nonparametric conditional independence test that enhances the speed and accuracy of causal discovery by employing a two-stage regression approach. By first addressing broad smooth dependence through low-order polynomial regression and then applying a nonlinear feature map with shallow tree regressions, BLITZ effectively reduces conditional-mean bias while maintaining calibration. In simulations and causal discovery experiments, BLITZ outperforms existing methods in terms of null calibration and structural recovery, demonstrating its potential as a scalable solution for causal inference tasks.
BLITZ achieves superior null calibration and faster performance, making it a game-changer for nonparametric conditional independence testing in causal discovery.
Constraint-based causal discovery relies on repeated conditional independence tests, but fast nonparametric tests often sacrifice calibration, especially when variables depend on the conditioning set through nonlinear relationships. We introduce BLITZ (Broad-to-Local Independence Testing via residualiZation), a nonparametric conditional independence test designed to run well under a second while maintaining the accuracy needed for the thousands of queries performed by constraint-based causal discovery algorithms. BLITZ first removes broad smooth dependence on the conditioning set using low-order polynomial regression, then applies a small nonlinear feature map and residualizes those features with shallow tree regressions. The resulting statistic tests residual cross-covariance, with a moment-matched chi-square approximation to the null distribution. We show theoretically that the two-stage design reduces the effective complexity faced by the tree residualizers, allowing shallow trees to control residual conditional-mean bias while avoiding excessive overfitting. In simulations, BLITZ provides better null calibration than fast kernel, random-feature, and regression-based competitors while remaining among the fastest methods tested. In causal discovery experiments on synthetic graphs and flow-cytometry data, BLITZ yields more reliable endpoint orientations among retained adjacencies and competitive structural recovery. These results suggest that broad-to-local residualization is a practical route to calibrated, scalable nonparametric conditional independence testing for causal discovery.
