← Search

Ameesh Shah

6 accepted papers

2026

Automata-Conditioned Cooperative Multi-Agent Reinforcement Learning

ICML 2026poster

We study learning multi-task, multi-agent policies for cooperative, temporal objectives, under centralized training, decentralized execution. In this setting, using automata to represent tasks assigned to agents enables breaking down a team-level objective into simpler, smaller sub-tasks. However, e…

Cited by 0SourceScholar
2026

Learning Affordances at Inference-Time for Vision-Language-Action Models

ICRA 2026poster

Solving complex real-world control tasks often takes multiple tries: if we fail at first, we reflect on what went wrong, and change our strategy accordingly to avoid making the same mistake. In robotics, Vision-Language-Action models (VLAs) offer a promising path towards solving complex control task…

2025

Robust and Diverse Multi-Agent Learning via Rational Policy Gradient

NeurIPS 2025poster

Adversarial optimization algorithms that explicitly search for flaws in agents' policies have been successfully applied to finding robust and diverse policies in the context of multi-agent learning. However, the success of adversarial optimization has been largely limited to zero-sum settings becaus…

Cited by 0SourcecodeScholar
2023

Who Needs to Know? Minimal Knowledge for Optimal Coordination

ICML 2023poster

To optimally coordinate with others in cooperative games, it is often crucial to have information about one’s collaborators: successful driving requires understanding which side of the road to drive on. However, not every feature of collaborators is strategically relevant: the fine-grained accelerat…

2020

Learning Differentiable Programs with Admissible Neural Heuristics

NeurIPS 2020poster

We study the problem of learning differentiable functions expressed as programs in a domain-specific language. Such programmatic models can offer benefits such as composability and interpretability; however, learning them requires optimizing over a combinatorial space of program "architectures". We…

Cited by 58SourcePDFScholar
2019

Representing Formal Languages: A Comparison Between Finite Automata and Recurrent Neural Networks

ICLR 2019poster

We investigate the internal representations that a recurrent neural network (RNN) uses while learning to recognize a regular formal language. Specifically, we train a RNN on positive and negative examples from a regular language, and ask if there is a simple decoding function that maps states of thi…

Cited by 33SourcePDFScholar