arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1805.06334cs.CVcs.LG
收藏

多任务学习中的辅助任务

Auxiliary Tasks in Multi-task Learning

Technical University of Munich(慕尼黑工业大学)

Lukas Liebel, Marco Körner

摘要:

多任务卷积神经网络(CNN)在某些任务组合方面表现出了令人印象深刻的结果,例如单图像深度估计(SIDE)和语义分割。这是通过推动网络学习一种鲁棒的表示来实现的,这种表示可以很好地推广到不同的原子任务。我们通过添加辅助任务来扩展这个概念,这些辅助任务与应用程序的相关性较小,可以学习到一组任务。作为一种额外的正则化,它们有望提高最终期望的主要任务的性能。为了研究所提出的方法,我们选择了基于视觉的道路场景理解(RSU)作为一个示例性的应用。由于多任务学习需要专门的数据集,特别是在使用大量任务集时,我们为多任务RSU提供了一个多模态数据集,称为synMT。从视频游戏《侠盗猎车手V》(GTA V)中获取了超过2.5个$\cdot$ 10^5合成图像,用21个不同的标签进行注释。我们提出的深度多任务CNN架构使用synMT在各种任务组合上进行训练。实验证实,辅助任务确实可以提高网络性能,无论是在最终结果还是训练时间方面。

英文摘要:

Multi-task convolutional neural networks (CNNs) have shown impressive results for certain combinations of tasks, such as single-image depth estimation (SIDE) and semantic segmentation. This is achieved by pushing the network towards learning a robust representation that generalizes well to different atomic tasks. We extend this concept by adding auxiliary tasks, which are of minor relevance for the application, to the set of learned tasks. As a kind of additional regularization, they are expected to boost the performance of the ultimately desired main tasks. To study the proposed approach, we picked vision-based road scene understanding (RSU) as an exemplary application. Since multi-task learning requires specialized datasets, particularly when using extensive sets of tasks, we provide a multi-modal dataset for multi-task RSU, called synMT. More than 2.5 $\cdot$ 10^5 synthetic images, annotated with 21 different labels, were acquired from the video game Grand Theft Auto V (GTA V). Our proposed deep multi-task CNN architecture was trained on various combination of tasks using synMT. The experiments confirmed that auxiliary tasks can indeed boost network performance, both in terms of final results and training time.

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

↑