Generalization to New Actions in Reinforcement Learning. - Ayush Jain 003; Andrew Szot; Joseph J. Lim (ICML, 2020)

Know Your Action Set: Learning Action Relations for Reinforcement Learning. - Ayush Jain 003; Norio Kosaka; Kyung-Min Kim; Joseph J. Lim (ICLR, 2022)

Efficient Multi-Task Reinforcement Learning via Selective Behavior Sharing. - Grace Zhang; Ayush Jain 003; Injune Hwang; Shao-Hua Sun; Joseph J. Lim (CoRR, 2023)

Mitigating Suboptimality of Deterministic Policy Gradients in Complex Q-functions. - Ayush Jain 003; Norio Kosaka; Xinhu Li; Kyung-Min Kim; Erdem Biyik; Joseph J. Lim (CoRR, 2024)

QMP: Q-switch Mixture of Policies for Multi-Task Behavior Sharing. - Grace Zhang; Ayush Jain 003; Injune Hwang; Shao-Hua Sun; Joseph J. Lim (ICLR, 2025)