Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentation

1 July 2021

Papers citing "Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentation"

50 / 80 papers shown

Title
Merging and Disentangling Views in Visual Reinforcement Learning for Robotic Manipulation Abdulaziz Almuzairee Rohan Patil Dwait Bhatt Henrik I. Christensen 34 0 0 07 May 2025
Accelerating Visual Reinforcement Learning with Separate Primitive Policy for Peg-in-Hole Tasks Zichun Xu Zhaomin Wang Yuntao Li Lei Zhuang Zhiyuan Zhao Guocai Yang Jingdong Zhao 26 0 0 21 Apr 2025
Novel Demonstration Generation with Gaussian Splatting Enables Robust One-Shot Manipulation Sizhe Yang Wenye Yu Jia Zeng Jun Lv Kerui Ren Cewu Lu Dahua Lin Jiangmiao Pang 3DGS 50 1 0 17 Apr 2025
HASARD: A Benchmark for Vision-Based Safe Reinforcement Learning in Embodied Agents Tristan Tomilin Meng Fang Mykola Pechenizkiy 55 0 0 11 Mar 2025
Efficient Reinforcement Learning Through Adaptively Pretrained Visual Encoder Yuhan Zhang Guoqing Ma Guangfu Hao Liangxuan Guo Yang Chen S. Yu OnRL 74 0 0 08 Feb 2025
State Combinatorial Generalization In Decision Making With Conditional Diffusion Models Xintong Duan Yutong He Fahim Tajwar Wen-Tse Chen Ruslan Salakhutdinov Jeff Schneider OffRL AI4CE 99 0 0 22 Jan 2025
Reinforcement Learning with Euclidean Data Augmentation for State-Based Continuous Control Jinzhu Luo Dingyang Chen Qi Zhang OffRL 26 0 0 16 Oct 2024
Make the Pertinent Salient: Task-Relevant Reconstruction for Visual Control with Distractions Kyungmin Kim JB Lanier Pierre Baldi Charless C. Fowlkes Roy Fox 28 1 0 13 Oct 2024
Zero-Shot Generalization of Vision-Based RL Without Data Augmentation S. Batra Gaurav Sukhatme OffRL DRL 26 1 0 09 Oct 2024
Focus On What Matters: Separated Models For Visual-Based RL Generalization Di Zhang Bowen Lv Hai Zhang Feifan Yang Junqiao Zhao Hang Yu Chang Huang Hongtu Zhou Chen Ye Changjun Jiang 27 3 0 29 Sep 2024
Adapting Image-based RL Policies via Predicted Rewards Weiyao Wang Xinyuan Fang Gregory D. Hager 41 0 0 23 Jul 2024
Domain Adaptation of Visual Policies with a Single Demonstration Weiyao Wang Gregory D. Hager 38 0 0 23 Jul 2024
Learning to Manipulate Anywhere: A Visual Generalizable Framework For Reinforcement Learning Zhecheng Yuan Tianming Wei Shuiqi Cheng Gu Zhang Yuanpei Chen Huazhe Xu 45 22 0 22 Jul 2024
Green Screen Augmentation Enables Scene Generalisation in Robotic Manipulation Eugene Teoh Sumit Patidar Xiao Ma Stephen James 40 5 0 10 Jul 2024
SiT: Symmetry-Invariant Transformers for Generalisation in Reinforcement Learning Matthias Weissenbacher Rishabh Agarwal Yoshinobu Kawahara OffRL 32 1 0 21 Jun 2024
Adapting Pretrained ViTs with Convolution Injector for Visuo-Motor Control Dongyoon Hwang ByungKun Lee Hojoon Lee Hyunseung Kim Jaegul Choo 53 0 0 10 Jun 2024
A Recipe for Unbounded Data Augmentation in Visual Reinforcement Learning Abdulaziz Almuzairee Nicklas Hansen Henrik I. Christensen 42 6 0 27 May 2024
Vision Transformers for End-to-End Vision-Based Quadrotor Obstacle Avoidance Anish Bhattacharya Nishanth Rao Dhruv Parikh Pratik Kunapuli Nikolai Matni Vijay R. Kumar Nikolai Matni Vijay Kumar 66 6 0 16 May 2024
Learning Latent Dynamic Robust Representations for World Models Ruixiang Sun Hongyu Zang Xin-hui Li Riashat Islam 39 5 0 10 May 2024
Point Cloud Models Improve Visual Robustness in Robotic Learners Skand Peri Iain Lee Chanho Kim Fuxin Li Tucker Hermans Stefan Lee 3DPC 44 3 0 29 Apr 2024
3D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations Yanjie Ze Gu Zhang Kangning Zhang Chenyuan Hu Muhan Wang Huazhe Xu VGen 37 77 0 06 Mar 2024
Revisiting Data Augmentation in Deep Reinforcement Learning Jianshu Hu Yunpeng Jiang Paul Weng OffRL 21 5 0 19 Feb 2024
Understanding What Affects Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidence Jiafei Lyu Le Wan Xiu Li Zongqing Lu CML OffRL 35 4 0 05 Feb 2024
Generalizable Visual Reinforcement Learning with Segment Anything Model Ziyu Wang Yanjie Ze Yifei Sun Zhecheng Yuan Huazhe Xu VLM 25 8 0 28 Dec 2023
MaDi: Learning to Mask Distractions for Generalization in Visual Deep Reinforcement Learning Bram Grooten Tristan Tomilin G. Vasan M. E. Taylor A. R. Mahmood Meng Fang Mykola Pechenizkiy D. Mocanu 24 8 0 23 Dec 2023
The Generalization Gap in Offline Reinforcement Learning Ishita Mediratta Qingfei You Minqi Jiang Roberta Raileanu OffRL 78 10 0 10 Dec 2023
FoMo Rewards: Can we cast foundation models as reward functions? Ekdeep Singh Lubana Johann Brehmer P. D. Haan Taco S. Cohen OffRL LRM 48 2 0 06 Dec 2023
Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learning Nicholas Corrado Yu-Tao Qu John U. Balis Adam Labiosa Josiah P. Hanna OffRL 30 2 0 27 Oct 2023
Understanding when Dynamics-Invariant Data Augmentations Benefit Model-Free Reinforcement Learning Updates Nicholas Corrado Josiah P. Hanna 21 4 0 26 Oct 2023
TD-MPC2: Scalable, Robust World Models for Continuous Control Nicklas Hansen Hao Su Xiaolong Wang MU 32 121 0 25 Oct 2023
Finetuning Offline World Models in the Real World Yunhai Feng Nicklas Hansen Ziyan Xiong Chandramouli Rajagopalan Xiaolong Wang OffRL OnRL 17 20 0 24 Oct 2023
Learning Generalizable Agents via Saliency-Guided Features Decorrelation Sili Huang Yanchao Sun Jifeng Hu Siyuan Guo Hechang Chen Yi-Ju Chang Lichao Sun Bo Yang 21 4 0 08 Oct 2023
Adversarial Style Transfer for Robust Policy Optimization in Deep Reinforcement Learning Md Masudur Rahman Yexiang Xue 26 4 0 29 Aug 2023
Improving Generalization in Visual Reinforcement Learning via Conflict-aware Gradient Agreement Augmentation Siao Liu Zhaoyu Chen Yang Liu Yuzheng Wang Dingkang Yang ... Ziqing Zhou Xie Yi Wei Li Wenqiang Zhang Zhongxue Gan 33 22 0 02 Aug 2023
Seeing is not Believing: Robust Reinforcement Learning against Spurious Correlation Wenhao Ding Laixi Shi Yuejie Chi Ding Zhao OOD 27 18 0 15 Jul 2023
RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalization Zhecheng Yuan Sizhe Yang Pu Hua C. Chang Kaizhe Hu Huazhe Xu OOD OffRL 24 17 0 15 Jul 2023
Decomposing the Generalization Gap in Imitation Learning for Visual Robotic Manipulation Annie Xie Lisa Lee Ted Xiao Chelsea Finn 23 54 0 07 Jul 2023
MoVie: Visual Model-Based Policy Adaptation for View Generalization Sizhe Yang Yanjie Ze Huazhe Xu 52 12 0 03 Jul 2023
TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learning Ruijie Zheng Xiyao Wang Yanchao Sun Shuang Ma Jieyu Zhao Huazhe Xu Hal Daumé Furong Huang 43 35 0 22 Jun 2023
On the Efficacy of 3D Point Cloud Reinforcement Learning Z. Ling Yuan Yao Xuanlin Li H. Su 3DPC 31 13 0 11 Jun 2023
On the Importance of Exploration for Generalization in Reinforcement Learning Yiding Jiang J. Zico Kolter Roberta Raileanu UQCV OffRL 30 20 0 08 Jun 2023
Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Data Chongyi Zheng Benjamin Eysenbach Homer Walke Patrick Yin Kuan Fang Ruslan Salakhutdinov Sergey Levine SSL OffRL 34 4 0 06 Jun 2023
Normalization Enhances Generalization in Visual Reinforcement Learning Lu Li Jiafei Lyu Guozheng Ma Zilin Wang Zhen Yang Xiu Li Zhiheng Li OOD 17 8 0 01 Jun 2023
Learning Better with Less: Effective Augmentation for Sample-Efficient Visual Reinforcement Learning Guozheng Ma Linrui Zhang Haoyu Wang Lu Li Zilin Wang Zhen Wang Li Shen Xueqian Wang Dacheng Tao 42 10 0 25 May 2023
Conditional Mutual Information for Disentangled Representations in Reinforcement Learning Mhairi Dunion Trevor A. McInroe K. Luck Josiah P. Hanna Stefano V. Albrecht OOD DRL 20 17 0 23 May 2023
Model-Based Reinforcement Learning with Isolated Imaginations Minting Pan Xiangming Zhu Yitao Zheng Yunbo Wang Xiaokang Yang 18 0 0 27 Mar 2023
A Survey of Demonstration Learning André Rosa de Sousa Porfírio Correia Luís A. Alexandre OffRL 28 17 0 20 Mar 2023
Sample-efficient Adversarial Imitation Learning Dahuin Jung Hyungyu Lee Sung-Hoon Yoon SSL 18 2 0 14 Mar 2023
The Dormant Neuron Phenomenon in Deep Reinforcement Learning Ghada Sokar Rishabh Agarwal P. S. Castro Utku Evci CLL 40 88 0 24 Feb 2023
Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractions Qiyuan Liu Qi Zhou Rui Yang Jie Wang OffRL OOD 25 14 0 12 Feb 2023