A Study of Policy Gradient on a Class of Exactly Solvable Models. - Gavin McCracken; Colin Daniels; Rosie Zhao; Anna M. Brandenberger; Prakash Panangaden; Doina Precup (CoRR, 2020)

Bridging the gap between supervised classification and unsupervised topic modelling for social-media assisted crisis management. - Mikael Brunila; Rosie Zhao; Andrei Mircea; Sam Lumley; Renée Sieber (CoRR, 2021)

Continuous MDP Homomorphisms and Homomorphic Policy Gradient. - Sahand Rezaei-Shoshtari; Rosie Zhao; Prakash Panangaden; David Meger; Doina Precup (NeurIPS, 2022)

Leaf multiplicity in a Bienaymé-Galton-Watson tree. - Anna M. Brandenberger; Luc Devroye; Marcel K. Goh; Rosie Zhao (Discret. Math. Theor. Comput. Sci., 2022)

Boolean functions with small approximate spectral norm. - TsunMing Cheung; Hamed Hatami; Rosie Zhao; Itai Zilberstein (Electron. Colloquium Comput. Complex., 2022)

Lower Bound Methods for Sign-rank and their Limitations. - Hamed Hatami; Pooya Hatami; William Pires; Ran Tao 013; Rosie Zhao (Electron. Colloquium Comput. Complex., 2022)

Loss of Plasticity in Continual Deep Reinforcement Learning. - Zaheer Abbas; Rosie Zhao; Joseph Modayil; Adam White 001; Marlos C. Machado (CoLLAs, 2023)

On the peel number and the leaf-height of Galton-Watson trees. - Luc Devroye; Marcel K. Goh; Rosie Y. Zhao (Comb. Probab. Comput., 2023)

Feature emergence via margin maximization: case studies in algebraic tasks. - Depen Morwani; Benjamin L. Edelman; Costin-Andrei Oncescu; Rosie Zhao; Sham M. Kakade (ICLR, 2024)

Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learning. - Nikhil Vyas 001; Depen Morwani; Rosie Zhao; Gal Kaplun; Sham M. Kakade; Boaz Barak (ICML, 2024)

Deconstructing What Makes a Good Optimizer for Language Models. - Rosie Zhao; Depen Morwani; David Brandfonbrener; Nikhil Vyas 001; Sham M. Kakade (CoRR, 2024)

SOAP: Improving and Stabilizing Shampoo using Adam. - Nikhil Vyas 001; Depen Morwani; Rosie Zhao; Itai Shapira; David Brandfonbrener; Lucas Janson; Sham M. Kakade (CoRR, 2024)

Creating a Cooperative AI Policymaking Platform through Open Source Collaboration. - Aiden Lewington; Alekhya Vittalam; Anshumaan Singh; Anuja Uppuluri; Arjun Ashok; Ashrith Mandayam Athmaram; Austin Milt; Benjamin Smith; Charlie Weinberger; Chatanya Sarin; Christoph Bergmeir; Cliff Chang; et al. (CoRR, 2024)

Policy Gradient Methods in the Presence of Symmetries and State Abstractions. - Prakash Panangaden; Sahand Rezaei-Shoshtari; Rosie Zhao; David Meger; Doina Precup (J. Mach. Learn. Res., 2024)

SOAP: Improving and Stabilizing Shampoo using Adam for Language Modeling. - Nikhil Vyas 001; Depen Morwani; Rosie Zhao; Itai Shapira; David Brandfonbrener; Lucas Janson; Sham M. Kakade (ICLR, 2025)

Deconstructing What Makes a Good Optimizer for Autoregressive Language Models. - Rosie Zhao; Depen Morwani; David Brandfonbrener; Nikhil Vyas 001; Sham M. Kakade (ICLR, 2025)

Distributional Scaling Laws for Emergent Capabilities. - Rosie Zhao; Tian Qin; David Alvarez-Melis; Sham M. Kakade; Naomi Saphra (CoRR, 2025)

Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining. - Rosie Zhao; Alexandru Meterez; Sham M. Kakade; Cengiz Pehlevan; Samy Jelassi; Eran Malach (CoRR, 2025)

Inside you are many wolves: Using cognitive models to interpret value trade-offs in LLMs. - Sonia K. Murthy; Rosie Zhao; Jennifer Hu 001; Sham M. Kakade; Markus Wulfmeier; Peng Qian; Tomer D. Ullman (CoRR, 2025)