ツリー構造のマルチタスクアーキテクチャは、マルチタスク学習(MTL)のコンテキストで複数のビジョンタスクに共同で取り組むために採用されています。主な課題は、タスクの精度と計算効率の両方を最適化するために、バックボーンモデルを指定して各タスクの分岐先を決定することです。この課題に対処するために、このペーパーでは、一連のタスクと畳み込みニューラルネットワークベースのバックボーンモデルを前提として、ユーザー指定の計算を満たしながら高いタスクパフォーマンスを実現できるツリー構造のマルチタスクアーキテクチャを自動的に提案する推奨事項を提案します。モデルトレーニングを実行せずに予算を立てます。人気のあるMTLベンチマークの広範な評価は、推奨されるアーキテクチャが、最先端のMTLメソッドと比較して競争力のあるタスク精度と計算効率を達成できることを示しています。ツリー構造のマルチタスクモデルレコメンダーはオープンソースであり、https://github.com/zhanglijun95/TreeMTLで入手できます。
Tree-structured multi-task architectures have been employed to jointly tackle multiple vision tasks in the context of multi-task learning (MTL). The major challenge is to determine where to branch out for each task given a backbone model to optimize for both task accuracy and computation efficiency. To address the challenge, this paper proposes a recommender that, given a set of tasks and a convolutional neural network-based backbone model, automatically suggests tree-structured multi-task architectures that could achieve a high task performance while meeting a user-specified computation budget without performing model training. Extensive evaluations on popular MTL benchmarks show that the recommended architectures could achieve competitive task accuracy and computation efficiency compared with state-of-the-art MTL methods. Our tree-structured multi-task model recommender is open-sourced and available at https://github.com/zhanglijun95/TreeMTL.