arXiv reaDer
DynaShare: マルチタスク学習のためのタスクとインスタンスの条件付きパラメータ共有
DynaShare: Task and Instance Conditioned Parameter Sharing for Multi-Task Learning
マルチタスク ネットワークは、効果的なパラメータ共有に依存して、タスク全体にわたる堅牢な一般化を実現します。この論文では、推論時にタスクと中間特徴表現の両方でパラメータ共有を条件付ける、マルチタスク学習のための新しいパラメータ共有方法を紹介します。トレーニング中に決定論的な共有パターンを修正または学習し、推論中にすべての例に同じパターンを適用する従来のパラメータ共有アプローチとは対照的に、タスクと入力インスタンスの両方に基づいてネットワークのどの部分をアクティブにするかを動的に決定することを提案します。 。私たちのアプローチは、粗いレイヤー選択のためのタスク固有のポリシーと個々の入力インスタンスのゲート ユニットで構成される階層型ゲート ポリシーを学習します。これらは連携して推論時に実行パスを決定します。 NYU v2、Cityscapes、および MIMIC-III データセットでの実験は、提案されたアプローチの可能性と問題領域全体への適用可能性を示しています。
Multi-task networks rely on effective parameter sharing to achieve robust generalization across tasks. In this paper, we present a novel parameter sharing method for multi-task learning that conditions parameter sharing on both the task and the intermediate feature representations at inference time. In contrast to traditional parameter sharing approaches, which fix or learn a deterministic sharing pattern during training and apply the same pattern to all examples during inference, we propose to dynamically decide which parts of the network to activate based on both the task and the input instance. Our approach learns a hierarchical gating policy consisting of a task-specific policy for coarse layer selection and gating units for individual input instances, which work together to determine the execution path at inference time. Experiments on the NYU v2, Cityscapes and MIMIC-III datasets demonstrate the potential of the proposed approach and its applicability across problem domains.
updated: Fri May 26 2023 23:43:21 GMT+0000 (UTC)
published: Fri May 26 2023 23:43:21 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アソシエイト