arXiv:cs.LG· Songyuan Zhang, Oswin So, Eric Yang Yu, Matthew Cleaveland, Peter Crowley-Dolen, Chuchu Fan·· 4 小时前AI 评分37
LASER:面向支持约束熵正则离线 RL 的潜空间伴随匹配
LASER: Latent Space Adjoint Matching for Support-Constrained Entropy-Regularized Offline RL
AI 导读
研究者提出离线 RL 算法 LASER,通过潜空间伴随匹配实现熵正则化的潜空间强化学习,并使用表达力强的流策略,同时避免沿时间反向传播。在 40 个不同数据质量的 OGBench 任务上,LASER 取得 SOTA 性能,且所有任务共用固定超参数,优于经过任务与数据集调优的基线。该工作已被 NeurIPS 2026 接收。
正文
Abstract:While offline reinforcement learning (RL) enables policy optimization from static datasets without costly online interaction, it remains bottlenecked by the risk of executing out-of-distribution (OOD) actions. Recent approaches mitigate this by learning a behavior-cloning policy through flow matching and then performing RL within its constrained latent space. However, naively optimizing the latent policy can easily cause the policy to collapse into a brittle mode or exploit sharp artifacts of the learned critic. In this work, we find that entropy regularization is essential in latent-space RL for addressing these challenges. We introduce LASER, a novel offline RL algorithm that applies latent-space adjoint matching to achieve entropy-regularized latent-space RL with expressive flow policies while avoiding backpropagation through time. Through comprehensive experiments on 40 challenging OGBench tasks with varying dataset qualities, we show that LASER achieves state-of-the-art performance. Notably, LASER uses fixed method-specific hyperparameters across all tasks and outperforms the evaluated baselines, including those with task- and dataset-specific tuning, which highlights the robust applicability of LASER. Project website: this https URL.
| Comments: | 29 pages, 16 figures. Accepted at NeurIPS 2026 |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Robotics (cs.RO); Optimization and Control (math.OC); Machine Learning (stat.ML) |
| Cite as: | arXiv:2610.08989 [cs.LG] |
| (or arXiv:2610.08989v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.08989 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Songyuan Zhang [view email]
[v1]
Tue, 6 Oct 2026 18:47:21 UTC (5,974 KB)
来源:arXiv:cs.LG · arxiv.org