Yu Bai
Affiliations:- Salesforce Research, Palo Alto, CA, USA
- Stanford University, CA, USA (PhD 2019)
According to our database1,
Yu Bai
authored at least 50 papers
between 2019 and 2024.
Collaborative distances:
Collaborative distances:
Timeline
Legend:
Book In proceedings Article PhD thesis Dataset OtherLinks
Online presence:
-
on yubai.org
-
on twitter.com
-
on github.com
On csauthors.net:
Bibliography
2024
Efficient Reinforcement Learning With Impaired Observability: Learning to Act With Delayed and Missing State Observations.
IEEE Trans. Inf. Theory, October, 2024
Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs.
CoRR, 2024
Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning.
CoRR, 2024
Is Inverse Reinforcement Learning Harder than Standard Reinforcement Learning? A Theoretical Perspective.
Proceedings of the Forty-first International Conference on Machine Learning, 2024
Transformers as Decision Makers: Provable In-Context Reinforcement Learning via Supervised Pretraining.
Proceedings of the Twelfth International Conference on Learning Representations, 2024
Proceedings of the Twelfth International Conference on Learning Representations, 2024
How Do Transformers Learn In-Context Beyond Simple Functions? A Case Study on Learning with Representations.
Proceedings of the Twelfth International Conference on Learning Representations, 2024
2023
CoRR, 2023
Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023
Efficient RL with Impaired Observability: Learning to Act with Delayed and Missing State Observations.
Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023
Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selection.
Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023
Proceedings of the International Conference on Machine Learning, 2023
Proceedings of the International Conference on Machine Learning, 2023
Proceedings of the Eleventh International Conference on Learning Representations, 2023
Partially Observable RL with B-Stability: Unified Structural Condition and Sharp Sample-Efficient Algorithms.
Proceedings of the Eleventh International Conference on Learning Representations, 2023
Proceedings of the Eleventh International Conference on Learning Representations, 2023
Breaking the Curse of Multiagency: Provably Efficient Decentralized Multi-Agent RL with Function Approximation.
Proceedings of the Thirty Sixth Annual Conference on Learning Theory, 2023
2022
Unified Algorithms for RL with Decision-Estimation Coefficients: No-Regret, PAC, and Reward-Free Learning.
CoRR, 2022
CoRR, 2022
Proceedings of the Uncertainty in Artificial Intelligence, 2022
Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022
Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022
Identifying good directions to escape the NTK regime and efficiently learn low-degree plus sparse polynomials.
Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022
Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022
Proceedings of the International Conference on Machine Learning, 2022
When Can We Learn General-Sum Markov Games with a Large Number of Players Sample-Efficiently?
Proceedings of the Tenth International Conference on Learning Representations, 2022
Proceedings of the Tenth International Conference on Learning Representations, 2022
Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing, 2022
2021
Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021
Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement Learning.
Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021
Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021
Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021
Exact Gap between Generalization Error and Uniform Convergence in Random Feature Models.
Proceedings of the 38th International Conference on Machine Learning, 2021
Proceedings of the 38th International Conference on Machine Learning, 2021
Don't Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classification.
Proceedings of the 38th International Conference on Machine Learning, 2021
Proceedings of the 38th International Conference on Machine Learning, 2021
Near-Optimal Provable Uniform Convergence in Offline Policy Evaluation for Reinforcement Learning.
Proceedings of the 24th International Conference on Artificial Intelligence and Statistics, 2021
2020
Near Optimal Provable Uniform Convergence in Off-Policy Evaluation for Reinforcement Learning.
CoRR, 2020
Taylorized Training: Towards Better Approximation of Neural Network Training at Finite Width.
CoRR, 2020
Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020
Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020
Proceedings of the 37th International Conference on Machine Learning, 2020
Beyond Linearization: On Quadratic and Higher-Order Approximation of Wide Neural Networks.
Proceedings of the 8th International Conference on Learning Representations, 2020
2019
Proximal algorithms for constrained composite optimization, with applications to solving low-rank SDPs.
CoRR, 2019
Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019
Proceedings of the 7th International Conference on Learning Representations, 2019
Proceedings of the 7th International Conference on Learning Representations, 2019
Proceedings of the 7th International Conference on Learning Representations, 2019