3 个月前

超越卡尔曼滤波器:基于深度学习的滤波器在提升目标跟踪性能中的应用

超越卡尔曼滤波器:基于深度学习的滤波器在提升目标跟踪性能中的应用

摘要

传统的基于检测的跟踪系统通常采用卡尔曼滤波器(Kalman Filter, KF)进行状态估计。然而,卡尔曼滤波器依赖于特定领域的设计选择,且难以有效处理非线性运动模式。为克服这些局限性,本文提出两种创新的数据驱动滤波方法。第一种方法采用带有可训练运动模型的贝叶斯滤波器,用于预测目标未来的空间位置,并将预测结果与目标检测器提供的观测信息相结合,从而提升边界框预测的准确性。此外,该方法显著减少了传统卡尔曼滤波器所依赖的大量领域特定设计决策。第二种方法则进一步发展为端到端可训练的滤波器,能够学习对检测器误差进行修正,从而进一步降低对领域知识的依赖。同时,本文还提出了一系列基于循环神经网络(Recurrent Neural Networks)、神经微分方程(Neural Ordinary Differential Equations)以及条件神经过程(Conditional Neural Processes)的运动模型架构,并将其与上述滤波方法相结合。在多个数据集上的广泛实验表明,所提出的滤波器在目标跟踪任务中显著优于传统的卡尔曼滤波器,尤其是在非线性运动模式下——这正是我们所设计滤波器最擅长的应用场景。此外,我们还对滤波器的噪声鲁棒性进行了分析,结果表现出令人信服的优越性能。为进一步提升跟踪效果,本文还提出了一种新的观测与轨迹关联成本函数。将该关联成本与所提出的滤波方法相结合后,所构建的跟踪器在Motion-Rich的DanceTrack和SportsMOT数据集上,基于多项评估指标,均优于传统的SORT方法及其他基于运动模型的跟踪算法。

代码仓库

Robotmurlock/NODETracker
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
multi-object-tracking-on-dancetrackMoveSORT
AssA: 38.7
DetA: 81.6
HOTA: 56.1
IDF1: 56.0
MOTA: 91.8
multi-object-tracking-on-sportsmotMoveSORT
AssA: 63.7
DetA: 87.5
HOTA: 74.6
IDF1: 76.9
MOTA: 96.7
multiple-object-tracking-on-sportsmotMoveSORT
AssA: 63.7
DetA: 87.5
HOTA: 74.6
IDF1: 76.9
MOTA: 96.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
超越卡尔曼滤波器:基于深度学习的滤波器在提升目标跟踪性能中的应用 | 论文 | HyperAI超神经