Doubly Robust Counterfactual Classification
Kwangho Kim, Edward H. Kennedy, José R. Zubizarreta
Abstract
We study counterfactual classification as a new tool for decision-making under hypothetical (contrary to fact) scenarios. We propose a doubly-robust nonparametric estimator for a general counterfactual classifier, where we can incorporate flexible constraints by casting the classification problem as a nonlinear mathematical program involving counterfactuals. We go on to analyze the rates of convergence of the estimator and provide a closed-form expression for its asymptotic distribution. Our analysis shows that the proposed estimator is robust against nuisance model misspecification, and can attain fast rates with tractable inference even when using nonparametric machine learning approaches. We study the empirical performance of our methods by simulation and apply them for recidivism risk prediction.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 27271d84-0f6c-415e-89e5-c7f72e89fd9cCited by top-tier papers2
- Fair and Robust Estimation of Heterogeneous Treatment Effects for Policy LearningKwangho Kim, José R. ZubizarretaICML 2023 · 12 citations
- Omitted Variable Bias in Language Models Under Distribution ShiftVictoria Lin, Louis-Philippe Morency, Eli Ben-MichaelICML 2026 · 1 citation
Builds on1
Related papers
- Treatment Responder Classification with AbstentionHaoxiang Wang, Haoxuan Li, Ziyan Wang, Zhiheng Zhang et al.ICML 2026
- Doubly-Robust Estimation of Counterfactual Policy Mean EmbeddingsHoussam Zenati, Bariscan Bozkurt, Arthur GrettonNeurIPS 2025 · 3 citations
- Counterfactual Density Estimation using Kernel Stein DiscrepanciesDiego Martinez-Taboada, Edward KennedyICLR 2024 · 8 citations
- Probably Approximately Correct Constrained LearningLuiz F. O. Chamon, Alejandro RibeiroNeurIPS 2020 · 67 citations
- Counterfactually Comparing Abstaining ClassifiersYo Joong Choe, Aditya Gangrade, Aaditya RamdasNeurIPS 2023 · 2 citations
