QHyer: Q-conditioned Hybrid Attention-mamba Transformer for Offline Goal-conditioned RL
Xing Lei, Jincheng Wang, Xuetao Zhang, Donglin Wang
Abstract
Offline goal-conditioned RL (GCRL) learns goal-reaching policies from static datasets, but real-world environments are often partially observable, so the collected trajectories are only partly consistent with the Markov assumption while other segments remain history-dependent. History-aware sequence models such as Decision Transformer (DT) are a natural fit for long-term dependency modeling, yet pure attention is inefficient and brittle when handling local Markovian structure and long-range context simultaneously. Although recent hybrid architectures (e.g., LSDT) introduce local extractors, their fixed-window extraction cannot adapt the effective memory to varying dependency lengths, often truncating long-range context instead of compressing it. Moreover, under sparse rewards, return-to-go (RTG) becomes non-discriminative across sub-trajectories, offering little guidance for stitching goal-reaching behaviors from diverse demonstrations. To address these limitations, we propose QHyer (Q-conditioned Hybrid Attention-Mamba Transformer), which replaces RTG with a Normalizing Flows (NFs) parameterized goal-reaching Q-estimator used directly as conditioning tokens, and a gated Hybrid Attention-Mamba backbone whose selective state-space dynamics enable content-adaptive history compression while attention captures global goal-directed dependencies. Extensive experiments on OGBench and D4RL demonstrate that QHyer achieves state-of-the-art performance on both non-Markovian and Markovian datasets, validating its effectiveness for diverse scenarios.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7ef84f28-c506-45e0-a13a-9b129e12b071Builds on46
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Conservative Q-Learning for Offline Reinforcement LearningAviral Kumar, Aurick Zhou, George Tucker, Sergey LevineNeurIPS 2020 · 2,881 citations
- Decision Transformer: Reinforcement Learning via Sequence ModelingLili Chen, Kevin Lu, Aravind Rajeswaran, Kimin Lee et al.NeurIPS 2021 · 2,557 citations
- Offline Reinforcement Learning with Implicit Q-LearningIlya Kostrikov, Ashvin Nair, Sergey LevineICLR 2022 · 1,402 citations
- A Minimalist Approach to Offline Reinforcement LearningScott Fujimoto, Shixiang Shane GuNeurIPS 2021 · 1,292 citations
Related papers
- Long-Short Decision Transformer: Bridging Global and Local Dependencies for Generalized Decision-MakingJincheng Wang, Penny Karanasou, Pengyuan Wei, Elia Gatti et al.ICLR 2025
- Decision Mamba: Reinforcement Learning via Hybrid Selective Sequence ModelingSili Huang, Jifeng Hu, Zhejian Yang, Liwei Yang et al.NeurIPS 2024 · 65 citations
- Q-learning Decision Transformer: Leveraging Dynamic Programming for Conditional Sequence Modelling in Offline RLTaku Yamagata, Ahmed Khalil, Raúl Santos-RodríguezICML 2023 · 121 citations
- Less is More: an Attention-free Sequence Prediction Modeling for Offline Embodied LearningWei Huang, Jianshu Zhang, Leiyu Wang, Heyue Li et al.NeurIPS 2025
- Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RLQi Lv, Xiang Deng, Gongwei Chen, Michael Yu Wang et al.NeurIPS 2024 · 25 citations
