MEGABYTE: Predicting Million-byte Sequences with Multiscale Transformers

AI-generated keywords: Megabyte Autoregressive Transformers Multi-Scale Decoder Long Sequences Performance

AI-generated Key Points

  • Introduces Megabyte, a multi-scale decoder architecture for modeling sequences over one million bytes
  • Addresses limitations of autoregressive transformers in scaling to long sequences such as high-resolution images, podcasts, code, or books
  • Segments sequences into patches and utilizes local submodel within patches and global model between patches
  • Allows for sub-quadratic self-attention, larger feedforward layers, and improved parallelism during decoding
  • Better performance at reduced cost for both training and generation
  • Byte-level models using Megabyte can compete with subword models in long context language modeling
  • Achieves state-of-the-art density estimation on ImageNet and successfully models audio from raw files
  • Establishes viability of tokenization-free autoregressive sequence modeling at scale
  • Provides additional context regarding dataset sizes and mean document lengths across different modalities such as PG-19, Stories, Books, arXiv, and Code
  • Implementation shares similarities with other models to ensure fair comparisons in compute-controlled experiments
  • Training procedure involves using the Metaseq2 codebase with PyTorch framework and fairscale for memory efficiency through fully sharded model and optimizer states
  • Mixed precision training is employed to improve training efficiency while preserving accuracy at scale
  • Reproduced experiments on downsized ImageNet at 64 pixels to validate implementation of PerceiverAR; achieved a bits per byte (bpb) score of 3.53 compared to the reported 3.54 in the original study
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Lili Yu, Dániel Simig, Colin Flaherty, Armen Aghajanyan, Luke Zettlemoyer, Mike Lewis

License: CC BY 4.0

Abstract: Autoregressive transformers are spectacular models for short sequences but scale poorly to long sequences such as high-resolution images, podcasts, code, or books. We proposed Megabyte, a multi-scale decoder architecture that enables end-to-end differentiable modeling of sequences of over one million bytes. Megabyte segments sequences into patches and uses a local submodel within patches and a global model between patches. This enables sub-quadratic self-attention, much larger feedforward layers for the same compute, and improved parallelism during decoding -- unlocking better performance at reduced cost for both training and generation. Extensive experiments show that Megabyte allows byte-level models to perform competitively with subword models on long context language modeling, achieve state-of-the-art density estimation on ImageNet, and model audio from raw files. Together, these results establish the viability of tokenization-free autoregressive sequence modeling at scale.

Submitted to arXiv on 12 May. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2305.07185v2

The paper introduces Megabyte, a multi-scale decoder architecture that enables the modeling of sequences over one million bytes. It addresses the limitations of autoregressive transformers in scaling to long sequences such as high-resolution images, podcasts, code, or books. Megabyte segments sequences into patches and utilizes a local submodel within patches and a global model between patches. This approach allows for sub-quadratic self-attention, larger feedforward layers, and improved parallelism during decoding. The result is better performance at reduced cost for both training and generation. Extensive experiments demonstrate the effectiveness of Megabyte in various tasks. It shows that byte-level models using Megabyte can compete with subword models in long context language modeling. Additionally, it achieves state-of-the-art density estimation on ImageNet and successfully models audio from raw files. These results establish the viability of tokenization-free autoregressive sequence modeling at scale. The paper also provides additional context regarding dataset sizes and mean document lengths across different modalities such as PG-19, Stories, Books, arXiv, and Code. The implementation of Megabyte shares similarities with other models to ensure fair comparisons in compute-controlled experiments. The training procedure involves using the Metaseq2 codebase with PyTorch framework and fairscale for memory efficiency through fully sharded model and optimizer states. Mixed precision training is employed to improve training efficiency while preserving accuracy at scale. In order to validate their implementation of PerceiverAR, the authors reproduce experiments on downsized ImageNet at 64 pixels by matching hyperparameters carefully; they achieve a bits per byte (bpb) score of 3.53 compared to the reported 3.54 in the original study.
Created on 28 Jul. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.