Publications

2026

  1. Provably Efficient and Agile Randomized Q-Learning
    Wang, He, Xu, Xingyu, and Chi, Yuejie
    AISTATS 2026

2025

  1. DSMentor: Enhancing Data Science Agents with Curriculum Learning and Online Knowledge Accumulation
    Wang, He, Li, Alexander Hanbo, Hu, Yiqun, Zhang, Sheng, Kobayashi, Hideo, Zhang, Jiani, Zhu, Henry, Hang, Chung-Wei, and Ng, Patrick
    arxiv 2025

2024

  1. Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes
    Wang, He, Shi, Laixi, and Chi, Yuejie
    RLC 2024
  2. Communication-Efficient Federated Optimization over Semi-Decentralized Networks
    Wang, He, and Chi, Yuejie
    IEEE Transactions on Signal and Information Processing over Networks (TSIPN) & ICASSP 2024

2022

  1. Distributed Optimization with Coupling Constraints
    Wu, Xuyang,  Wang, He, and Lu, Jie
    Transactions on Automatic Control (TAC) 2022
  2. Stable and Efficient Shapley Value-Based Reward Reallocation for Multi-Agent Reinforcement Learning of Autonomous Vehicles
    Han, Songyang,  Wang, He, Su, Sanbao, Shi, Yuanyuan, and Miao, Fei
    ICRA 2022

2021

  1. Decentralized Statistical Inference with Unrolled Graph Neural Networks
    Wang, He, Shen, Yifei, Wang, Ziyuan, Li, Dongsheng, Zhang, Jun, Letaief, Khaled B., and Lu, Jie
    CDC 2021

2020

  1. An Inexact Fenchel Dual Gradient Algorithm for Distributed Optimization
    Wang, He, and Lu, Jie
    ICCA 2020