Demo Data from EDBT 2026 / reviewers in the wild / expert
Hengrui Cai
dblp:277/5831
· DBLP profile ↗
8ranked-venue papers
5as first author
7since 2021 · last 2026
0000-0002-5679-8862ORCID · reported
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 8 · 5 first-author · 7 since 2021
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
6 papers |
Probabilistic and Bayesian machine learning · 67% Knowledge representation and reasoning · 14% Trustworthy machine learning · 14% | |
| Interdisciplinary, comprehensive, and emerging computing
1 paper |
Computational science and engineering · 100% | |
| Theoretical computer science
1 paper |
Mathematical optimization · 100% |
Topics — the 13 heaviest of 15, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Probabilistic and Bayesian machine learning
causal inference |
2.8 | 6 | 2023 | Jump Interval-Learning for Individualized Decision Making with Continuous Treatments · J. Mach. Learn. Res. 2023 On Learning Necessary and Sufficient Causal Graphs · NeurIPS 2023 Towards Trustworthy Explanation: On Causal Rationalization · ICML 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference
causal discovery |
0.7 | 1 | 2023 | On Learning Necessary and Sufficient Causal Graphs · NeurIPS 2023 |
Knowledge, reasoning and agents › Knowledge representation and reasoning › causal reasoning
causal graphical model |
0.7 | 1 | 2023 | On Heterogeneous Treatment Effects in Heterogeneous Causal Graphs · ICML 2023 |
Knowledge, reasoning and agents › Knowledge representation and reasoning
causal reasoning |
0.7 | 1 | 2023 | On Heterogeneous Treatment Effects in Heterogeneous Causal Graphs · ICML 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference › causal effect estimation
continuous treatment |
0.7 | 1 | 2023 | Jump Interval-Learning for Individualized Decision Making with Continuous Treatments · J. Mach. Learn. Res. 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference
individualized treatment rule |
0.7 | 1 | 2023 | Jump Interval-Learning for Individualized Decision Making with Continuous Treatments · J. Mach. Learn. Res. 2023 |
Machine learning › Trustworthy machine learning
interpretability |
0.7 | 1 | 2023 | Towards Trustworthy Explanation: On Causal Rationalization · ICML 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference › counterfactual prediction
probability of causation |
0.7 | 1 | 2023 | On Learning Necessary and Sufficient Causal Graphs · NeurIPS 2023 |
Machine learning › Trustworthy machine learning › interpretability
rationalization |
0.7 | 1 | 2023 | Towards Trustworthy Explanation: On Causal Rationalization · ICML 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference › causal effect estimation
mediation analysis |
0.5 | 1 | 2021 | ANOCE: Analysis of Causal Effects with Multiple Mediators via Constrained Structural Learning · ICLR 2021 |
Machine learning › Reinforcement learning
off-policy evaluation |
0.5 | 1 | 2021 | Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment Settings · NeurIPS 2021 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference
heterogeneous treatment effect estimation |
0.2 | 1 | 2023 | On Heterogeneous Treatment Effects in Heterogeneous Causal Graphs · ICML 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference › causal effect estimation
treatment effect estimation |
0.1 | 1 | 2021 | Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment Settings · NeurIPS 2021 |
Methods — techniques the papers use, named apart from their topics
probabilities of causation · 1.3augmented inverse probability weighting · 0.9asymptotic analysis · 0.9structural causal model · 0.7jump penalized regression · 0.7interactive structural learning · 0.7dynamic programming · 0.7deep neural network · 0.7confidence intervals · 0.7causal structural learning · 0.7deep discretization · 0.5constrained structural learning · 0.5
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | A Review of Causal Decision MakingabstractTo make effective decisions, it is important to have a thorough understanding of the causal relationships among actions, environments, and outcomes. This review aims to surface three crucial aspects of decision making through a causal lens: 1) the discovery of causal relationships through causal structure learning, 2) understanding the impacts of these relationships through causal effect learning, and 3) applying the knowledge gained from the first two aspects to support decision making via causal policy learning. Moreover, we identify challenges that hinder the broader utilization of causal decision making and discuss recent advances in overcoming these challenges. Finally, we provide future research directions to address these challenges and further enhance the implementation of causal decision making in practice, with real-world applications illustrated through the proposed causal decision-making workflow. To facilitate broader adoption, we additionally integrate relevant methods into a unified Python-based collection, offering a methodological and practical framework for the community (available at https://causaldm.github.io/Causal-Decision-Making). Hengrui Cai, Runzhe Wan, Yang Xu 0089, Rui Song 0006 |
J. Artif. Intell. Res. | 2 |
| 2023 | On Heterogeneous Treatment Effects in Heterogeneous Causal GraphsabstractHeterogeneity and comorbidity are two interwoven challenges associated with various healthcare problems that greatly hampered research on developing effective treatment and understanding of the underlying neurobiological mechanism. Very few studies have been conducted to investigate heterogeneous causal effects (HCEs) in graphical contexts due to the lack of statistical methods. To characterize this heterogeneity, we first conceptualize heterogeneous causal graphs (HCGs) by generalizing the causal graphical model with confounder-based interactions and multiple mediators. Such confounders with an interaction with the treatment are known as moderators. This allows us to flexibly produce HCGs given different moderators and explicitly characterize HCEs from the treatment or potential mediators on the outcome. We establish the theoretical forms of HCEs and derive their properties at the individual level in both linear and nonlinear models. An interactive structural learning is developed to estimate the complex HCGs and HCEs with confidence intervals provided. Our method is empirically justified by extensive simulations and its practical usefulness is illustrated by exploring causality among psychiatric disorders for trauma survivors. Code implementing the proposed algorithm is open-source and publicly available at: https://github.com/richard-watson/ISL. Richard A. Watson, Hengrui Cai, Xinming An, Samuel A. McLean, Rui Song 0006 |
ICML | 2 |
| 2023 | Towards Trustworthy Explanation: On Causal RationalizationabstractWith recent advances in natural language processing, rationalization becomes an essential self-explaining diagram to disentangle the black box by selecting a subset of input texts to account for the major variation in prediction. Yet, existing association-based approaches on rationalization cannot identify true rationales when two or more snippets are highly inter-correlated and thus provide a similar contribution to prediction accuracy, so-called spuriousness. To address this limitation, we novelly leverage two causal desiderata, non-spuriousness and efficiency, into rationalization from the causal inference perspective. We formally define a series of probabilities of causation based on a newly proposed structural causal model of rationalization, with its theoretical identification established as the main component of learning necessary and sufficient rationales. The superior performance of the proposed causal rationalization is demonstrated on real-world review and medical datasets with extensive experiments compared to state-of-the-art methods. Wenbo Zhang 0010, Hengrui Cai |
ICML | 5 |
| 2023 | On Learning Necessary and Sufficient Causal GraphsabstractThe causal revolution has stimulated interest in understanding complex relationships in various fields. Most of the existing methods aim to discover causal relationships among all variables within a complex large-scale graph. However, in practice, only a small subset of variables in the graph are relevant to the outcomes of interest. Consequently, causal estimation with the full causal graph---particularly given limited data---could lead to numerous *falsely discovered, spurious* variables that exhibit high correlation with, but exert no causal impact on, the target outcome. In this paper, we propose learning a class of *necessary and sufficient causal graphs (NSCG)* that exclusively comprises causally relevant variables for an outcome of interest, which we term *causal features*. The key idea is to employ *probabilities of causation* to systematically evaluate the importance of features in the causal graph, allowing us to identify a subgraph relevant to the outcome of interest. To learn NSCG from data, we develop a *necessary and sufficient causal structural learning (NSCSL)* algorithm, by establishing theoretical properties and relationships between probabilities of causation and natural causal effects of features. Across empirical studies of simulated and real data, we demonstrate that NSCSL outperforms existing algorithms and can reveal crucial yeast genes for target heritable traits of interest. Hengrui Cai, Yixin Wang 0002, Michael I. Jordan, Rui Song 0006 |
NeurIPS | 1 |
| 2023 | Jump Interval-Learning for Individualized Decision Making with Continuous TreatmentsabstractAn individualized decision rule (IDR) is a decision function that assigns each individual a given treatment based on his/her observed characteristics. Most of the existing works in the literature consider settings with binary or finitely many treatment options. In this paper, we focus on the continuous treatment setting and propose a jump interval-learning to develop an individualized interval-valued decision rule (I2DR) that maximizes the expected outcome. Unlike IDRs that recommend a single treatment, the proposed I2DR yields an interval of treatment options for each individual, making it more flexible to implement in practice. To derive an optimal I2DR, our jump interval-learning method estimates the conditional mean of the outcome given the treatment and the covariates via jump penalized regression, and derives the corresponding optimal I2DR based on the estimated outcome regression function. The regressor is allowed to be either linear for clear interpretation or deep neural network to model complex treatment-covariates interactions. To implement jump interval-learning, we develop a searching algorithm based on dynamic programming that efficiently computes the outcome regression function. Statistical properties of the resulting I2DR are established when the outcome regression function is either a piecewise or continuous function over the treatment space. We further develop a procedure to infer the mean outcome under the (estimated) optimal policy. Extensive simulations and a real data application to a Warfarin study are conducted to demonstrate the empirical validity of the proposed I2DR. Hengrui Cai, Chengchun Shi, Rui Song 0006, Wenbin Lu |
J. Mach. Learn. Res. | 1 |
| 2021 | ANOCE: Analysis of Causal Effects with Multiple Mediators via Constrained Structural Learning
Hengrui Cai, Rui Song 0006, Wenbin Lu |
ICLR | 1 |
| 2021 | Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment SettingsabstractWe consider off-policy evaluation (OPE) in continuous treatment settings, such as personalized dose-finding. In OPE, one aims to estimate the mean outcome under a new treatment decision rule using historical data generated by a different decision rule. Most existing works on OPE focus on discrete treatment settings. To handle continuous treatments, we develop a novel estimation method for OPE using deep jump learning. The key ingredient of our method lies in adaptively discretizing the treatment space using deep discretization, by leveraging deep learning and multi-scale change point detection. This allows us to apply existing OPE methods in discrete treatments to handle continuous treatments. Our method is further justified by theoretical results, simulations, and a real application to Warfarin Dosing. Hengrui Cai, Chengchun Shi, Rui Song 0006, Wenbin Lu |
NeurIPS | 1 |
| 2020 | On Validation and Planning of An Optimal Decision Rule with Application in Healthcare StudiesabstractIn the current era of personalized recommendation, one major interest is to develop an optimal individualized decision rule that assigns individuals with the best treatment option according to their covariates. Estimation of optimal decision rules (ODR) has been extensively investigated recently, however, at present, no testing procedure is proposed to verify whether these ODRs are significantly better than the naive decision rule that always assigning individuals to a fixed treatment option. In this paper, we propose a testing procedure for detecting the existence of an ODR that is better than the naive decision rule under the randomized trials. We construct the proposed test based on the difference of estimated value functions using the augmented inverse probability weighted method. The asymptotic distributions of the proposed test statistic under the null and local alternative hypotheses are established. Based on the established asymptotic distributions, we further develop a sample size calculation formula for testing the existence of an ODR in designing A/B tests. Extensive simulations and a real data application to a schizophrenia clinical trial data are conducted to demonstrate the empirical validity of the proposed methods. Hengrui Cai, Wenbin Lu, Rui Song 0006 |
ICML | 1 |