Vec-QMDP:面向实时自动驾驶的CPU矢量化POMDP规划
Vec-QMDP: Vectorized POMDP Planning on CPUs for Real-Time Autonomous Driving
AI总结:
提出Vec-QMDP,一种CPU原生并行规划器,通过数据导向设计、层次化并行和矢量化碰撞检测,实现比最先进串行规划器快227-1073倍的速度,在毫秒级延迟下完成大规模自动驾驶规划。
摘要:
现实世界机器人任务(如自动驾驶)中的不确定性规划需要在巨大的高维信念空间中进行推理,导致计算密集型问题。虽然并行化提供了可扩展性,但现有的混合CPU-GPU求解器由于主机-设备同步延迟和SIMT架构上的分支发散而面临关键瓶颈,限制了其在实时规划中的实用性并阻碍了实际机器人部署。我们提出了Vec-QMDP,一种CPU原生并行规划器,它将POMDP搜索与现代CPU的SIMD架构对齐,实现了比最先进串行规划器快227倍到1073倍的加速。Vec-QMDP采用数据导向设计(DOD),将分散的、基于指针的数据结构重构为连续的、缓存高效的内存布局。我们进一步引入了一种层次化并行方案:将子树分布到独立的CPU核心和SIMD通道上,实现完全矢量化的树扩展和碰撞检测。通过树间的UCB负载均衡和用于粗粒度碰撞检测的矢量化STR树,效率得以最大化。在大规模自动驾驶基准测试中,Vec-QMDP以毫秒级延迟实现了最先进的规划性能,将CPU确立为大规模不确定性规划的高性能计算平台。
英文摘要:
Planning under uncertainty for real-world robotics tasks, such as autonomous driving, requires reasoning in enormous high-dimensional belief spaces, rendering the problem computationally intensive. While parallelization offers scalability, existing hybrid CPU-GPU solvers face critical bottlenecks due to host-device synchronization latency and branch divergence on SIMT architectures, limiting their utility for real-time planning and hindering real-robot deployment. We present Vec-QMDP, a CPU-native parallel planner that aligns POMDP search with modern CPUs' SIMD architecture, achieving $227\times$--$1073\times$ speedup over state-of-the-art serial planners. Vec-QMDP adopts a Data-Oriented Design (DOD), refactoring scattered, pointer-based data structures into contiguous, cache-efficient memory layouts. We further introduce a hierarchical parallelism scheme: distributing sub-trees across independent CPU cores and SIMD lanes, enabling fully vectorized tree expansion and collision checking. Efficiency is maximized with the help of UCB load balancing across trees and a vectorized STR-tree for coarse-level collision checking. Evaluated on large-scale autonomous driving benchmarks, Vec-QMDP achieves state-of-the-art planning performance with millisecond-level latency, establishing CPUs as a high-performance computing platform for large-scale planning under uncertainty.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。