Generative Modeling for Multi-task Visual Learning

July 2022

Generative Modeling for Multi-task Visual Learning

Authors:

Zhipeng Bao, Martial Hebert, and Yu-Xiong Wang
Conference Paper
Proceedings of (ICML) International Conference on Machine Learning

Abstract:

Generative modeling has recently shown great promise in computer vision, but it has mostly focused on synthesizing visually realistic images. In this paper, motivated by multi-task learning of shareable feature representations, we consider a novel problem of learning a shared generative model that is useful across various visual perception tasks. Correspondingly, we propose a general multi-task oriented generative modeling (MGM) framework, by coupling a discriminative multi-task network with a generative network. While it is challenging to synthesize both RGB images and pixel-level annotations in multi-task scenarios, our framework enables us to use synthesized images paired with only weak annotations (i.e., image-level scene labels) to facilitate multiple visual tasks. Experimental evaluation on challenging multi-task benchmarks, including NYUv2 and Taskonomy, demonstrates that our MGM framework improves the performance of all the tasks by large margins, consistently outperforming state-of-the-art multi-task approaches in different sample-size regimes.
@conference{Bao-2022-132251,
author = {Zhipeng Bao And Martial Hebert And Yu-Xiong Wang},
title = {Generative Modeling for Multi-task Visual Learning},
booktitle = {Proceedings of (ICML) International Conference on Machine Learning},
year = {2022},
month = {July},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.