轻量模型实时预测多车轨迹,兼顾精度与安全
MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- 用教师-学生架构,分层蒸馏复杂模型知识
- 推理延迟仅14.6毫秒,参数压缩6.2倍
- 结合强化学习优化避障、舒适性等实际表现
轨迹预测是自动驾驶系统的核心,直接影响碰撞检测、行为规划与控制。在密集交互、行为多样、未来多模态且车载算力有限的条件下,现有方法或计算成本高,或无法纠正教师模型的安全偏差。本文提出MAVEN-T,一种强化型异构蒸馏框架:高容量教师模型通过环视图编码器建模有向局部交互,结合时序滤波与移位窗口注意力,并用稀疏专家混合头生成驾驶意图相关的未来轨迹;轻量学生模型由GRU与挤压激励结构构成,采用特征、注意力和语义级蒸馏训练,并通过近端策略优化(PPO)奖励函数进一步优化避障、舒适性与前进效率。训练过程引入复杂度感知课程与弹性权重固化以稳定阶段训练。在NGSIM、HighD、MoCAD、Argoverse~2及Waymo Open Motion Dataset上的实验表明,该学生模型实现6.2倍参数压缩、3.7倍推理加速,推理延迟低至14.6毫秒(NVIDIA Jetson AGX Orin),同时保持竞争力的预测精度。
原文摘要 · Abstract (English)
Trajectory prediction is a key component of autonomous driving systems because future motions directly affect collision checking, behavior planning, and control. The task remains challenging under dense interactions, heterogeneous behaviors, multimodal futures, and limited on-board computation. Existing graph, attention, and generative predictors improve interaction reasoning or uncertainty modeling, but their high-capacity designs are often costly for real-time deployment. Lightweight predictors and conventional distillation reduce inference cost, yet usually rely on static imitation and do not explicitly correct safety-relevant teacher bias. This paper proposes \textbf{MAVEN-T}, a reinforced heterogeneous distillation framework for real-time multi-agent trajectory prediction. A high-capacity teacher models directed local interactions with a surround-aware graph encoder, combines efficient temporal filtering with shifted-window spatial attention, and decodes maneuver-specific futures through a sparse Mixture-of-Experts head. A compact GRU--Squeeze-and-Excitation student with a Low-Rank Adapted policy head is trained by feature-, attention-, and semantic-level distillation. To align prediction with downstream behavior, the student is further refined by Proximal Policy Optimization rewards for collision avoidance, comfort, and progress, while a complexity-aware curriculum and Elastic Weight Consolidation stabilize stage-wise training. Experiments on NGSIM, HighD, MoCAD, Argoverse~2, and the Waymo Open Motion Dataset evaluate accuracy, efficiency, generalization, robustness, and closed-loop safety. The student achieves 6.2$\times$ parameter compression, 3.7$\times$ inference acceleration, and 14.6,ms latency on an NVIDIA Jetson AGX Orin while maintaining competitive accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。