HackerPapers

arXiv:1706.05137cs.LG

One Model To Learn Them All

Lukasz Kaiser, Aidan N. Gomez, Noam Shazeer, Ashish Vaswani, Niki Parmar, Llion Jones, Jakob Uszkoreit

Abstract

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.

arXiv comments
Journal reference
DOI
Subjects
cs.LGMachine Learning stat.MLMachine Learning (Statistics)

On Hacker News

Linked by 5 identifiable commenters in 5 comments across 5 threads, from Jun 2017 to Mar 2019. Found in 10 HN submissions from Jun 2017 to Oct 2024. Open submission discussions.

Comments linking this paper, per year.

Discussed in comments5 threads

    • The success of Transformers aside, I'm not sure you should be relying on model titles for anything, lest we forget papers like "One Model To Learn Them All" [1]. [1] https://arxiv.org/abs/1706.05137

      phowonView on HNParent context

    • The author is Douglas Hofstadter, author of Gödel, Escher, Bach.[a] In this article, he shows with concrete examples how Google Translate falls short, and then offers two criticisms: * Feeding more data to current models won't bring them any closer to understanding, since…

      cs702View on HNParent context

    • A nice complement to this is Andrej Karpathy's Software 2.0 essay. Namely that it is getting much easier to collect the data. Than it ever will be to write the complex programs which can anticipate every case. Software 2.0 https://medium.com/@karpathy/software-2-0-a64152b37c35…

      indescions_2017View on HNParent context

  1. The future of deep learning 1 comment
    • > we need better ways of composing this `deep` abstractions Tensor2Tensor[0], from the Google Brain team, has some strong recent results in that direction. Related paper: "One Model To Learn Them All"[1]. [0]: https://github.com/tensorflow/tensor2tensor [1]…

      espadrineView on HNParent context

    • 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…

      bdcsView on HNParent context

Submitted to Hacker News

Connected through shared HN threads, not inferred similarity or a recommendation.