Learning to Control in Metric Space with Optimal Regret. - Chengzhuo Ni; Lin F. Yang; Mengdi Wang (Allerton, 2019)

Maximum Likelihood Tensor Decomposition of Markov Decision Process. - Chengzhuo Ni; Mengdi Wang (ISIT, 2019)

Learning Good State and Action Representations via Tensor Decomposition. - Chengzhuo Ni; Anru R. Zhang; Yaqi Duan; Mengdi Wang (ISIT, 2021)

On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Method. - Junyu Zhang; Chengzhuo Ni; Zheng Yu; Csaba Szepesvári; Mengdi Wang (NeurIPS, 2021)

Optimal Estimation of Policy Gradient via Double Fitted Iteration. - Chengzhuo Ni; Ruiqi Zhang; Xiang Ji; Xuezhou Zhang; Mengdi Wang (ICML, 2022)

Off-Policy Fitted Q-Evaluation with Differentiable Function Approximators: Z-Estimation and Inference Theory. - Ruiqi Zhang; Xuezhou Zhang; Chengzhuo Ni; Mengdi Wang (ICML, 2022)

Bandit Theory and Thompson Sampling-Guided Directed Evolution for Sequence Optimization. - Hui Yuan 002; Chengzhuo Ni; Huazheng Wang; Xuezhou Zhang; Le Cong; Csaba Szepesvári; Mengdi Wang (NeurIPS, 2022)

Optimal Estimation of Off-Policy Policy Gradient via Double Fitted Iteration. - Chengzhuo Ni; Ruiqi Zhang; Xiang Ji; Xuezhou Zhang; Mengdi Wang (CoRR, 2022)

Representation Learning for General-sum Low-rank Markov Games. - Chengzhuo Ni; Yuda Song 001; Xuezhou Zhang; Chi Jin 001; Mengdi Wang (CoRR, 2022)

Representation Learning for Low-rank General-sum Markov Games. - Chengzhuo Ni; Yuda Song 001; Xuezhou Zhang; Zihan Ding; Chi Jin 001; Mengdi Wang (ICLR, 2023)

Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement. - Hui Yuan 002; Kaixuan Huang; Chengzhuo Ni; Minshuo Chen; Mengdi Wang (NeurIPS, 2023)

Learning Good State and Action Representations for Markov Decision Process via Tensor Decomposition. - Chengzhuo Ni; Yaqi Duan; Munther A. Dahleh; Mengdi Wang; Anru R. Zhang (J. Mach. Learn. Res., 2023)

Diffusion Model for Data-Driven Black-Box Optimization. - Zihao Li; Hui Yuan 002; Kaixuan Huang; Chengzhuo Ni; Yinyu Ye 001; Minshuo Chen; Mengdi Wang (CoRR, 2024)