Stack Exchange Tagger. - Sanket Vaibhav Mehta; Shagun Sodhani (CoRR, 2015)

Reinforcement Learning Based Conversational Search Assistant. - Milan Aggarwal; Aarushi Arora; Shagun Sodhani; Balaji Krishnamurthy (CoRR, 2017)

Survey of Recent Advances in Visual Question Answering. - Supriya Pandhre; Shagun Sodhani (CoRR, 2017)

News Category Network based Approach for News Source Recommendations. - Swati Gupta; Shagun Sodhani; Dhaval Patel 002; Biplab Banerjee (ICACCI, 2018)

Improving Search Through A3C Reinforcement Learning Based Conversational Agent. - Milan Aggarwal; Aarushi Arora; Shagun Sodhani; Balaji Krishnamurthy (ICCS (2), 2018)

Reproducibility Report for "Learning To Count Objects In Natural Images For Visual Question Answering". - Shagun Sodhani; Vardaan Pahuja (CoRR, 2018)

Memory Augmented Self-Play. - Shagun Sodhani; Vardaan Pahuja (CoRR, 2018)

Spatial Co-location Pattern Mining - A new perspective using Graph Database. - Sanket Vaibhav Mehta; Shagun Sodhani; Dhaval Patel 002 (CoRR, 2018)

Compositional Language Understanding with Text-based Relational Reasoning. - Koustuv Sinha; Shagun Sodhani; William L. Hamilton; Joelle Pineau (CoRR, 2018)

On Training Recurrent Neural Networks for Lifelong Learning. - Shagun Sodhani; Sarath Chandar; Yoshua Bengio (CoRR, 2018)

Environments for Lifelong Reinforcement Learning. - Khimya Khetarpal; Shagun Sodhani; Sarath Chandar; Doina Precup (CoRR, 2018)

CLUTRR: A Diagnostic Benchmark for Inductive Reasoning from Text. - Koustuv Sinha; Shagun Sodhani; Jin Dong; Joelle Pineau; William L. Hamilton (EMNLP/IJCNLP (1), 2019)

Learning Powerful Policies by Using Consistent Dynamics Model. - Shagun Sodhani; Anirudh Goyal; Tristan Deleu; Yoshua Bengio; Sergey Levine; Jian Tang 005 (CoRR, 2019)

Attending Over Triads for Learning Signed Network Embedding. - Shagun Sodhani; Meng Qu; Jian Tang 005 (Frontiers Big Data, 2019)

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

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

Evaluating Logical Generalization in Graph Neural Networks. - Koustuv Sinha; Shagun Sodhani; Joelle Pineau; William L. Hamilton (CoRR, 2020)

Toward Trustworthy AI Development: Mechanisms for Supporting Verifiable Claims. - Miles Brundage; Shahar Avin; Jasmine Wang; Haydn Belfield; Gretchen Krueger; Gillian K. Hadfield; Heidy Khlaaf; Jingying Yang; Helen Toner; Ruth Fong; Tegan Maharaj; Pang Wei Koh; et al. (CoRR, 2020)

Multi-Task Reinforcement Learning as a Hidden-Parameter Block MDP. - Amy Zhang 001; Shagun Sodhani; Khimya Khetarpal; Joelle Pineau (CoRR, 2020)

Ideas for Improving the Field of Machine Learning: Summarizing Discussion from the NeurIPS 2019 Retrospectives Workshop. - Shagun Sodhani; Mayoore S. Jaiswal; Lauren Baker; Koustuv Sinha; Carl Shneider; Peter Henderson 002; Joel Lehman; Ryan Lowe (CoRR, 2020)

A Closer Look at Codistillation for Distributed Training. - Shagun Sodhani; Olivier Delalleau; Mahmoud Assran; Koustuv Sinha; Nicolas Ballas; Michael G. Rabbat (CoRR, 2020)

Toward Training Recurrent Neural Networks for Lifelong Learning. - Shagun Sodhani; Sarath Chandar; Yoshua Bengio (Neural Comput., 2020)

IIRC: Incremental Implicitly-Refined Classification. - Mohamed A. Abdelsalam; Mojtaba Faramarzi; Shagun Sodhani; Sarath Chandar (CVPR, 2021)

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

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

Multi-Task Reinforcement Learning with Context-based Representations. - Shagun Sodhani; Amy Zhang 001; Joelle Pineau (ICML, 2021)

The Neural Race Reduction: Dynamics of Abstraction in Gated Networks. - Andrew M. Saxe; Shagun Sodhani; Sam Jay Lewallen (ICML, 2022)

Robust Policy Learning over Multiple Uncertainty Sets. - Annie Xie; Shagun Sodhani; Chelsea Finn; Joelle Pineau; Amy Zhang 001 (ICML, 2022)

Block Contextual MDPs for Continual Learning. - Shagun Sodhani; Franziska Meier; Joelle Pineau; Amy Zhang 001 (L4DC, 2022)

An Introduction to Lifelong Supervised Learning. - Shagun Sodhani; Mojtaba Faramarzi; Sanket Vaibhav Mehta; Pranshu Malviya; Mohamed A. Abdelsalam; Janarthanan Rajendran; Sarath Chandar (CoRR, 2022)

EpiK-Eval: Evaluation for Language Models as Epistemic Models. - Gabriele Prato 001; Jerry Huang; Prasanna Parthasarathi; Shagun Sodhani; Sarath Chandar (EMNLP, 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 001 (ICLR, 2023)

Sequence Modeling is a Robust Contender for Offline Reinforcement Learning. - Prajjwal Bhargava; Rohan Chitnis; Alborz Geramifard; Shagun Sodhani; Amy Zhang 001 (CoRR, 2023)

Improving Generalization with Approximate Factored Value Functions. - Shagun Sodhani; Sergey Levine; Amy Zhang 001 (Trans. Mach. Learn. Res., 2023)

Do Large Language Models Know How Much They Know? - Gabriele Prato 001; Jerry Huang; Prasanna Parthasarathi; Shagun Sodhani; Sarath Chandar (EMNLP, 2024)

When should we prefer Decision Transformers for Offline Reinforcement Learning? - Prajjwal Bhargava; Rohan Chitnis; Alborz Geramifard; Shagun Sodhani; Amy Zhang 001 (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)

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

Harnessing small projectors and multiple views for efficient vision pretraining. - Arna Ghosh; Kumar Krishna Agrawal; Shagun Sodhani; Adam Oberman; Blake A. Richards (NeurIPS, 2024)

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

Scaling and Distilling Transformer Models for sEMG. - Nicholas Mehlman; Jean-Christophe Gagnon-Audet; Michael Shvartsman; Kelvin Niu; Alexander H. Miller; Shagun Sodhani (Trans. Mach. Learn. Res., 2025)

Do Large Language Models Know How Much They Know? - Gabriele Prato 001; Jerry Huang; Prasannna Parthasarathi; Shagun Sodhani; Sarath Chandar (CoRR, 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; et al. (CoRR, 2025)

AI Research Agents for Machine Learning: Search, Exploration, and Generalization in MLE-bench. - Edan Toledo; Karen Hambardzumyan; Martin Josifoski; Rishi Hazra; Nicolas Mario Baldwin; Alexis Audran-Reiss; Michael Kuchnik; Despoina Magka; Minqi Jiang; Alisia Maria Lupidi; Andrei Lupu; Roberta Raileanu; et al. (CoRR, 2025)