Multi-Task Reinforcement Learning with Context-based Representations
Shagun Sodhani, Amy Zhang, Joelle Pineau
People
Amy Zhang
PhD Student
I am a PhD candidate at McGill University in the Reasoning and Learning Lab supervised by Joelle Pineau and a researcher at Facebook Artificial Intelligence Research. I hold dual B.S. degrees in EECS and math, and an M.Eng. in EECS from MIT. I previously did research in computer vision, privacy-preserving recommendation, and network coding. I am also a board member of Women in Machine Learning.
Interests
I work on state abstractions and deep reinforcement learning, with a focus on model-based methods and generalization.
Latest Publications
ICML
Learning Invariant Representations for Reinforcement Learning without Reconstruction
Amy Zhang, Rowan McAllister, Roberto Calandra, Yarin Gal, Sergey Levine
Learning Robust State Abstractions for Hidden-Parameter Block MDPs
Amy Zhang, Shagun Sodhani, Khimya Khetarpal, Joelle Pineau
MBRL-Lib: A Modular Library for Model-based Reinforcement Learning
Luis Pineda, Brandon Amos, Amy Zhang, Nathan O. Lambert, Roberto Calandra
Stable Policy Optimization via Off-Policy Divergence Regularization
Ahmed Touati, Amy Zhang, Joelle Pineau, Pascal Vincent
Invariant Causal Prediction for Block MDPs
Amy Zhang, Clare Lyle, Shagun Sodhani, Angelos Filos, Marta Kwiatkowska, Joelle Pineau, Yarin Gal, Doina Precup