CONF Ewerton_ICRAW_2021/IDIAP An Attention Mechanism for Deep Q-Networks with Applications in Robotic Pushing Ewerton, Marco Calinon, Sylvain Odobez, Jean-Marc https://publications.idiap.ch/index.php/publications/showcite/Ewerton_Idiap-RR-03-2021 Related documents Proc. of Workshop on Emerging paradigms for robotic manipulation: from the lab to the productive world, ICRA 2021 REPORT Ewerton_Idiap-RR-03-2021/IDIAP An Attention Mechanism for Deep Q-Networks with Applications in Robotic Pushing Ewerton, Marco Calinon, Sylvain Odobez, Jean-Marc EXTERNAL https://publications.idiap.ch/attachments/reports/2021/Ewerton_Idiap-RR-03-2021.pdf PUBLIC Idiap-RR-03-2021 2021 Idiap April 2021 Humans effortlessly solve push tasks in everyday life but unlocking these capabilities remains a research challenge in robotics. Physical models are often inaccurate or unattainable. State-of-the-art data-driven approaches learn to compensate for these inaccuracies or get rid of the approximated physical models altogether. Nevertheless, data-driven approaches such as Deep Q-Networks (DQNs) get frequently stuck in local optima in large state-action spaces. We propose an attention mechanism for DQNs to improve their sampling efficiency and demonstrate in simulation experiments with a UR5 robot arm that such a mechanism helps the DQN learn faster and achieve higher performance in a push task involving objects with unknown dynamics.