arXiv:2605.01862cs.LG2026-05

用自适应记忆压缩机制提升离线目标导向强化学习的泛化能力

QHyer: Q-conditioned Hybrid Attention-mamba Transformer for Offline Goal-conditioned RL

论文配图:QHyer: Q-conditioned Hybrid Attention-mamba Transformer for Offline Goal-conditioned RL
图 1 · 摘自论文原文
  • 引入门控混合注意力-马尔可夫结构,自适应压缩历史信息
  • 用状态相关Q估计替代返回值,增强稀疏奖励下的行为拼接能力
  • 在非马尔可夫与马尔可夫数据上均达领先性能,适合复杂轨迹学习

离线目标导向强化学习(GCRL)从静态数据集中学习目标达成策略,但真实数据常具部分可观测性和历史依赖性,混合了马尔可夫与非马尔可夫特性,违背标准RL假设。决策变压器(DT)等序列模型能建模长时依赖,但纯注意力机制在同时处理局部马尔可夫结构和长程上下文时效率低且脆弱。尽管近期混合架构(如LSDT)引入局部提取器改善局部依赖建模,但固定窗口提取无法适应时间异质场景中的变化依赖长度,常截断长程上下文而非自适应压缩内容。此外,序列离线GCRL面临关键瓶颈:在稀疏奖励下,回报至目标(RTG)在子轨迹间失去区分性,难以指导从不同演示中拼接目标达成行为。为此,我们提出QHyer,用流参数化、状态条件化的目标达成Q估计器替代RTG,支持跨演示拼接;并设计门控混合注意力-马尔可夫主干,实现内容自适应历史压缩,同时保留局部动态。大量实验表明,QHyer在非马尔可夫与马尔可夫数据集上均达到当前最优性能,验证其在多样化场景中的有效性。

原文摘要 · Abstract (English)

Offline goal-conditioned RL (GCRL) learns goal-reaching policies from static datasets, but real-world datasets are often partially observable and history-dependent, exhibiting a mix of Markovian and non-Markovian that violate standard RL assumptions. 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 to improve local dependencies modeling, the fixed-window extraction cannot adapt its effective memory to varying dependency lengths in temporally heterogeneous settings, often truncating long-range context rather than compressing its content adaptively. Moreover, sequential offline GCRL faces a key bottleneck: under sparse rewards, return-to-go (RTG) becomes non-discriminative across sub-trajectories, providing little guidance signal for stitching goal-reaching behaviors from diverse demonstrations. To address these, we propose \textbf{QHyer}, which replaces RTG with a flow-parameterized, state-conditioned goal-reaching Q-estimator to support stitching across demonstrations, and introduces a gated Hybrid Attention-Mamba backbone that performs content-adaptive history compression while preserving local dynamics. Extensive experiments demonstrate that \textbf{QHyer} achieves state-of-the-art performance on both non-Markovian and Markovian datasets, validating its effectiveness for diverse scenarios.

强化学习序列建模离线学习目标导向

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。