arXiv reaDer
マルチタスク学習のための動的ニューラル ネットワーク 多様なネットワーク トポロジを横断する検索
Dynamic Neural Network for Multi-Task Learning Searching across Diverse Network Topologies
この論文では、多様なグラフ トポロジを持つ複数のタスクに最適化された構造を検索し、タスク間で機能を共有する新しい MTL フレームワークを提示します。検索スペースと時間を制限しながら、トポロジ的に多様なタスク適応構造を構築するために、読み取り/読み取りレイヤーを備えた制限付きの DAG ベースの中央ネットワークを設計します。 3段階のトレーニングプロセスを使用して、複数のタスク適応サブネットワークとして機能する単一の最適化されたネットワークを検索します。ネットワークをコンパクトで離散化するために、フローベースのリダクション アルゴリズムと、トレーニング プロセスで使用されるスクイーズ ロスを提案します。さまざまなパブリック MTL データセットで最適化されたネットワークを評価し、最先端のパフォーマンスを達成することを示します。大規模なアブレーション研究では、フレームワークのサブモジュールとスキームの有効性が実験的に検証されています。
In this paper, we present a new MTL framework that searches for structures optimized for multiple tasks with diverse graph topologies and shares features among tasks. We design a restricted DAG-based central network with read-in/read-out layers to build topologically diverse task-adaptive structures while limiting search space and time. We search for a single optimized network that serves as multiple task adaptive sub-networks using our three-stage training process. To make the network compact and discretized, we propose a flow-based reduction algorithm and a squeeze loss used in the training process. We evaluate our optimized network on various public MTL datasets and show ours achieves state-of-the-art performance. An extensive ablation study experimentally validates the effectiveness of the sub-module and schemes in our framework.
updated: Mon Mar 13 2023 05:01:50 GMT+0000 (UTC)
published: Mon Mar 13 2023 05:01:50 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アソシエイト