CausalKANs: interpretable treatment effect estimation with Kolmogorov-Arnold networks
Not on the map. Why?
Abstract (arXiv)
Deep neural networks achieve state-of-the-art performance in estimating heterogeneous treatment effects, but their opacity limits trust and adoption in sensitive domains such as medicine, economics, and public policy. Building on well-established and high-performing causal neural architectures, we propose causalKANs, a framework that transforms neural estimators of conditional average treatment effects (CATEs) into Kolmogorov--Arnold Networks (KANs). By incorporating pruning and symbolic simplification, causalKANs yields interpretable closed-form formulas while preserving predictive accuracy. Experiments on benchmark datasets demonstrate that causalKANs perform on par with neural baselines in CATE error metrics, and that even simple KAN variants achieve competitive performance, offering a favorable accuracy--interpretability trade-off. By combining reliability with analytic accessibility, causalKANs provide auditable estimators supported by closed-form expressions and interpretable plots, enabling trustworthy individualized decision-making in high-stakes settings. We release the code for reproducibility at https://github.com/aalmodovares/causalkans .
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 45 of its 90 references that are in the map. First posted on arXiv 26 Sep 2025.
Builds on
45 of 90 references.
References · 45
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
- 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,416
- 2024KAN: Kolmogorov-Arnold Networkscited by 649
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2016European Union regulations on algorithmic decision-making and a "right to explanation"cited by 344
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 332
- 2016Learning Representations for Counterfactual Inferencecited by 329
- 2020Neural Additive Models: Interpretable Machine Learning with Neural Netscited by 319
- 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 287
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
33 more
- 2016Visualizing the Effects of Predictor Variables in Black Box Supervised Learning Modelscited by 213
- 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
- 2019What Clinicians Want: Contextualizing Explainable Machine Learning for Clinical End Usecited by 135
- 2024KAN 2.0: Kolmogorov-Arnold Networks Meet Sciencecited by 118
- 2017Automated versus do-it-yourself methods for causal inference: Lessons learned from a data analysis competitioncited by 106
- 2019Learning Counterfactual Representations for Estimating Individual Dose-Response Curvescited by 90
- 2020Causal Interpretability for Machine Learning -- Problems, Methods and Evaluationcited by 82
- 2024Convolutional Kolmogorov-Arnold Networkscited by 77
- 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 75
- 2022Causal Machine Learning for Healthcare and Precision Medicinecited by 68
- 2024Kolmogorov Arnold Informed neural network: A physics-informed deep learning framework for solving forward and inverse problems based on Kolmogorov Arnold Networkscited by 67
- 2018Perfect Match: A Simple Method for Learning Representations For Counterfactual Inference With Neural Networkscited by 56
- 2024KAGNNs: Kolmogorov-Arnold Networks meet Graph Learningcited by 49
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
- 2024CoxKAN: Kolmogorov-Arnold Networks for Interpretable, High-Performance Survival Analysiscited by 22
- 2022Estimating individual treatment effects under unobserved confounding using binary instrumentscited by 18
- 2022Benchmarking Heterogeneous Treatment Effect Models through the Lens of Interpretabilitycited by 15
- 2024Physics-Informed Kolmogorov-Arnold Networks for Power System Dynamicscited by 14
- 2020Causal Rule Ensemble: Interpretable Discovery and Inference of Heterogeneous Treatment Effectscited by 13
- 2016Regularization and confounding in linear regression for treatment effect estimationcited by 12
- 2019Assessing Treatment Effect Variation in Observational Studies: Results from a Data Challengecited by 12
- 2019Learning Interpretable Models with Causal Guaranteescited by 11
- 2024Kolmogorov-Arnold Network Autoencoderscited by 8
- 2022Covariate-Balancing-Aware Interpretable Deep Learning models for Treatment Effect Estimationcited by 5
- 2024Residual Kolmogorov-Arnold Network for Enhanced Deep Learningcited by 5
- 2021Heterogeneous Treatment Effect Estimation using machine learning for Healthcare application: tutorial and benchmarkcited by 4
- 2021A causal fused lasso for interpretable heterogeneous treatment effects estimationcited by 4
- 2023Causal Dependence Plotscited by 3
- 2023A New Causal Rule Learning Approach to Interpretable Estimation of Heterogeneous Treatment Effectcited by 3
- 2025Forecasting VIX using interpretable Kolmogorov-Arnold networkscited by 3
- 2025Interpretable Graph Kolmogorov-Arnold Networks for Multi-Cancer Classification and Biomarker Identification using Multi-Omics Datacited by 3
- 2025Interpretable Clinical Classification with Kolmogorov-Arnold Networkscited by 3
- 2025KANITE: Kolmogorov-Arnold Networks for ITE estimationcited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.