用多模态知识蒸馏让小模型也能精准预测行人轨迹。
Multi-modal Knowledge Distillation-based Human Trajectory Forecasting
- 大模型融合轨迹、姿态和文本,小模型仅靠轨迹或姿态就能学得相似能力。
- 在三个数据集上,小模型性能提升最高达13%,且对实时观测也有效。
- 适合资源受限的自动驾驶和机器人系统使用,无需依赖复杂文本生成。
行人轨迹预测在自动驾驶和移动机器人导航中至关重要。基于摄像头的感知可提取人体姿态、文本等额外模态以提升预测精度。我们发现,文本描述在融合多模态信息中起关键作用。然而,在线提取文本需依赖视觉语言模型(VLM),这对资源受限系统不现实。为此,我们提出一种多模态知识蒸馏框架:将训练于完整模态(轨迹、人体姿态、文本)的教师模型知识,蒸馏到仅使用轨迹或人体姿态作为补充的轻量学生模型。我们分别从代理内多模态和代理间交互中蒸馏核心运动规律。该通用框架在两个前沿模型上,于三个数据集(JRDB、SIT、ETH/UCY)的自车视角(ego-view)与俯视图(BEV-view)设置下验证,使用标注和VLM生成的文本标题。蒸馏后的学生模型在所有预测指标上均取得一致提升,全时和瞬时观测下最高提升约13%。代码已公开于https://github.com/Jaewoo97/KDTF。
原文摘要 · Abstract (English)
Pedestrian trajectory forecasting is crucial in various applications such as autonomous driving and mobile robot navigation. In such applications, camera-based perception enables the extraction of additional modalities (human pose, text) to enhance prediction accuracy. Indeed, we find that textual descriptions play a crucial role in integrating additional modalities into a unified understanding. However, online extraction of text requires the use of VLM, which may not be feasible for resource-constrained systems. To address this challenge, we propose a multi-modal knowledge distillation framework: a student model with limited modality is distilled from a teacher model trained with full range of modalities. The comprehensive knowledge of a teacher model trained with trajectory, human pose, and text is distilled into a student model using only trajectory or human pose as a sole supplement. In doing so, we separately distill the core locomotion insights from intra-agent multi-modality and inter-agent interaction. Our generalizable framework is validated with two state-of-the-art models across three datasets on both ego-view (JRDB, SIT) and BEV-view (ETH/UCY) setups, utilizing both annotated and VLM-generated text captions. Distilled student models show consistent improvement in all prediction metrics for both full and instantaneous observations, improving up to ~13%. The code is available at https://github.com/Jaewoo97/KDTF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。