TACO: Temporal Latent Action-driven Contrastive Loss For Visual Reinforcement Learning

Abstract

Despite recent progress in reinforcement learning (RL) from raw pixel data, sample inefficiency continues to present a substantial obstacle. Prior works have attempted to address this challenge by creating self-supervised auxiliary tasks, aiming to enrich the agent's learned representations with control-relevant information for future state prediction. However, these objectives are often insufficient to learn representations that can represent the optimal policy or value function, and they often consider tasks with small, abstract discrete action spaces and thus overlook the importance of action representation learning in continuous control. In this paper, we introduce TACO: Temporal Action-driven Contrastive Learning, a simple yet powerful temporal contrastive learning approach that facilitates the concurrent acquisition of latent state and action representations for agents. TACO simultaneously learns a state and an action representation by optimizing the mutual information between re

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keyzheng2023taco

Related papers

• Return-based Contrastive Representation Learning For Reinforcement Learning (2021)12.17
• MOOSS: Mask-enhanced Temporal Contrastive Learning For Smooth State Evolution In Visual Reinforcement Learning (2024)0.00
• Temporal Abstractions-augmented Temporally Contrastive Learning: An Alternative To The Laplacian In RL (2022)0.00
• Contrastive Learning As Goal-conditioned Reinforcement Learning (2022)0.00
• Learning Temporally-consistent Representations For Data-efficient Reinforcement Learning (2021)0.00
• Improving Context-based Meta-reinforcement Learning With Self-supervised Trajectory Contrastive Learning (2021)0.00
• Value-consistent Representation Learning For Data-efficient Reinforcement Learning (2022)0.00
• A Temporally Correlated Latent Exploration For Reinforcement Learning (2024)0.00