publications

Please see Google Scholar for more recent works and arXiv papers.

2026

  1. Preprint
    Train Less, Learn More: Adaptive and Efficient Rollout Optimization for Group-Based Reinforcement Learning
    Zhi Zhang, Zhen Han, and  others
    2026
  2. arXiv
    Flow Matching Generalizes Through Discretization Bias
    Zhixin Zhou, Zhi Zhang, and Arash Amini
    arXiv preprint 2026
  3. ICLR
    EDIVAL-Agent: An Object-Centric Framework for Automated, Fine-Grained Evaluation of Multi-Turn Editing
    Tianyu Chen, Yasi Zhang, Zhi Zhang, and  others
    In International Conference on Learning Representations 2026
  4. ICLR
    Single Index Bandits: Generalized Linear Contextual Bandits with Unknown Reward Functions
    Yue Kang, Mingshuo Liu, Bongsoo Yi, Jing Lyu, Zhi Zhang, Doudou Zhou, and Yao Li
    In International Conference on Learning Representations 2026

2025

  1. AISTATS
    Quantile Additive Trend Filtering
    Zhi Zhang, Kyle Ritscher, and Oscar Madrid Padilla
    In International Conference on Artificial Intelligence and Statistics 2025
  2. arXiv
    Dense ReLU Neural Network for Temporal-Spatial Model
    Carlos Padilla, Zhi Zhang, Oscar Padilla, and Daren Wang
    arXiv preprint 2025
  3. AISTATS
    Statistical Guarantees for Lifelong Reinforcement Learning Using PAC-Bayes Theory
    Zhi Zhang, Chris Chow, Yasi Zhang, and  others
    In International Conference on Artificial Intelligence and Statistics 2025
  4. arXiv
    Unlocking the Potential of Text-to-Image Diffusion with PAC-Bayesian Theory
    Eric Hanchen Jiang, Yasi Zhang, and Zhi Zhang
    2025

2024

  1. arXiv
    DODT: Enhanced Online Decision Transformer Learning through Dreamer’s Actor-Critic Trajectory Forecasting
    Eric Hanchen Jiang, Zhi Zhang, and  others
    2024
  2. AISTATS
    Multivariate Time Series Forecasting by Graph Attention Networks with Theoretical Guarantees
    Zhi Zhang, Weijian Li, and Han Liu
    In International Conference on Artificial Intelligence and Statistics 2024

2023