Zhenke Wu

dblp:259/3143 · DBLP profile ↗
← Back
6ranked-venue papers
0as first author
5since 2021 · last 2025
0000-0001-7582-669XORCID · reported

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 5 · 4 since 2021Applied, interdisciplinary, general and emerging computing · 1 · 1 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
5 papers
Reinforcement learning · 55% Probabilistic and Bayesian machine learning · 31% Deep learning architectures and training · 14%
Theoretical computer science
1 paper
Information theory · 100%
Interdisciplinary, comprehensive, and emerging computing
2 papers
Medical and health informatics · 100%

Topics — the 15 heaviest of 18, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Reinforcement learning
policy optimization
0.912025
Beyond Average Value Function in Precision Medicine: Maximum Probability-Driven Reinforcement Learning for Survival Analysis · NeurIPS 2025
Machine learning › Reinforcement learning
reinforcement learning for healthcare
0.912025
Beyond Average Value Function in Precision Medicine: Maximum Probability-Driven Reinforcement Learning for Survival Analysis · NeurIPS 2025
Machine learning › Probabilistic and Bayesian machine learning › statistical inference
survival analysis
0.912025
Beyond Average Value Function in Precision Medicine: Maximum Probability-Driven Reinforcement Learning for Survival Analysis · NeurIPS 2025
Machine learning › Probabilistic and Bayesian machine learning
causal inference
0.712023
A Reinforcement Learning Framework for Dynamic Mediation Analysis · ICML 2023
Machine learning › Reinforcement learning
non-stationary reinforcement learning
0.712023
A Robust Test for the Stationarity Assumption in Sequential Decision Making · ICML 2023
Machine learning › Reinforcement learning
offline reinforcement learning
0.712023
A Robust Test for the Stationarity Assumption in Sequential Decision Making · ICML 2023
Machine learning › Reinforcement learning
off-policy evaluation
0.712023
A Reinforcement Learning Framework for Dynamic Mediation Analysis · ICML 2023
Machine learning › Reinforcement learning
semiparametric efficient estimation
0.712023
A Reinforcement Learning Framework for Dynamic Mediation Analysis · ICML 2023
Information theory › hypothesis testing
change-point detection
0.712023
A Robust Test for the Stationarity Assumption in Sequential Decision Making · ICML 2023
Information theory
hypothesis testing
0.712023
A Robust Test for the Stationarity Assumption in Sequential Decision Making · ICML 2023
Machine learning › Deep learning architectures and training
attention mechanism
0.612022
Kernel Multimodal Continuous Attention · NeurIPS 2022
Machine learning › Deep learning architectures and training › attention mechanism › temporal attention
continuous-time attention
0.612022
Kernel Multimodal Continuous Attention · NeurIPS 2022
Machine learning › Probabilistic and Bayesian machine learning › statistical inference › parameter estimation
expectation-maximization
0.412020
A Robust Functional EM Algorithm for Incomplete Panel Count Data · NeurIPS 2020
Medical and health informatics
precision medicine
0.312025
Beyond Average Value Function in Precision Medicine: Maximum Probability-Driven Reinforcement Learning for Survival Analysis · NeurIPS 2025
Data integration and cleaning
missing data
0.112020
A Robust Functional EM Algorithm for Incomplete Panel Count Data · NeurIPS 2020

Methods — techniques the papers use, named apart from their topics

dynamic programming · 1.7model-based testing · 1.3doubly robust estimation · 1.3EM algorithm · 1.3poisson process · 0.9nonparametric estimation · 0.9semiparametric efficiency theory · 0.7robust estimation · 0.7numerical integration · 0.6kernel exponential families · 0.6deformed exponential families · 0.6non-parametric estimation · 0.4
YearPublicationVenuePosition
2025 Beyond Average Value Function in Precision Medicine: Maximum Probability-Driven Reinforcement Learning for Survival Analysis
abstract
Constructing multistage optimal decisions for alternating recurrent event data is critically important in medical and healthcare research. Current reinforcement learning (RL) algorithms have only been applied to time-to-event data, with the objective of maximizing expected survival time. However, alternating recurrent event data has a different structure, which motivates us to model the probability and frequency of event occurrences rather than a single terminal outcome. In this paper, we introduce an RL framework specifically designed for alternating recurrent event data. Our goal is to maximize the probability that the duration between consecutive events exceeds a clinically meaningful threshold. To achieve this, we identify a lower bound of this probability, which transforms the problem into maximizing a cumulative sum of log probabilities, thus enabling direct application of standard RL algorithms. We establish the theoretical properties of the resulting optimal policy and demonstrate through numerical experiments that our proposed algorithm yields a larger probability of that the time between events exceeds a critical threshold compared with existing state-of-the-art algorithms.
Jianqi Feng, Chengchun Shi, Zhenke Wu
NeurIPS3
2025 Generating synthetic electronic health record data: a methodological scoping review with benchmarking on phenotype data and open-source software
abstract
OBJECTIVES: To conduct a scoping review (ScR) of existing approaches for synthetic Electronic Health Records (EHR) data generation, to benchmark major methods, and to provide an open-source software and offer recommendations for practitioners. MATERIALS AND METHODS: We search three academic databases for our scoping review. Methods are benchmarked on open-source EHR datasets, Medical Information Mart for Intensive Care III and IV (MIMIC-III/IV). Seven existing methods covering major categories and two baseline methods are implemented and compared. Evaluation metrics concern data fidelity, downstream utility, privacy protection, and computational cost. RESULTS: Forty-eight studies are identified and classified into five categories. Seven open-source methods covering all categories are selected, trained on MIMIC-III, and evaluated on MIMIC-III or MIMIC-IV for transportability considerations. Among them, Generative Adversarial Network (GAN)-based methods demonstrate competitive performance in fidelity and utility on MIMIC-III, rule-based methods excel in privacy protection. Similar findings are observed on MIMIC-IV, except that GAN-based methods further outperform the baseline methods in preserving fidelity. DISCUSSION: Method choice is governed by the relative importance of the evaluation metrics in downstream use cases. We provide a decision tree to guide the choice among the benchmarked methods. An extensible Python package, "SynthEHRella", is provided to facilitate streamlined evaluations. CONCLUSION: GAN-based methods excel when distributional shifts exist between the training and testing populations. Otherwise, CorGAN and MedGAN are most suitable for association modeling and predictive modeling, respectively. Future research should prioritize enhancing fidelity of the synthetic data while controlling privacy exposure, and comprehensive benchmarking of longitudinal or conditional generation methods.
Xingran Chen, Zhenke Wu, Hyunghoon Cho, Bhramar Mukherjee
J. Am. Medical Informatics Assoc.2
2023 A Reinforcement Learning Framework for Dynamic Mediation Analysis
abstract
Mediation analysis learns the causal effect transmitted via mediator variables between treatments and outcomes, and receives increasing attention in various scientific domains to elucidate causal relations. Most existing works focus on point-exposure studies where each subject only receives one treatment at a single time point. However, there are a number of applications (e.g., mobile health) where the treatments are sequentially assigned over time and the dynamic mediation effects are of primary interest. Proposing a reinforcement learning (RL) framework, we are the first to evaluate dynamic mediation effects in settings with infinite horizons. We decompose the average treatment effect into an immediate direct effect, an immediate mediation effect, a delayed direct effect, and a delayed mediation effect. Upon the identification of each effect component, we further develop robust and semi-parametrically efficient estimators under the RL framework to infer these causal effects. The superior performance of the proposed method is demonstrated through extensive numerical studies, theoretical results, and an analysis of a mobile health dataset. A Python implementation of the proposed procedure is available at https://github.com/linlinlin97/MediationRL.
Jitao Wang, Chengchun Shi, Zhenke Wu, Rui Song 0006
ICML4
2023 A Robust Test for the Stationarity Assumption in Sequential Decision Making
abstract
Reinforcement learning (RL) is a powerful technique that allows an autonomous agent to learn an optimal policy to maximize the expected return. The optimality of various RL algorithms relies on the stationarity assumption, which requires time-invariant state transition and reward functions. However, deviations from stationarity over extended periods often occur in real-world applications like robotics control, health care and digital marketing, resulting in suboptimal policies learned under stationary assumptions. In this paper, we propose a model-based doubly robust procedure for testing the stationarity assumption and detecting change points in offline RL settings with certain degree of homogeneity. Our proposed testing procedure is robust to model misspecifications and can effectively control type-I error while achieving high statistical power, especially in high-dimensional settings. Extensive comparative simulations and a real-world interventional mobile health example illustrate the advantages of our method in detecting change points and optimizing long-term rewards in high-dimensional, non-stationary environments.
Jitao Wang, Chengchun Shi, Zhenke Wu
ICML3
2022 Kernel Multimodal Continuous Attention
abstract
Attention mechanisms take an expectation of a data representation with respect to probability weights. Recently, (Martins et al. 2020, 2021) proposed continuous attention mechanisms, focusing on unimodal attention densities from the exponential and deformed exponential families: the latter has sparse support. (Farinhas et al 2021) extended this to to multimodality via Gaussian mixture attention densities. In this paper, we extend this to kernel exponential families (Canu and Smola 2006) and our new sparse counterpart, kernel deformed exponential families. Theoretically, we show new existence results for both kernel exponential and deformed exponential families, and that the deformed case has similar approximation capabilities to kernel exponential families. Lacking closed form expressions for the context vector, we use numerical integration: we show exponential convergence for both kernel exponential and deformed exponential families. Experiments show that kernel continuous attention often outperforms unimodal continuous attention, and the sparse variant tends to highlight peaks of time series.
Alexander Moreno, Zhenke Wu, Supriya Nagesh, Walter H. Dempsey, James M. Rehg
NeurIPS2
2020 A Robust Functional EM Algorithm for Incomplete Panel Count Data
abstract
Panel count data describes aggregated counts of recurrent events observed at discrete time points. To understand dynamics of health behaviors and predict future negative events, the field of quantitative behavioral research has evolved to increasingly rely upon panel count data collected via multiple self reports, for example, about frequencies of smoking using in-the-moment surveys on mobile devices. However, missing reports are common and present a major barrier to downstream statistical learning. As a first step, under a missing completely at random assumption (MCAR), we propose a simple yet widely applicable functional EM algorithm to estimate the counting process mean function, which is of central interest to behavioral scientists. The proposed approach wraps several popular panel count inference methods, seamlessly deals with incomplete counts and is robust to misspecification of the Poisson process assumption. Theoretical analysis of the proposed algorithm provides finite-sample guarantees by extending parametric EM theory to the general non-parametric setting. We illustrate the utility of the proposed algorithm through numerical experiments and an analysis of smoking cessation data. We also discuss useful extensions to address deviations from the MCAR assumption and covariate effects.
Alexander Moreno, Zhenke Wu, Jamie Yap, Cho Lam, David W. Wetter, Inbal Nahum-Shani, Walter H. Dempsey, James M. Rehg
NeurIPS2