cs.LGOct 13, 2026

Causal Discovery Under Hard Selection Bias: A New Robust Score-Matching Approach

Authors: Yiwen Qiu, Francesco Montagna, Shimeng Huang, Francesco Locatello

Organizations: IST Austria, Am Campus 1, Klosterneuburg, 3400

Abstract

Selection bias is a difficult yet widespread problem in causal discovery, occurring whenever non-random selection processes lead to data that is not representative of the underlying populations. Due to its practical importance, related works have been proposed to address the problem under local [Versteeg et al., 2022] or interventional settings [Dai et al., 2025a], but general algorithms for observational data remain elusive. In this paper, we focus on truncated data (hard selection), and provide a general result for Additive Noise Models (ANM). We show that score-based methods are a surprisingly suitable candidate due to a special property of the score function (∇log⁡p(x)\nabla \log p(x)): deterministic truncation preserves both the score of the joint distribution as well as the Jacobian of the score. While established score-matching algorithms still fail in most settings, we actively leverage this observation to propose a new method for causal discovery that is robust under truncated data with ANMs. Overall, our approach offers a general, robust solution that is agnostic to external information about the selection process and still achieves comparable performance to the state-of-the-art approaches when selection is not present.

Figures & tables

Appendix figures & tables20 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Identifying Structural Biases from Causal Mechanism Shifts

    Jun 17, 2026Praharsh Nanavati, Jilles Vreeken, David KaltenpothStructural Causal ModelsCausal Discovery

  2. Generalizing Identifiability Results in Causal Discovery from Complete Data to Missing Data Settings

    Oct 13, 2026Matteo Ceriscioli, Shohei Shimizu, Mallika Gupta +1Causal DiscoveryCausal Identifiability

  3. Discrete Score Matching Enables Causal Discovery from Count Data

    Sep 30, 2026Euijong Song, Hyewon Park, Gunwoong ParkScore MatchingCausal Discovery