ViSA: Visited-State Augmentation for Generalized Goal-Space Contrastive Reinforcement Learning
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2026
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866918395052556288 |
|---|---|
| author | Nakamura, Issa Yamanokuchi, Tomoya Kadokawa, Yuki Qu, Jia Otsub, Shun Miyamoto, Ken Miwa, Shotaro Matsubara, Takamitsu |
| author_facet | Nakamura, Issa Yamanokuchi, Tomoya Kadokawa, Yuki Qu, Jia Otsub, Shun Miyamoto, Ken Miwa, Shotaro Matsubara, Takamitsu |
| contents | Goal-Conditioned Reinforcement Learning (GCRL) is a framework for learning a policy that can reach arbitrarily given goals. In particular, Contrastive Reinforcement Learning (CRL) provides a framework for policy updates using an approximation of the value function estimated via contrastive learning, achieving higher sample efficiency compared to conventional methods. However, since CRL treats the visited state as a pseudo-goal during learning, it can accurately estimate the value function only for limited goals. To address this issue, we propose a novel data augmentation approach for CRL called ViSA (Visited-State Augmentation). ViSA consists of two components: 1) generating augmented state samples, with the aim of augmenting hard-to-visit state samples during on-policy exploration, and 2) learning consistent embedding space, which uses an augmented state as auxiliary information to regularize the embedding space by reformulating the objective function of the embedding space based on mutual information. We evaluate ViSA in simulation and real-world robotic tasks and show improved goal-space generalization, which permits accurate value estimation for hard-to-visit goals. Further details can be found on the project page: https://issa-n.github.io/projectPage_ViSA/ |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2603_14887 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | ViSA: Visited-State Augmentation for Generalized Goal-Space Contrastive Reinforcement Learning Nakamura, Issa Yamanokuchi, Tomoya Kadokawa, Yuki Qu, Jia Otsub, Shun Miyamoto, Ken Miwa, Shotaro Matsubara, Takamitsu Robotics Goal-Conditioned Reinforcement Learning (GCRL) is a framework for learning a policy that can reach arbitrarily given goals. In particular, Contrastive Reinforcement Learning (CRL) provides a framework for policy updates using an approximation of the value function estimated via contrastive learning, achieving higher sample efficiency compared to conventional methods. However, since CRL treats the visited state as a pseudo-goal during learning, it can accurately estimate the value function only for limited goals. To address this issue, we propose a novel data augmentation approach for CRL called ViSA (Visited-State Augmentation). ViSA consists of two components: 1) generating augmented state samples, with the aim of augmenting hard-to-visit state samples during on-policy exploration, and 2) learning consistent embedding space, which uses an augmented state as auxiliary information to regularize the embedding space by reformulating the objective function of the embedding space based on mutual information. We evaluate ViSA in simulation and real-world robotic tasks and show improved goal-space generalization, which permits accurate value estimation for hard-to-visit goals. Further details can be found on the project page: https://issa-n.github.io/projectPage_ViSA/ |
| title | ViSA: Visited-State Augmentation for Generalized Goal-Space Contrastive Reinforcement Learning |
| topic | Robotics |
| url | https://arxiv.org/abs/2603.14887 |