Sodhani, Shagun

28 publications

NeurIPS 2025 AI Research Agents for Machine Learning: Search, Exploration, and Generalization in MLE-Bench Edan Toledo, Karen Hambardzumyan, Martin Josifoski, Rishi Hazra, Nicolas Baldwin, Alexis Audran-Reiss, Michael Kuchnik, Despoina Magka, Minqi Jiang, Alisia Maria Lupidi, Andrei Lupu, Roberta Raileanu, Tatiana Shavrina, Kelvin Niu, Jean-Christophe Gagnon-Audet, Michael Shvartsman, Shagun Sodhani, Alexander H Miller, Abhishek Charnalia, Derek Dunfield, Carole-Jean Wu, Pontus Stenetorp, Nicola Cancedda, Jakob Nicolaus Foerster, Yoram Bachrach

ICLR 2025 MaestroMotif: Skill Design from Artificial Intelligence Feedback Martin Klissarov, Mikael Henaff, Roberta Raileanu, Shagun Sodhani, Pascal Vincent, Amy Zhang, Pierre-Luc Bacon, Doina Precup, Marlos C. Machado, Pierluca D'Oro

TMLR 2025 Scaling and Distilling Transformer Models for sEMG Nick Mehlman, Jean-Christophe Gagnon-Audet, Michael Shvartsman, Kelvin Niu, Alexander H Miller, Shagun Sodhani

NeurIPS 2025 The Automated LLM Speedrunning Benchmark: Reproducing NanoGPT Improvements Bingchen Zhao, Despoina Magka, Minqi Jiang, Xian Li, Roberta Raileanu, Tatiana Shavrina, Jean-Christophe Gagnon-Audet, Kelvin Niu, Shagun Sodhani, Michael Shvartsman, Andrei Lupu, Alisia Maria Lupidi, Karen Hambardzumyan, Martin Josifoski, Edan Toledo, Thomas Foster, Lucia Cipolina-Kun, Derek Dunfield, Abhishek Charnalia, Alexander H Miller, Oisin Mac Aodha, Jakob Nicolaus Foerster, Yoram Bachrach

NeurIPS 2024 Harnessing Small Projectors and Multiple Views for Efficient Vision Pretraining Arna Ghosh, Kumar Krishna Agrawal, Shagun Sodhani, Adam M. Oberman, Blake A. Richards

ICLR 2024 Motif: Intrinsic Motivation from Artificial Intelligence Feedback Martin Klissarov, Pierluca D'Oro, Shagun Sodhani, Roberta Raileanu, Pierre-Luc Bacon, Pascal Vincent, Amy Zhang, Mikael Henaff

ICLR 2024 TorchRL: A Data-Driven Decision-Making Library for PyTorch Albert Bou, Matteo Bettini, Sebastian Dittert, Vikash Kumar, Shagun Sodhani, Xiaomeng Yang, Gianni De Fabritiis, Vincent Moens

ICLR 2024 When Should We Prefer Decision Transformers for Offline Reinforcement Learning? Prajjwal Bhargava, Rohan Chitnis, Alborz Geramifard, Shagun Sodhani, Amy Zhang

TMLR 2023 Improving Generalization with Approximate Factored Value Functions Shagun Sodhani, Sergey Levine, Amy Zhang

NeurIPSW 2023 Motif: Intrinsic Motivation from Artificial Intelligence Feedback Martin Klissarov, Pierluca D'Oro, Shagun Sodhani, Roberta Raileanu, Pierre-Luc Bacon, Pascal Vincent, Amy Zhang, Mikael Henaff

NeurIPSW 2023 Motif: Intrinsic Motivation from Artificial Intelligence Feedback Martin Klissarov, Pierluca D'Oro, Shagun Sodhani, Roberta Raileanu, Pierre-Luc Bacon, Pascal Vincent, Amy Zhang, Mikael Henaff

ICLR 2023 VIP: Towards Universal Visual Reward and Representation via Value-Implicit Pre-Training Yecheng Jason Ma, Shagun Sodhani, Dinesh Jayaraman, Osbert Bastani, Vikash Kumar, Amy Zhang

L4DC 2022 Block Contextual MDPs for Continual Learning Shagun Sodhani, Franziska Meier, Joelle Pineau, Amy Zhang

ICLRW 2022 Improving Generalization with Approximate Factored Value Functions Shagun Sodhani, Sergey Levine, Amy Zhang

ICML 2022 Robust Policy Learning over Multiple Uncertainty Sets Annie Xie, Shagun Sodhani, Chelsea Finn, Joelle Pineau, Amy Zhang

ICML 2022 The Neural Race Reduction: Dynamics of Abstraction in Gated Networks Andrew Saxe, Shagun Sodhani, Sam Jay Lewallen

NeurIPSW 2022 Towards Universal Visual Reward and Representation via Value-Implicit Pre-Training Yecheng Jason Ma, Shagun Sodhani, Dinesh Jayaraman, Osbert Bastani, Vikash Kumar, Amy Zhang

NeurIPSW 2022 VIP: Towards Universal Visual Reward and Representation via Value-Implicit Pre-Training Yecheng Jason Ma, Shagun Sodhani, Dinesh Jayaraman, Osbert Bastani, Vikash Kumar, Amy Zhang

NeurIPSW 2022 VIP: Towards Universal Visual Reward and Representation via Value-Implicit Pre-Training Yecheng Jason Ma, Shagun Sodhani, Dinesh Jayaraman, Osbert Bastani, Vikash Kumar, Amy Zhang

NeurIPSW 2021 Block Contextual MDPs for Continual Learning Shagun Sodhani, Franziska Meier, Joelle Pineau, Amy Zhang

CVPR 2021 IIRC: Incremental Implicitly-Refined Classification Mohamed Abdelsalam, Mojtaba Faramarzi, Shagun Sodhani, Sarath Chandar

NeurIPSW 2021 IIRC: Incremental Implicitly-Refined Classification Mohamed Abdelsalam, Mojtaba Faramarzi, Shagun Sodhani, Sarath Chandar

ICLR 2021 Learning Robust State Abstractions for Hidden-Parameter Block MDPs Amy Zhang, Shagun Sodhani, Khimya Khetarpal, Joelle Pineau

ICML 2021 Multi-Task Reinforcement Learning with Context-Based Representations Shagun Sodhani, Amy Zhang, Joelle Pineau

ICLR 2021 Recurrent Independent Mechanisms Anirudh Goyal, Alex Lamb, Jordan Hoffmann, Shagun Sodhani, Sergey Levine, Yoshua Bengio, Bernhard Schölkopf

ICMLW 2020 Evaluating Logical Generalization in Graph Neural Networks Koustuv Sinha, Shagun Sodhani, Joelle Pineau, William L. Hamilton

ICML 2020 Invariant Causal Prediction for Block MDPs Amy Zhang, Clare Lyle, Shagun Sodhani, Angelos Filos, Marta Kwiatkowska, Joelle Pineau, Yarin Gal, Doina Precup

ICLR 2020 Reinforcement Learning with Competitive Ensembles of Information-Constrained Primitives Anirudh Goyal, Shagun Sodhani, Jonathan Binas, Xue Bin Peng, Sergey Levine, Yoshua Bengio