Proceedings

Learning Kernelized Hypothesis for Hidden Confounder Detection

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 4197-4205.

https://doi.org/10.24963/ijcai.2026/467

Abstract

Detecting hidden confounding is crucial for reliable causal analysis from observational data, directly determining which downstream causal inference method to be deployed. Inspired by the theory of higher-order regression, recent sample-efficient hypothesis testing strategies overcome the restrictive requirement of multiple heterogeneous data environment. Despite their progress on single-environment confounder detection, such methods suffer from intrinsic flaws that the structural functions of the causal models should be specified in prior (linear or specific kernel functions). By contrast, real-world data acquisition exhibits diverse, unknown forms of structural functions, imposing an important but challenging gap between theories of higher-order regressions to practical confounder detection. In this paper, we contribute a Bi-level Kernel Confounder Detection (BiKCD) framework by learning adaptive kernelized structural space of structural functions. Subsequently, our BiKCD constructs hypothesis testing by comparing coefficients from the higher-order regression and the classical ordinary least squares in learned kernelized space. Finally, the hypothesis is calibrated to ensure valid inference under adaptivity. Theoretically, we establish an oracle-type risk bound for the selected structural space over a candidate kernel family, with the Type-I error control for the downstream test. Extensive experiments on synthetic and real-world datasets demonstrate the effectiveness of the proposed BiKCD.

  1. [Almond et al., 2005] Douglas Almond, Kenneth Y Chay, and David S Lee. The costs of low birth weight. The Quar- terly Journal of Economics, 120(3):1031-1083, 2005.
  2. [Angrist et al., 1996] Joshua D Angrist, Guido W Imbens, and Donald B Rubin. Identification of causal effects using instrumental variables. Journal of the American statistical Association, 91(434):444-455, 1996.
  3. [Aronszajn, 1950] Nachman Aronszajn. Theory of reproduc- ing kernels. Transactions of the American mathematical society, 68(3):337-404, 1950.
  4. [Buja et al., 2019a] Andreas Buja, Lawrence Brown, Richard Berk, Edward George, Emil Pitkin, Mikhail Traskin, Kai Zhang, and Linda Zhao. Models as ap- proximations i. Statistical Science, 34(4):523-544, 2019.
  5. [Buja et al., 2019b] Andreas Buja, Lawrence Brown, Arun Kumar Kuchibhotla, Richard Berk, Edward George, and Linda Zhao. Models as approximations ii. Statistical Science, 34(4):545-565, 2019.
  6. [Cai et al., 2023] Ruichu Cai, Zhiyi Huang, Wei Chen, Zhifeng Hao, and Kun Zhang. Causal discovery with latent confounders based on higher-order cumulants. In Interna- tional conference on machine learning, pages 3380-3407.
  7. PMLR, 2023.
  8. [Chen et al., 2024] Wei Chen, Zhiyi Huang, Ruichu Cai, Zhifeng Hao, and Kun Zhang. Identification of causal structure with latent variables based on higher order cumu- lants. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 20353-20361, 2024.
  9. [Chen et al., 2026] Yikai Chen, Yunxin Mao, Chunyuan Zheng, Hao Zou, Shanzhi Gu, Shixuan Liu, Yang Shi, Wenjing Yang, Kun Kuang, and Haotian Wang. Detect- ing unobserved confounders: A kernelized regression ap- proach. arXiv preprint arXiv:2601.00200, 2026.
  10. [Dahabreh and Bibbins-Domingo, 2024] Issa J Dahabreh and Kirsten Bibbins-Domingo. Causal inference about the effects of interventions from observational studies in medical journals. Jama, 331(21):1845-1853, 2024.
  11. [Hausman, 1978] Jerry A Hausman. Specification tests in econometrics. Econometrica: Journal of the econometric society, pages 1251-1271, 1978.
  12. [Hazlett, 2016] Chad Hazlett. Kernel balancing: A flexible non-parametric weighting procedure for estimating causal effects, 2016.
  13. [Hu et al., 2024] Kun Hu, Wenjing Yang, Wanrong Huang, Xianchen Zhou, Mingyu Cao, Jing Ren, and Huibin Tan. Sequential fusion based multi-granularity consistency for space-time transformer tracking. In Proceedings of the AAAI conference on artificial intelligence, volume 38, pages 12519-12527, 2024.
  14. [Janzing and Schölkopf, 2018] Dominik Janzing and Bern- hard Schölkopf. Detecting confounding in multivariate linear models via spectral analysis. Journal of Causal In- ference, 6(1):20170013, 2018.
  15. [Kallus et al., 2018] Nathan Kallus, Aahlad Manas Puli, and Uri Shalit. Removing hidden confounding by experimen- tal grounding. Advances in neural information processing systems, 31, 2018.
  16. [Karlsson and Krijthe, 2023] Rickard Karlsson and Jesse Krijthe. Detecting hidden confounding in observational data using multiple environments. Advances in Neural In- formation Processing Systems, 36:44280-44309, 2023.
  17. [Kimeldorf and Wahba, 1971] George Kimeldorf and Grace Wahba. Some results on tchebycheffian spline func- tions. Journal of mathematical analysis and applications, 33(1):82-95, 1971.
  18. [Kou et al., 2025] Zhiqiang Kou, Si Qin, Hailin Wang, Mingkun Xie, Shuo Chen, Yuheng Jia, Tongliang Liu, Masashi Sugiyama, and Xin Geng. Label distribution learning with biased annotations by learning multi-label representation. arXiv preprint arXiv:2502.01170, 2025.
  19. [Kühne et al., 2022] Felicitas Kühne, Michael Schomaker, Igor Stojkov, Beate Jahn, Annette Conrads-Frank, Silke Siebert, Gaby Sroczynski, Sibylle Puntscher, Daniela Schmid, Petra Schnell-Inderst, et al. Causal evidence in health decision making: methodological approaches of causal inference and health decision science. GMS Ger- man Medical Science, 20:Doc12, 2022.
  20. [Lee and Lemieux, 2010] David S Lee and Thomas Lemieux. Regression discontinuity designs in economics. Journal of economic literature, 48(2):281-355, 2010.
  21. [Listl et al., 2016] Stefan Listl, Hendrik Jürges, and Richard G Watt. Causal inference from observational data. Community dentistry and oral epidemiology, 44(5):409-415, 2016.
  22. [Liu and Chan, 2018] Furui Liu and Laiwan Chan. Con- founder detection in high-dimensional linear models using first moments of spectral measures. Neural Computation, 30(8):2284-2318, 2018.
  23. [Mameche et al., 2024] Sarah Mameche, Jilles Vreeken, and David Kaltenpoth. Identifying confounding from causal mechanism shifts. In International Conference on Artifi- cial Intelligence and Statistics, pages 4897-4905. PMLR, 2024.
  24. [Mastouri et al., 2021] Afsaneh Mastouri, Yuchen Zhu, Limor Gultchin, Anna Korba, Ricardo Silva, Matt Kusner, Arthur Gretton, and Krikamol Muandet. Proximal causal learning with kernels: Two-stage estimation and moment restriction. In International conference on machine learn- ing, pages 7512-7523. PMLR, 2021.
  25. [Matthay and Glymour, 2022] Ellicott C Matthay and M Maria Glymour. Causal inference challenges and new directions for epidemiologic research on the health effects of social policies. Current Epidemiology Reports, 9(1):22-37, 2022.
  26. [Mooij et al., 2020] Joris M Mooij, Sara Magliacane, and Tom Claassen. Joint causal inference from multiple con- texts. Journal of machine learning research, 21(99):1- 108, 2020.
  27. [Muandet et al., 2017] Krikamol Muandet, Kenji Fukumizu, Bharath Sriperumbudur, Bernhard Schölkopf, et al. Ker- nel mean embedding of distributions: A review and be- yond. Foundations and Trends® in Machine Learning, 10(1-2):1-141, 2017.
  28. [Newey and Powell, 2003] Whitney K Newey and James L Powell. Instrumental variable estimation of nonparametric models. Econometrica, 71(5):1565-1578, 2003.
  29. [Pearl, 2009] Judea Pearl. Causality. Cambridge university press, 2009.
  30. [Perry et al., 2022] Ronan Perry, Julius Von Kügelgen, and Bernhard Schölkopf. Causal discovery in heterogeneous environments under the sparse mechanism shift hypothe- sis. Advances in Neural Information Processing Systems, 35:10904-10917, 2022.
  31. [Pfister et al., 2018] Niklas Pfister, Peter Bühlmann, Bern- hard Schölkopf, and Jonas Peters. Kernel-based tests for joint independence. Journal of the Royal Statistical Soci- ety Series B: Statistical Methodology, 80(1):5-31, 2018.
  32. [Prosperi et al., 2020] Mattia Prosperi, Yi Guo, Matt Sper- rin, James S Koopman, Jae S Min, Xing He, Shannan Rich, Mo Wang, Iain E Buchan, and Jiang Bian. Causal in- ference and counterfactual prediction in machine learning for actionable healthcare. Nature Machine Intelligence, 2(7):369-375, 2020.
  33. [Reddy and N Balasubramanian, 2024] Abbavaram Gowtham Reddy and Vineeth N Bala- subramanian. Detecting and measuring confounding using causal mechanism shifts. Advances in Neural Information Processing Systems, 37:61677-61699, 2024.
  34. [Schultheiss et al., 2024] Christoph Schultheiss, Peter Bühlmann, and Ming Yuan. Higher-order least squares: assessing partial goodness of fit of linear causal mod- els. Journal of the American Statistical Association, 119(546):1019-1031, 2024.
  35. [Shimizu et al., 2006] Shohei Shimizu, Patrik O Hoyer, Aapo Hyvärinen, Antti Kerminen, and Michael Jordan. A linear non-gaussian acyclic model for causal discovery. Journal of Machine Learning Research, 7(10), 2006.
  36. [Singh et al., 2019] Rahul Singh, Maneesh Sahani, and Arthur Gretton. Kernel instrumental variable regression. Advances in Neural Information Processing Systems, 32, 2019.
  37. [Vanderschueren, 2024] Toon Vanderschueren. Operational decision-making with machine learning and causal infer- ence. PhD thesis, University of Antwerp, 2024.
  38. [Wang et al., 2022] Haotian Wang, Wenjing Yang, Longqi Yang, Anpeng Wu, Liyang Xu, Jing Ren, Fei Wu, and Kun Kuang. Estimating individualized causal effect with confounded instruments. In Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 1857-1867, 2022.
  39. [Wang et al., 2023a] Haotian Wang, Kun Kuang, Haoang Chi, Longqi Yang, Mingyang Geng, Wanrong Huang, and Wenjing Yang. Treatment effect estimation with adjust- ment feature selection. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 2290-2301, 2023.
  40. [Wang et al., 2023b] Haotian Wang, Kun Kuang, Long Lan, Zige Wang, Wanrong Huang, Fei Wu, and Wenjing Yang. Out-of-distribution generalization with causal feature sep- aration. IEEE Transactions on Knowledge and Data Engi- neering, 36(4):1758-1772, 2023.
  41. [Wang et al., 2025] Haotian Wang, Haoxuan Li, Hao Zou, Haoang Chi, Long Lan, Wanrong Huang, and Wenjing Yang. Effective and efficient time-varying counterfac- tual prediction with state-space models. In The Thir- teenth International Conference on Learning Representa- tions, 2025.
  42. [Wang et al., 2026] Haotian Wang, Hao Zou, Haoxuan Li, Haoang Chi, Yang Shi, Yuanxing Zhang, Wenjing Yang, Xinwang Liu, and Zhouchen Lin. Transformers with en- dogenous in-context learning: Bias characterization and mitigation. In The Fourteenth International Conference on Learning Representations, 2026.
  43. [Wang, 2025] Meng-zhu Wang. Simprof: A simple proba- bilistic framework for unsupervised domain adaptation. In Proceedings of the AAAI Conference on Artificial Intelli- gence, volume 39, pages 21153-21161, 2025.
  44. [Yan et al., 2025] Yuguang Yan, Zongyu Li, Haolin Yang, Zeqin Yang, Hao Zhou, Ruichu Cai, and Zhifeng Hao. Re- ducing confounding bias without data splitting for causal inference via optimal transport. In Forty-second Interna- tional Conference on Machine Learning, 2025.
  45. [Yin et al., 2023] Nan Yin, Li Shen, Mengzhu Wang, Long Lan, Zeyu Ma, Chong Chen, Xian-Sheng Hua, and Xiao Luo. Coco: A coupled contrastive framework for unsu- pervised domain adaptive graph classification. In Inter- national Conference on Machine Learning, pages 40040- 40053. PMLR, 2023.
  46. [Zhang et al., 2015] Kun Zhang, Mingming Gong, and Bern- hard Schölkopf. Multi-source domain adaptation: A causal view. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 29, 2015.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards