Reinforcement Learning with Convex Constraints. - Sobhan Miryoosefi; Kianté Brantley; Hal Daumé III; Miroslav Dudík; Robert E. Schapire (NeurIPS, 2019)

Constrained episodic reinforcement learning in concave-convex and knapsack settings. - Kianté Brantley; Miroslav Dudík; Thodoris Lykouris; Sobhan Miryoosefi; Max Simchowitz; Aleksandrs Slivkins; Wen Sun 002 (NeurIPS, 2020)

Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithms. - Chi Jin 001; Qinghua Liu; Sobhan Miryoosefi (NeurIPS, 2021)

Provable Reinforcement Learning with a Short-Term Memory. - Yonathan Efroni; Chi Jin 001; Akshay Krishnamurthy; Sobhan Miryoosefi (ICML, 2022)

A Simple Reward-free Approach to Constrained Reinforcement Learning. - Sobhan Miryoosefi; Chi Jin 001 (ICML, 2022)

Provable Reinforcement Learning with Constraints and Function Approximation - Sobhan Miryoosefi (2022)

Efficient Training of Language Models using Few-Shot Learning. - Sashank J. Reddi; Sobhan Miryoosefi; Stefani Karp; Shankar Krishnan; Satyen Kale; Seungyeon Kim 001; Sanjiv Kumar (ICML, 2023)

ReST meets ReAct: Self-Improvement for Multi-Step Reasoning LLM Agent. - Renat Aksitov; Sobhan Miryoosefi; Zonglin Li; Daliang Li; Sheila Babayan; Kavya Kopparapu; Zachary Fisher; Ruiqi Guo; Sushant Prakash; Pranesh Srinivasan; Manzil Zaheer; Felix X. Yu; et al. (CoRR, 2023)

On the Inductive Bias of Stacking Towards Improving Reasoning. - Nikunj Saunshi; Stefani Karp; Shankar Krishnan; Sobhan Miryoosefi; Sashank Jakkam Reddi; Sanjiv Kumar (NeurIPS, 2024)

Landscape-Aware Growing: The Power of a Little LAG. - Stefani Karp; Nikunj Saunshi; Sobhan Miryoosefi; Sashank J. Reddi; Sanjiv Kumar (CoRR, 2024)

Efficient stagewise pretraining via progressive subnetworks. - Abhishek Panigrahi; Nikunj Saunshi; Kaifeng Lyu; Sobhan Miryoosefi; Sashank J. Reddi; Satyen Kale; Sanjiv Kumar (ICLR, 2025)