SimonVandenhende / Multi-Task-Learning-PyTorch

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
Other
752 stars 114 forks source link

[Question] About dataset condition for MTL #21

Open qjadud1994 opened 2 years ago

qjadud1994 commented 2 years ago

Thank you for your awesome work. Your work might be greatly helpful to all people who interest in the MTL.

I'm about to study MTL and have one question.

I think the dataset for MTL should be in form of {Input: X(i), GT: Y_task1(i), Y_task2(i), ..., Y_taskT(i)}.

However, I think that it is difficult to satisfy this condition in a real-world environment. When we should train task-specific datasets D_task1 {Input: X_task1, GT: Y_task1}, D_task2 {Input: X_task2, GT: Y_task2} simultaneously, how we do MTL?

For example, we aim to set MTL for both salient object detection and depth estimation. For the salient object detection task, we use saliency labels from Pascal VOC dataset. For the depth estimation task, we use depth-map labels from NYUD dataset. (Both datasets totally consist of different input images, and Pascal VOC does not contain depth-map labels and NYUD does not contain saliency labels)

In this condition, how we construct MTL? Does anyone know about MTL for task-specific datasets or related works?

wyfeng1020 commented 2 years ago

you can read the Ubernet: https://arxiv.org/pdf/1609.02132.pdf

BQT11 commented 2 years ago

I also want to ask how you can get the label of salient task in Pascal Dataset. The Pascal-S Dataset only has 850 images, but your dataset about sal has 10500 images. Thank you.

yuan243212790 commented 1 year ago

你可以阅读 Ubernet:https ://arxiv.org/pdf/1609.02132.pdf

Ubernet is not open source, and its training strategy is that there is no assignment of gt to 0 for loss. In fact, different data sets are trained separately and sequentially, right?