arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2309.04422cs.CV
收藏

Video Task Decathlon:在自动驾驶中统一图像与视频任务

Video Task Decathlon: Unifying Image and Video Tasks in Autonomous Driving

ETH Zürich(苏黎世联邦理工学院);KU Leuven(鲁汶大学)

Thomas E. Huang, Yifan Liu, Luc Van Gool, Fisher Yu

AI总结:

为解决自动驾驶中多视觉任务统一问题,提出VTD挑战及统一网络VTDNet,通过CPF训练方案在仅需20%计算量下显著优于单任务模型。

摘要:

在动态场景中执行多种异构视觉任务是人类感知能力的标志。尽管通过表征学习在图像和视频识别方面取得了显著进展,但当前研究仍侧重于为单一、同质或简单的任务组合设计专用网络。我们转而探索构建一个统一模型,以处理自动驾驶中具有多样输入和输出结构的主要图像与视频识别任务。为了实现此类研究,我们设计了一项新挑战Video Task Decathlon(VTD),它包含十项代表性的图像与视频任务,涵盖对象与像素的分类、分割、定位和关联。在VTD上,我们开发了统一网络VTDNet,它对所有十项任务使用单一结构和单一权重集。VTDNet将相似任务进行分组,并采用任务交互阶段在任务组内和组间交换信息。鉴于对所有帧上的所有任务进行标注的不切实际性,以及多任务联合训练带来的性能下降,我们设计了一种课程训练、伪标签和微调(CPF)方案,以成功在所有任务上训练VTDNet并减轻性能损失。借助CPF,VTDNet在大多数任务上显著优于其单任务对应模型,且仅需20%的整体计算量。VTD是探索自动驾驶感知任务统一的一个有前景的新方向。

英文摘要:

Performing multiple heterogeneous visual tasks in dynamic scenes is a hallmark of human perception capability. Despite remarkable progress in image and video recognition via representation learning, current research still focuses on designing specialized networks for singular, homogeneous, or simple combination of tasks. We instead explore the construction of a unified model for major image and video recognition tasks in autonomous driving with diverse input and output structures. To enable such an investigation, we design a new challenge, Video Task Decathlon (VTD), which includes ten representative image and video tasks spanning classification, segmentation, localization, and association of objects and pixels. On VTD, we develop our unified network, VTDNet, that uses a single structure and a single set of weights for all ten tasks. VTDNet groups similar tasks and employs task interaction stages to exchange information within and between task groups. Given the impracticality of labeling all tasks on all frames, and the performance degradation associated with joint training of many tasks, we design a Curriculum training, Pseudo-labeling, and Fine-tuning (CPF) scheme to successfully train VTDNet on all tasks and mitigate performance loss. Armed with CPF, VTDNet significantly outperforms its single-task counterparts on most tasks with only 20% overall computations. VTD is a promising new direction for exploring the unification of perception tasks in autonomous driving.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑