Simplifying Transformer Blocks
AI-generated Key Points
- The standard transformer block used in deep Transformers is complex
- Complexity can lead to reduced training speed or untrainable models with minor changes
- Authors propose modifications to simplify the standard transformer block
- Modifications include removing skip connections, projection or value parameters, sequential sub-blocks, and normalization layers
- Simplified transformers achieve comparable training speed and performance to standard transformers
- Simplified transformers have a 15% faster training throughput and use 15% fewer parameters
- Simplified blocks perform well on different datasets and architectures, including bidirectional encoder-only BERT models
- Low-resource setting with 24-hour training on a single consumer GPU is used for experiments
- Results show that simplified blocks match pre-training speed of baseline within given runtime
- Proposed modifications improve training speed without sacrificing performance and reduce parameter count
Authors: Bobby He, Thomas Hofmann
Abstract: A simple design recipe for deep Transformers is to compose identical building blocks. But standard transformer blocks are far from simple, interweaving attention and MLP sub-blocks with skip connections & normalisation layers in precise arrangements. This complexity leads to brittle architectures, where seemingly minor changes can significantly reduce training speed, or render models untrainable. In this work, we ask to what extent the standard transformer block can be simplified? Combining signal propagation theory and empirical observations, we motivate modifications that allow many block components to be removed with no loss of training speed, including skip connections, projection or value parameters, sequential sub-blocks and normalisation layers. In experiments on both autoregressive decoder-only and BERT encoder-only models, our simplified transformers emulate the per-update training speed and performance of standard transformers, while enjoying 15% faster training throughput, and using 15% fewer parameters.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
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 representationLook 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.