arXiv reaDer
AutoTaskFormer: マルチタスク学習のためのビジョン トランスフォーマーの検索
AutoTaskFormer: Searching Vision Transformers for Multi-task Learning
ビジョン トランスフォーマーは、分類やセグメンテーションなどの単一のタスクで優れたパフォーマンスを示しています。ただし、現実世界の問題は分離されていないため、複数のタスクを同時に実行できるビジョン トランスフォーマーが必要です。既存のマルチタスク ビジョン トランスフォーマーは手作りであり、人間の専門知識に大きく依存しています。この作業では、このプロセスを自動化するために、AutoTaskFormer (Automated Multi-Task Vision Transformer) と呼ばれる新しいワンショット ニューラル アーキテクチャ検索フレームワークを提案します。 AutoTaskFormer は、複数のタスク間で共有する重みを自動的に識別するだけでなく、さまざまなリソースの制約下で展開するための幅広いパラメーター (ヘッドの数やネットワークの深さなど) を使用して、十分にトレーニングされた何千ものビジョン トランスフォーマーを提供します。小規模 (2 タスクの Cityscapes と 3 タスクの NYUv2) および大規模 (16 タスクの Taskonomy) データセットの両方での実験では、AutoTaskFormer が最先端の手作りのビジョン トランスフォーマーよりもマルチタスク学習で優れていることが示されています。コードとモデル全体がオープンソースになります。
Vision Transformers have shown great performance in single tasks such as classification and segmentation. However, real-world problems are not isolated, which calls for vision transformers that can perform multiple tasks concurrently. Existing multi-task vision transformers are handcrafted and heavily rely on human expertise. In this work, we propose a novel one-shot neural architecture search framework, dubbed AutoTaskFormer (Automated Multi-Task Vision TransFormer), to automate this process. AutoTaskFormer not only identifies the weights to share across multiple tasks automatically, but also provides thousands of well-trained vision transformers with a wide range of parameters (e.g., number of heads and network depth) for deployment under various resource constraints. Experiments on both small-scale (2-task Cityscapes and 3-task NYUv2) and large-scale (16-task Taskonomy) datasets show that AutoTaskFormer outperforms state-of-the-art handcrafted vision transformers in multi-task learning. The entire code and models will be open-sourced.
updated: Tue Apr 18 2023 06:30:20 GMT+0000 (UTC)
published: Tue Apr 18 2023 06:30:20 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト