Publications (117)
From Pen to Prompt: How Creative Writers Integrate AI into their Writing Practice
Alicia Guo, Shreya Sathyanarayanan, Leijie Wang +2
Learning a Fast Mixing Exogenous Block MDP using a Single Trajectory
Alexander Levine, Peter Stone, Amy Zhang
Robot Air Hockey: A Manipulation Testbed for Robot Learning with Reinforcement Learning
Caleb Chuck, Carl Qi, Michael J. Munje +13
Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning
Tongzhou Wang, Antonio Torralba, Phillip Isola +1
Dual RL: Unification and New Methods for Reinforcement and Imitation Learning
Harshit Sikchi, Qinqing Zheng, Amy Zhang +1
Decentralizing Platform Power: A Design Space of Multi-level Governance in Online Social Platforms
Shagun Jhaver, Seth Frey, Amy Zhang
$f$-Policy Gradients: A General Framework for Goal Conditioned RL using $f$-Divergences
Siddhant Agarwal, Ishan Durugkar, Peter Stone +1
A Dissection of Overfitting and Generalization in Continuous Reinforcement Learning
Amy Zhang, Nicolas Ballas, Joelle Pineau
Guiding Diffusion with Deep Geometric Moments: Balancing Fidelity and Variation
Sangmin Jung, Utkarsh Nath, Yezhou Yang +5
Viblio: Introducing Credibility Signals and Citations to Video-Sharing Platforms
Emelia Hughes, Renee Wang, Prerna Juneja +3
Who Puts the "Social" in "Social Computing"?: Using A Neurodiversity Framing to Review Social Computing Research
Philip Baillargeon, Jina Yoon, Amy Zhang
Unified Auto-Encoding with Masked Diffusion
Philippe Hansen-Estruch, Sriram Vishwanath, Amy Zhang +1
Learning Representations for Pixel-based Control: What Matters and Why?
Manan Tomar, Utkarsh A. Mishra, Amy Zhang +1
Multi-Task Reinforcement Learning with Context-based Representations
Shagun Sodhani, Amy Zhang, Joelle Pineau
Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observability
Dibya Ghosh, Jad Rahme, Aviral Kumar +3
Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangement
Michael Chang, Alyssa L. Dayan, Franziska Meier +3
LIV: Language-Image Representations and Rewards for Robotic Control
Yecheng Jason Ma, William Liang, Vaidehi Som +4
Exploiting Local Dynamics Regularity for Reusable Skills in Offline Hierarchical RL
Sarthak Dayal, Abhinav Peri, Carl Qi +4
Motif: Intrinsic Motivation from Artificial Intelligence Feedback
Martin Klissarov, Pierluca D'Oro, Shagun Sodhani +5
EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation
Carl Qi, Dan Haramati, Tal Daniel +2
Latent State Marginalization as a Low-cost Approach for Improving Exploration
Dinghuai Zhang, Aaron Courville, Yoshua Bengio +3
Automating Document Classification with Distant Supervision to Increase the Efficiency of Systematic Reviews
Xiaoxiao Li, Rabah Al-Zaidy, Amy Zhang +3
Mapping the world population one building at a time
Tobias G. Tiecke, Xianming Liu, Amy Zhang +8
Factored Latent Action World Models
Zizhao Wang, Chang Shi, Jiaheng Hu +4
Plan2Vec: Unsupervised Representation Learning by Latent Plans
Ge Yang, Amy Zhang, Ari S. Morcos +3
Stable Policy Optimization via Off-Policy Divergence Regularization
Ahmed Touati, Amy Zhang, Joelle Pineau +1
Reshaping Undergraduate Computer Science Education in the Generative AI Era
Yi-Chieh Lee, Nattapat Boonprakong, Yugin Tan +20
The PokeAgent Challenge: Competitive and Long-Context Learning at Scale
Seth Karten, Jake Grigsby, Tersoo Upaa +28
Diffusion-DICE: In-Sample Diffusion Guidance for Offline Reinforcement Learning
Liyuan Mao, Haoran Xu, Xianyuan Zhan +2
Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion
Dan Haramati, Carl Qi, Tal Daniel +3
Building Human Values into Recommender Systems: An Interdisciplinary Synthesis
Jonathan Stray, Alon Halevy, Parisa Assar +18
Structure in Deep Reinforcement Learning: A Survey and Open Problems
Aditya Mohan, Amy Zhang, Marius Lindauer
Invariant Causal Prediction for Block MDPs
Amy Zhang, Clare Lyle, Shagun Sodhani +5
Online Intrinsic Rewards for Decision Making Agents from Large Language Model Feedback
Qinqing Zheng, Mikael Henaff, Amy Zhang +2
Learning Invariant Representations for Reinforcement Learning without Reconstruction
Amy Zhang, Rowan McAllister, Roberto Calandra +2
Out-of-Distribution Generalization via Risk Extrapolation (REx)
David Krueger, Ethan Caballero, Joern-Henrik Jacobsen +5
Guess Who Rated This Movie: Identifying Users Through Subspace Clustering
Amy Zhang, Nadia Fawaz, Stratis Ioannidis +1
Bisimulation Makes Analogies in Goal-Conditioned Reinforcement Learning
Philippe Hansen-Estruch, Amy Zhang, Ashvin Nair +2
A Recipe for Stable Offline Multi-agent Reinforcement Learning
Dongsu Lee, Daehee Lee, Amy Zhang
Language Control Diffusion: Efficiently Scaling through Space, Time, and Tasks
Edwin Zhang, Yujie Lu, Shinda Huang +2
A Survey of Zero-shot Generalisation in Deep Reinforcement Learning
Robert Kirk, Amy Zhang, Edward Grefenstette +1
SMORE: Score Models for Offline Goal-Conditioned Reinforcement Learning
Harshit Sikchi, Rohan Chitnis, Ahmed Touati +3
Null Counterfactual Factor Interactions for Goal-Conditioned Reinforcement Learning
Caleb Chuck, Fan Feng, Carl Qi +4
Decoupling Dynamics and Reward for Transfer Learning
Amy Zhang, Harsh Satija, Joelle Pineau
Benchmarking Massively Parallelized Multi-Task Reinforcement Learning for Robotics Tasks
Viraj Joshi, Zifan Xu, Bo Liu +2
Reevaluating Policy Gradient Methods for Imperfect-Information Games
Max Rudolph, Nathan Lichtle, Sobhan Mohammadpour +6
VIP: Towards Universal Visual Reward and Representation via Value-Implicit Pre-Training
Yecheng Jason Ma, Shagun Sodhani, Dinesh Jayaraman +3
Confidence Contours: Uncertainty-Aware Annotation for Medical Semantic Segmentation
Andre Ye, Quan Ze Chen, Amy Zhang
ExPO: Unlocking Hard Reasoning with Self-Explanation-Guided Reinforcement Learning
Ruiyang Zhou, Shuozhe Li, Amy Zhang +1
Contrastive Distillation Is a Sample-Efficient Self-Supervised Loss Policy for Transfer Learning
Chris Lengerich, Gabriel Synnaeve, Amy Zhang +4
Provably Efficient Representation Selection in Low-rank Markov Decision Processes: From Online to Offline RL
Weitong Zhang, Jiafan He, Dongruo Zhou +2
Information Borrowing in Regression Models
Amy Zhang, Le Bao, Michael J. Daniels
A Dual Approach to Imitation Learning from Observations with Offline Datasets
Harshit Sikchi, Caleb Chuck, Amy Zhang +1
Learning Action-based Representations Using Invariance
Max Rudolph, Caleb Chuck, Kevin Black +3
Zero-Shot Reinforcement Learning via Function Encoders
Tyler Ingebrand, Amy Zhang, Ufuk Topcu
Augmented Conditioning Is Enough For Effective Training Image Generation
Jiahui Chen, Amy Zhang, Adriana Romero-Soriano
SkiLD: Unsupervised Skill Discovery Guided by Factor Interactions
Zizhao Wang, Jiaheng Hu, Caleb Chuck +5
Natural Environment Benchmarks for Reinforcement Learning
Amy Zhang, Yuxin Wu, Joelle Pineau
Generalization Across Observation Shifts in Reinforcement Learning
Anuj Mahajan, Amy Zhang
Composable Planning with Attributes
Amy Zhang, Adam Lerer, Sainbayar Sukhbaatar +2
Learning Robust Reasoning through Guided Adversarial Self-Play
Shuozhe Li, Vaishnav Tadiparthi, Kwonjoon Lee +6
Feedback Neural Network for Weakly Supervised Geo-Semantic Segmentation
Xianming Liu, Amy Zhang, Tobias Tiecke +2
Regularized Latent Dynamics Prediction is a Strong Baseline For Behavioral Foundation Models
Pranaya Jajoo, Harshit Sikchi, Siddhant Agarwal +3
Towards Robust Offline Reinforcement Learning under Diverse Data Corruption
Rui Yang, Han Zhong, Jiawei Xu +4
Robust Policy Learning over Multiple Uncertainty Sets
Annie Xie, Shagun Sodhani, Chelsea Finn +2
Reinforcement Learning via Value Gradient Flow
Haoran Xu, Kaiwen Hu, Somayeh Sojoudi +1
AI and the Future of Digital Public Squares
Beth Goldberg, Diana Acosta-Navas, Michiel Bakker +24
Designing Word Filter Tools for Creator-led Comment Moderation
Shagun Jhaver, Quan Ze Chen, Detlef Knauss +1
AMAGO-2: Breaking the Multi-Task Barrier in Meta-Reinforcement Learning with Transformers
Jake Grigsby, Justin Sasek, Samyak Parajuli +3
Fast Adaptation with Behavioral Foundation Models
Harshit Sikchi, Andrea Tirinzoni, Ahmed Touati +6
Learning Causal State Representations of Partially Observable Environments
Amy Zhang, Zachary C. Lipton, Luis Pineda +5
Improving Sample Efficiency in Model-Free Reinforcement Learning from Images
Denis Yarats, Amy Zhang, Ilya Kostrikov +3
Automated Discovery of Functional Actual Causes in Complex Environments
Caleb Chuck, Sankaran Vaidyanathan, Stephen Giguere +3
A Model-Based Solution to the Offline Multi-Agent Reinforcement Learning Coordination Problem
Paul Barde, Jakob Foerster, Derek Nowrouzezahrai +1
Guess Who Rated This Movie: Identifying Users Through Subspace Clustering
Amy Zhang, Nadia Fawaz, Stratis Ioannidis +1
CREStE: Scalable Mapless Navigation with Internet Scale Priors and Counterfactual Guidance
Arthur Zhang, Harshit Sikchi, Amy Zhang +1
Towards General-Purpose Model-Free Reinforcement Learning
Scott Fujimoto, Pierluca D'Oro, Amy Zhang +2
Filtered Reasoning Score: Evaluating Reasoning Quality on a Model's Most-Confident Traces
Manas Pathak, Xingyao Chen, Shuozhe Li +2
The paper introduces the Filtered Reasoning Score (FRS), a metric that evaluates the quality of reasoning traces from large language models by focusing on the most confident genera…
An Optimal Discriminator Weighted Imitation Perspective for Reinforcement Learning
Haoran Xu, Shuozhe Li, Harshit Sikchi +2
An Investigation of Time Reversal Symmetry in Reinforcement Learning
Brett Barkley, Amy Zhang, David Fridovich-Keil
CARE-RFT: Confidence-Anchored Reinforcement Finetuning for Reliable Reasoning in Large Language Models
Shuozhe Li, Jincheng Cao, Bodun Hu +3
Bringing Social Computing to Secondary School Classrooms
Kianna Bolante, Kevin Chen, Quan Ze Chen +1
Managing your Private and Public Data: Bringing down Inference Attacks against your Privacy
Salman Salamatian, Amy Zhang, Flavio du Pin Calmon +5
Offline Action-Free Learning of Ex-BMDPs by Comparing Diverse Datasets
Alexander Levine, Peter Stone, Amy Zhang
GitHub OSS Governance File Dataset
Yibo Yan, Seth Frey, Amy Zhang +2
Trajectory-Refined Distillation
Li Jiang, Haoran Xu, Yichuan Ding +1
Proto Successor Measure: Representing the Behavior Space of an RL Agent
Siddhant Agarwal, Harshit Sikchi, Peter Stone +1
Accelerating Exploration with Unlabeled Prior Data
Qiyang Li, Jason Zhang, Dibya Ghosh +2
Learning Robust State Abstractions for Hidden-Parameter Block MDPs
Amy Zhang, Shagun Sodhani, Khimya Khetarpal +1
AutoCAT: Reinforcement Learning for Automated Exploration of Cache-Timing Attacks
Mulong Luo, Wenjie Xiong, Geunbae Lee +6
Dissecting IoT Device Provisioning Process
Rostand A. K. Fezeu, Timothy J. Salo, Amy Zhang +1
Do Users Want Platform Moderation or Individual Control? Examining the Role of Third-Person Effects and Free Speech Support in Shaping Moderation Preferences
Shagun Jhaver, Amy Zhang
TRAM: Test-Time Risk Adaptation with Mixture of Agents
Mohamad Fares El Hajj Chehade, Amrit Singh Bedi, Amy Zhang +1
Unifying Agent Interaction and World Information for Multi-agent Coordination
Dongsu Lee, Daehee Lee, Yaru Niu +3
Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning
Zihan Ding, Amy Zhang, Yuandong Tian +1
Private Hierarchical Governance for Encrypted Messaging
Armin Namavari, Barry Wang, Sanketh Menda +5
Model-Invariant State Abstractions for Model-Based Reinforcement Learning
Manan Tomar, Amy Zhang, Roberto Calandra +2
MBRL-Lib: A Modular Library for Model-based Reinforcement Learning
Luis Pineda, Brandon Amos, Amy Zhang +2
Efficient RL for optimizing conversation level outcomes with an LLM-based tutor
Hyunji Nam, Omer Gottesman, Amy Zhang +3
Denoised MDPs: Learning World Models Better Than the World Itself
Tongzhou Wang, Simon S. Du, Antonio Torralba +3