← Search

Hengrui Cai

9 accepted papers

2026

Reasoning Is Not Free: Robust Adaptive Cost-Efficient Router for LLM-as-a-Judge

ICML 2026poster

Reasoning-capable large language models (LLMs) have recently been adopted as automated judges, but their benefits and costs in LLM-as-a-Judge settings remain unclear. Through controlled comparisons between reasoning and non-reasoning judges, we show that explicit reasoning substantially improves jud…

Cited by 0SourceScholar
2023

On Heterogeneous Treatment Effects in Heterogeneous Causal Graphs

ICML 2023poster

Heterogeneity 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 ca…

2023

Towards Trustworthy Explanation: On Causal Rationalization

ICML 2023poster

With 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 i…

2021

ANOCE: Analysis of Causal Effects with Multiple Mediators via Constrained Structural Learning

ICLR 2021poster

In the era of causal revolution, identifying the causal effect of an exposure on the outcome of interest is an important problem in many areas, such as epidemics, medicine, genetics, and economics. Under a general causal graph, the exposure may have a direct effect on the outcome and also an indirec…

Cited by 13SourcePDFScholar
2021

Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment Settings

NeurIPS 2021poster

We 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 tr…

2020

On Validation and Planning of An Optimal Decision Rule with Application in Healthcare Studies

ICML 2020poster

In 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, ho…

Cited by 1SourcePDFScholar