This work presents a method for adapting a single, fixed deep neural network
to multiple tasks without affecting performance on already learned tasks. By
building upon ideas from network quantization and pruning, we learn binary
masks that piggyback on an existing network, or are applied to unmodified
weights of that network to provide good performance on a new task. These masks
are learned in an end-to-end differentiable fashion, and incur a low overhead
of 1 bit per network parameter, per task. Even though the underlying network is
fixed, the ability to mask individual weights allows for the learning of a
large number of filters. We show performance comparable to dedicated fine-tuned
networks for a variety of classification tasks, including those with large
domain shifts from the initial task (ImageNet), and a variety of network
architectures. Unlike prior work, we do not suffer from catastrophic forgetting
or competition between tasks, and our performance is agnostic to task ordering.
Code available at https://github.com/arunmallya/piggyback.
Beschreibung
[1801.06519] Piggyback: Adapting a Single Network to Multiple Tasks by Learning to Mask Weights
%0 Journal Article
%1 mallya2018piggyback
%A Mallya, Arun
%A Davis, Dillon
%A Lazebnik, Svetlana
%D 2018
%K compression deep-learning
%T Piggyback: Adapting a Single Network to Multiple Tasks by Learning to
Mask Weights
%U http://arxiv.org/abs/1801.06519
%X This work presents a method for adapting a single, fixed deep neural network
to multiple tasks without affecting performance on already learned tasks. By
building upon ideas from network quantization and pruning, we learn binary
masks that piggyback on an existing network, or are applied to unmodified
weights of that network to provide good performance on a new task. These masks
are learned in an end-to-end differentiable fashion, and incur a low overhead
of 1 bit per network parameter, per task. Even though the underlying network is
fixed, the ability to mask individual weights allows for the learning of a
large number of filters. We show performance comparable to dedicated fine-tuned
networks for a variety of classification tasks, including those with large
domain shifts from the initial task (ImageNet), and a variety of network
architectures. Unlike prior work, we do not suffer from catastrophic forgetting
or competition between tasks, and our performance is agnostic to task ordering.
Code available at https://github.com/arunmallya/piggyback.
@article{mallya2018piggyback,
abstract = {This work presents a method for adapting a single, fixed deep neural network
to multiple tasks without affecting performance on already learned tasks. By
building upon ideas from network quantization and pruning, we learn binary
masks that piggyback on an existing network, or are applied to unmodified
weights of that network to provide good performance on a new task. These masks
are learned in an end-to-end differentiable fashion, and incur a low overhead
of 1 bit per network parameter, per task. Even though the underlying network is
fixed, the ability to mask individual weights allows for the learning of a
large number of filters. We show performance comparable to dedicated fine-tuned
networks for a variety of classification tasks, including those with large
domain shifts from the initial task (ImageNet), and a variety of network
architectures. Unlike prior work, we do not suffer from catastrophic forgetting
or competition between tasks, and our performance is agnostic to task ordering.
Code available at https://github.com/arunmallya/piggyback.},
added-at = {2020-02-06T00:58:31.000+0100},
author = {Mallya, Arun and Davis, Dillon and Lazebnik, Svetlana},
biburl = {https://www.bibsonomy.org/bibtex/23ed5c68506447bd6590d72cc27687002/kirk86},
description = {[1801.06519] Piggyback: Adapting a Single Network to Multiple Tasks by Learning to Mask Weights},
interhash = {5039e735df9352ff133099b80355cfa6},
intrahash = {3ed5c68506447bd6590d72cc27687002},
keywords = {compression deep-learning},
note = {cite arxiv:1801.06519},
timestamp = {2020-02-06T00:58:31.000+0100},
title = {Piggyback: Adapting a Single Network to Multiple Tasks by Learning to
Mask Weights},
url = {http://arxiv.org/abs/1801.06519},
year = 2018
}