Context-Aware Multi-Task Learning for Traffic Scene Recognition in Autonomous Vehicles

Younkwan Lee1, Jihyo Jeon1, Jongmin Yu, Moongu Jeon1

  • 1Gwangju Institute of Science and Technology (GIST)


14:15 - 14:30 | Wed 21 Oct | Virtual Room 1 | W1WePM1.2

Session: Environment Perception 1

Category: Regular
Theme: Environment Perception


Traffic scene recognition, which requires various visual classification tasks, is a critical ingredient in autonomous vehicles. However, most existing approaches treat each relevant task independently from one another, never considering the entire task as a wholeecause of this, they are limited to utilizing a task-specific set of features for all possible tasks of inference-time, which ignores the ability to leverage common task-invariant contextual knowledge for the task at hand. To address this problem, we propose an algorithm to jointly learn the task-specific and shared representations by adopting a multi-task learning network. Specifically, we present a lower bound of mutual information constraint between shared feature embedding and input that is considered to be able to extract common contextual information across tasks while preserving essential information of each task jointly. The learned representations capture richer contextual information without additional task-specific network. Extensive experiments on the large-scale dataset HSD demonstrate the effectiveness and superiority of our network over state-of-the-art methods.