about
One Model to Learn Them All (arxiv.org)
3 points by jonbaer on Jun 20, 2017 | hide | past | pdf | discuss on HN

In plain words: A single network mixes picture-scanning layers, attention, and layers that use only a few parts, so one model covers images, translation, captions, speech, and parsing. Trained together, it does well on every task, helping small-data tasks most while barely hurting big ones.

Abstract · One Model To Learn Them All

Deep learning yields great results across many fields, from speech recognition, image classification, to translation. But for each problem, getting a deep model to work well involves research into the architecture and a long period of tuning. We present a single model that yields good results on a number of problems spanning multiple domains. In particular, this single model is trained concurrently on ImageNet, multiple translation tasks, image captioning (COCO dataset), a speech recognition corpus, and an English parsing task. Our model architecture incorporates building blocks from multiple domains. It contains convolutional layers, an attention mechanism, and sparsely-gated layers. Each of these computational blocks is crucial for a subset of the tasks we train on. Interestingly, even if a block is not crucial for a task, we observe that adding it never hurts performance and in most cases improves it on all tasks. We also show that tasks with less data benefit largely from joint training with other tasks, while performance on large tasks degrades only slightly if at all.

Lukasz Kaiser, Aidan N. Gomez, Noam Shazeer, Ashish Vaswani, Niki Parmar, Llion Jones, Jakob Uszkoreit
arXiv:1706.05137 · cs.LG, stat.ML · submitted Jun 16, 2017
abstract · pdf · html

add comment on HN
Also discussed: Oct 2024 (5 points, 0 comments) · Jun 2018 (1 point, 0 comments) · Jul 2017 (4 points, 0 comments) · Jun 2017 (1 point, 0 comments) · Jun 2017 (2 points, 0 comments) · Jun 2017 (2 points, 0 comments) · Jun 2017 (6 points, 0 comments) · Jun 2017 (3 points, 0 comments) · Jun 2017 (3 points, 0 comments)