arXiv reaDer
マルチタスクCNNの確率フィルターグループ:スペシャリストおよびジェネラリストコンボリューションカーネルの学習
Stochastic Filter Groups for Multi-Task CNNs: Learning Specialist and Generalist Convolution Kernels
  畳み込みニューラルネットワーク(CNN)でのマルチタスク学習のパフォーマンスは、アーキテクチャ内のタスク間の機能共有の設計にかかっています。可能な共有パターンの数は、ネットワークの深さとタスク数の組み合わせであるため、純粋にタスク関係の人間の直感に基づいたアーキテクチャを手作業で作成することは、時間がかかり、最適ではない場合があります。この論文では、マルチタスク学習のためのCNNでタスク固有の共有表現を学習するための確率的アプローチを提示します。具体的には、各層の畳み込みカーネルを「スペシャリスト」または「ジェネラリスト」グループに割り当てるメカニズムである「確率的フィルターグループ」(SFG)を提案します。レイヤーとネットワーク内のタスク固有および共有表現の構造間の接続性を決定します。カーネルとネットワークパラメーターの可能なグループ化に対する事後分布を学習するために、変分推論を採用します。ベースラインメソッドよりもパフォーマンスが向上しました。
The performance of multi-task learning in Convolutional Neural Networks (CNNs) hinges on the design of feature sharing between tasks within the architecture. The number of possible sharing patterns are combinatorial in the depth of the network and the number of tasks, and thus hand-crafting an architecture, purely based on the human intuitions of task relationships can be time-consuming and suboptimal. In this paper, we present a probabilistic approach to learning task-specific and shared representations in CNNs for multi-task learning. Specifically, we propose "stochastic filter groups'' (SFG), a mechanism to assign convolution kernels in each layer to "specialist'' or "generalist'' groups, which are specific to or shared across different tasks, respectively. The SFG modules determine the connectivity between layers and the structures of task-specific and shared representations in the network. We employ variational inference to learn the posterior distribution over the possible grouping of kernels and network parameters. Experiments demonstrate that the proposed method generalises across multiple tasks and shows improved performance over baseline methods.
updated: Mon Aug 26 2019 11:09:44 GMT+0000 (UTC)
published: Mon Aug 26 2019 11:09:44 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アソシエイト