Research
I am broadly interested in applying deep reinforcement learning, imitation learning, and meta-learning towards developing intelligent behavior in robotics
and decision-making agents. To this end I'm actively applying to PhD programs in Fall 2022!
|
|
A General Pipeline for Autonomous Robotic Reinforcement Learning
Archit Sharma*, Ahmed M. Ahmed*, Rehaan Ahmad, Chelsea Finn
Conference on Robot Learning (CoRL) Workshop on Learning to Adapt and Improve in the Real World, 2022
paper /
poster /
We present an autonomous system that extends an algorithm for reset-free reinforcement learning known as MEDAL
to using visual inputs, allowing for a real world set-up where we use a small number of demonstrations
to learn reward functions for both solving a given task and resetting the environment.
We validate our approach in simulation and show that it can autonomously improve on manipulation tasks
for 24+ hours with limited initial resets and no additional reward specification on a Franka Panda Robot
|
|
Cross-Trajectory Representation Learning for Zero-Shot Generalization in RL
Bogdan Mazoure*, Ahmed M. Ahmed*, Patrick MacAlpine, R Devon Hjelm, Andrey Kolobov
International Conference on Learning Representations (ICLR), 2022
arxiv /
code /
poster /
We propose Cross-Trajectory Representation Learning (CTRL) a novel self-supervised learning
objective that maintains a set of trajectories embeddings representative of different behaviors
onto which we can project any given trajectory from the transition dynamics. This allows CTRL to avoid
overfitting to rewards in the encoder and improved generalization on the challenging Procgen Benchmark compared
to prior work. We also show a connection to psuedo-bisimulation metrics in Reinforcement Learning.
|
|
Hierarchical Imitation via Bayesian Meta-Learning
Ahmed M. Ahmed, James Harrison, Chelsea Finn
Black in AI NeurIPS Workshop, 2021
paper /
code /
workshop /
Devised a novel approach for hierarchical imitation learning through bayesian meta-learning algorithms. The
goal is to learn a multi-skill policy from expert demonstrations which are segmented through a adapted changepoint
detection scheme. Selected for an oral presentation at the Black in AI workshop and best theoretical course
project for Stanford's CS 236: Deep Generative Models
|
Outreach
Outside of my research, I'm passionate about addressing issues of diversity and inclusion in academia at large. To this end I've worked on improving outreach
and inclusion in CS research through my work as mentor CURIS, the Stanford CS department's REU program. I helped spearhead initatives
such as the CURIS fellows program, aimed to provide research opportunities for historically underrepresented students and PURE
which provides research funding for First-Generation/Low-Income students.
|
|