Flexible Multi-task Networks by Learning Parameter Allocation
  This paper proposes a novel learning method for multi-task applications.
Multi-task neural networks can learn to transfer knowledge across different
tasks by using parameter sharing. However, sharing parameters between unrelated
tasks can hurt performance. To address this issue, we propose a framework to
learn fine-grained patterns of parameter sharing. Assuming that the network is
composed of several components across layers, our framework uses learned binary
variables to allocate components to tasks in order to encourage more parameter
sharing between related tasks, and discourage parameter sharing otherwise. The
binary allocation variables are learned jointly with the model parameters by
standard back-propagation thanks to the Gumbel-Softmax reparametrization
method. When applied to the Omniglot benchmark, the proposed method achieves a
17% relative reduction of the error rate compared to state-of-the-art.
