Wang, Tianhao
31 publications
NeurIPSW
2024
Can Neural Networks Achieve Optimal Computational-Statistical Tradeoff? an Analysis on Single-Index Model
NeurIPS
2024
Unveiling Induction Heads: Provable Training Dynamics and Feature Learning in Transformers
ICMLW
2024
Unveiling Induction Heads: Provable Training Dynamics and Feature Learning in Transformers
AISTATS
2023
Finding Regularized Competitive Equilibria of Heterogeneous Agent Macroeconomic Models via Reinforcement Learning
NeurIPS
2022
A Simple and Provably Efficient Algorithm for Asynchronous Federated Contextual Linear Bandits
NeurIPS
2022
Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descent
NeurIPS
2021
Provably Efficient Reinforcement Learning with Linear Function Approximation Under Adaptivity Constraints