|
I am a Computer Science PhD student in at University of Southern California, co-advised by Prof. Joseph J. Lim and Prof. Erdem Bıyık. I was fortunate to intern at Meta Reality Labs, Microsoft Research Montreal and Naver AI, Seoul. Before joining USC, I spent two years in Seoul, working at Samsung Research Korea. Earlier, I graduated from IIT Delhi, where I worked under the guidance of Prof. Sumeet Agarwal and Prof. Rajakrishnan Rajkumar. Research Goal: To build adaptive and capable agents for both physical and virtual worlds. I work on reinforcement learning algorithms and architectures that learn under complex action spaces that are large, unseen, varying, or difficult to optimize. Email | Twitter | CV | Google Scholar | LinkedIn |
|
|
|
 |
Ayush Jain, Norio Kosaka, Xinhu Li, Kyung-Min Kim, Erdem Bıyık, Joseph J. Lim preprint We identify that TD3 gets stuck in local optima in tasks with complex Q-functions and propose a new actor architecture to find better optima. arXiv |
|
Grace Zhang*, Ayush Jain*, Injune Hwang, Shao-Hua Sun, Joseph J. Lim preprint Sharing behaviors between tasks to improve exploration for multitask reinforcement learning. arXiv | Project Page |
|
Ayush Jain*, Norio Kosaka*, Kyung-Min Kim, Joseph J. Lim International Conference on Learning Representations (ICLR), 2022 For optimal decision-making under a varying action space, we learn the relations between the available actions using a graph-attention network based policy architecture. Paper | Project Page | Code | Talk |
|
Ayush Jain*, Andrew Szot*, Joseph J. Lim International Conference on Machine Learning (ICML), 2020 Our proposed RL framework enables agents to solve sequential decision-making tasks even when the available actions (tools or skills) have not been seen before. Paper | Project Page | arXiv | Code | Talk | Environment |
 |
Ayush Jain*, Vishal Singh*, Sidharth Ranjan*, Rajakrishnan Rajkumar, Sumeet Agarwal Workshop on Linguistic Complexity and Natural Language Processing, COLING 2018 This work investigates the extent to which word order choices in Hindi language are influenced by the drive to minimize the information variance in a sentence. |
|
|
|
Teaching Assistant (USC):
|
|
|
|
|
Credits to the Coolest template! |