Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2024
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866916663696293888 |
|---|---|
| author | Zhang, Xinyu Qiu, Wenjie Li, Yi-Chen Yuan, Lei Jia, Chengxing Zhang, Zongzhang Yu, Yang |
| author_facet | Zhang, Xinyu Qiu, Wenjie Li, Yi-Chen Yuan, Lei Jia, Chengxing Zhang, Zongzhang Yu, Yang |
| contents | Developing policies that can adjust to non-stationary environments is essential for real-world reinforcement learning applications. However, learning such adaptable policies in offline settings, with only a limited set of pre-collected trajectories, presents significant challenges. A key difficulty arises because the limited offline data makes it hard for the context encoder to differentiate between changes in the environment dynamics and shifts in the behavior policy, often leading to context misassociations. To address this issue, we introduce a novel approach called Debiased Offline Representation for fast online Adaptation (DORA). DORA incorporates an information bottleneck principle that maximizes mutual information between the dynamics encoding and the environmental data, while minimizing mutual information between the dynamics encoding and the actions of the behavior policy. We present a practical implementation of DORA, leveraging tractable bounds of the information bottleneck principle. Our experimental evaluation across six benchmark MuJoCo tasks with variable parameters demonstrates that DORA not only achieves a more precise dynamics encoding but also significantly outperforms existing baselines in terms of performance. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_11317 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics Zhang, Xinyu Qiu, Wenjie Li, Yi-Chen Yuan, Lei Jia, Chengxing Zhang, Zongzhang Yu, Yang Machine Learning Artificial Intelligence Developing policies that can adjust to non-stationary environments is essential for real-world reinforcement learning applications. However, learning such adaptable policies in offline settings, with only a limited set of pre-collected trajectories, presents significant challenges. A key difficulty arises because the limited offline data makes it hard for the context encoder to differentiate between changes in the environment dynamics and shifts in the behavior policy, often leading to context misassociations. To address this issue, we introduce a novel approach called Debiased Offline Representation for fast online Adaptation (DORA). DORA incorporates an information bottleneck principle that maximizes mutual information between the dynamics encoding and the environmental data, while minimizing mutual information between the dynamics encoding and the actions of the behavior policy. We present a practical implementation of DORA, leveraging tractable bounds of the information bottleneck principle. Our experimental evaluation across six benchmark MuJoCo tasks with variable parameters demonstrates that DORA not only achieves a more precise dynamics encoding but also significantly outperforms existing baselines in terms of performance. |
| title | Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2402.11317 |