arXiv cs.AI· Songyuan Zhang, Oswin So, Eric Yang Yu, Matthew Cleaveland, Peter Crowley-Dolen, Chuchu Fan·· 2 天前AI 评分21
LASER:面向支持约束熵正则离线 RL 的潜在空间伴随匹配
LASER: Latent Space Adjoint Matching for Support-Constrained Entropy-Regularized Offline RL
AI 导读
LASER 是一种离线 RL 算法,通过潜在空间伴随匹配实现熵正则化的潜在空间 RL,并使用表达力强的流策略,同时避免沿时间反向传播。在 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.08989v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.08989 arXiv-issued DOI via DataCite |
Submission history
From: Songyuan Zhang [view email]
[v1]
Tue, 6 Oct 2026 18:47:21 UTC (5,974 KB)
[v2]
Wed, 14 Oct 2026 18:23:06 UTC (5,974 KB)
来源:arXiv cs.AI · arxiv.org