Reinforcement Learning under a Multi-agent Predictive State Representation Model: Method and Theory. - Zhi Zhang 012; Zhuoran Yang; Han Liu 001; Pratap Tokekar; Furong Huang (ICLR, 2022)

Multivariate Time Series Forecasting By Graph Attention Networks With Theoretical Guarantees. - Zhi Zhang 012; Weijian Li 002; Han Liu 001 (AISTATS, 2024)

DODT: Enhanced Online Decision Transformer Learning through Dreamer's Actor-Critic Trajectory Forecasting. - Eric Hanchen Jiang; Zhi Zhang 012; Dinghuai Zhang; Andrew Lizarraga; Chenheng Xu; Yasi Zhang; Siyan Zhao; Zhengjie Xu; Peiyu Yu; Yuer Tang; Deqian Kong; Ying Nian Wu (CoRR, 2024)

Statistical Guarantees for Lifelong Reinforcement Learning using PAC-Bayesian Theory. - Zhi Zhang 012; Chris Chow; Yasi Zhang; Yanchao Sun; Haochen Zhang; Eric Hanchen Jiang; Han Liu 001; Furong Huang; Yuchen Cui; Oscar Hernan Madrid Padilla (CoRR, 2024)

Dense ReLU Neural Networks for Temporal-spatial Model. - Zhi Zhang 012; Carlos Misael Madrid Padilla; Xiaokai Luo; Oscar Hernan Madrid Padilla; Daren Wang (CoRR, 2024)

Unlocking the Potential of Text-to-Image Diffusion with PAC-Bayesian Theory. - Eric Hanchen Jiang; Yasi Zhang; Zhi Zhang 012; Yixin Wan; Andrew Lizarraga; Shufan Li; Ying Nian Wu (CoRR, 2024)

Confidence Interval Construction and Conditional Variance Estimation with Dense ReLU Networks. - Carlos Misael Madrid Padilla; Oscar Hernan Madrid Padilla; Yik Lun Kei; Zhi Zhang 012; Yanzhen Chen (CoRR, 2024)

Statistical Guarantees for Lifelong Reinforcement Learning using PAC-Bayes Theory. - Zhi Zhang 012; Chris Chow; Yasi Zhang; Yanchao Sun; Haochen Zhang; Eric Hanchen Jiang; Han Liu 001; Furong Huang; Yuchen Cui; Oscar Hernan Madrid Padilla (AISTATS, 2025)

Quantile Additive Trend Filtering. - Zhi Zhang 012; Kyle Ritscher; Oscar Hernan Madrid Padilla (AISTATS, 2025)