解决自动驾驶强化学习中采样效率低的问题
FAST: A Framework for Aligned Sampling and Training in Parallel Reinforcement Learning for Autonomous Driving

- 通过虚拟延续机制保持并行采样同步
- 实现比单片段基准快1.78倍的运行速度
- 适合需要高效训练的自动驾驶仿真系统
深度强化学习对闭环自动驾驶至关重要,但采样效率严重受限。标准并行采样虽缓解此问题,却受‘慢速者效应’影响:单个环境提前终止需同步重初始化,导致样本利用率低下和高延迟。为此,我们提出FAST框架,专为闭环仿真设计。FAST采用动态并行采样对齐(DPSA),通过虚拟延续延长终止片段,解耦采样循环与个体终止。基于并行片段终止率动态触发全局截断,有效消除提前重置瓶颈,且不牺牲数据多样性。此外,为严格保证理论一致性,引入缩放掩码填充优化(SMPO),利用有效性掩码与自适应损失归一化,消除辅助填充数据带来的偏差。实证表明,FAST在保持统计无偏的前提下,相比单片段基线实现至少1.78倍的墙钟速度提升。
原文摘要 · Abstract (English)
Deep reinforcement learning is pivotal for closed-loop autonomous driving yet remains constrained by severe bottlenecks in sampling efficiency. Standard parallel sampling mitigates this but suffers from the straggler effect, where the premature termination of a single environment necessitates a synchronized batch re-initialization, leading to suboptimal sample utilization and prohibitive re-initialization latency. To address this, we propose FAST, a synchronous parallel framework tailored for closed-loop simulation. Specifically, FAST employs Dynamic Parallel Sampling Alignment (DPSA) to maintain vectorization synchronization by extending terminated episodes via virtual continuation, thereby decoupling the sampling loop from individual terminations. By dynamically triggering global truncation based on the termination rate of parallel clips, FAST effectively eliminates the bottleneck of premature resets without sacrificing data diversity. Furthermore, to strictly preserve theoretical consistency, we incorporate a Scaled Mask-Padding Optimization (SMPO) that leverages validity masking and adaptive loss normalization to nullify the bias from auxiliary padding data. Empirical evaluations demonstrate that FAST achieves at least a 1.78 times wall-clock speedup over the single-clip baseline while preserving statistical unbiasedness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。