统一框架同时解决自动驾驶中的运动模拟、预测与规划问题。
UniMotion: A Unified Motion Framework for Simulation, Prediction and Planning
- 基于解码器架构,共享交互建模与时空推理能力。
- 在Waymo数据集上联合训练后表现超越现有方法。
- 适合需要多任务协同的自动驾驶系统开发。
运动模拟、预测与规划是自动驾驶的核心任务,各自针对动态交通场景建模与推理。尽管目标不同(如生成多样运动状态或估算最优轨迹),这些任务本质上依赖于共同能力:理解多智能体交互、建模运动行为、处理时空动态。然而,现有方法通常采用专用模型设计,阻碍了跨任务泛化与系统扩展。更关键的是,任务间潜在的相互促进被忽视。为此,我们提出UniMotion,一个统一运动框架,在共享结构基础上兼顾各任务需求。基于解码器仅架构,UniMotion采用专门的交互模式与定制训练策略,实现多任务协同。该设计支持联合优化与表征共享,也允许针对性微调以专精特定任务。在Waymo Open Motion Dataset上的实验表明,联合训练带来稳健泛化与有效集成。进一步微调后,UniMotion在多项运动任务中达到领先性能,验证了其作为自动驾驶通用可扩展解决方案的潜力。
原文摘要 · Abstract (English)
Motion simulation, prediction and planning are foundational tasks in autonomous driving, each essential for modeling and reasoning about dynamic traffic scenarios. While often addressed in isolation due to their differing objectives, such as generating diverse motion states or estimating optimal trajectories, these tasks inherently depend on shared capabilities: understanding multi-agent interactions, modeling motion behaviors, and reasoning over temporal and spatial dynamics. Despite this underlying commonality, existing approaches typically adopt specialized model designs, which hinders cross-task generalization and system scalability. More critically, this separation overlooks the potential mutual benefits among tasks. Motivated by these observations, we propose UniMotion, a unified motion framework that captures shared structures across motion tasks while accommodating their individual requirements. Built on a decoder-only Transformer architecture, UniMotion employs dedicated interaction modes and tailored training strategies to simultaneously support these motion tasks. This unified design not only enables joint optimization and representation sharing but also allows for targeted fine-tuning to specialize in individual tasks when needed. Extensive experiments on the Waymo Open Motion Dataset demonstrate that joint training leads to robust generalization and effective task integration. With further fine-tuning, UniMotion achieves state-of-the-art performance across a range of motion tasks, establishing it as a versatile and scalable solution for autonomous driving.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。