References & Citations
Computer Science > Machine Learning
Title: Alternating Updates for Efficient Transformers
(Submitted on 30 Jan 2023)
Abstract: It is well established that increasing scale in deep transformer networks leads to improved quality and performance. This increase in scale often comes with an increase in compute cost and inference latency. Consequently, research into methods which help realize the benefits of increased scale without leading to an increase in the compute cost becomes important. We introduce Alternating Updates (AltUp), a simple-to-implement method to increase a model's capacity without the computational burden. AltUp enables the widening of the learned representation without increasing the computation time by working on a subblock of the representation at each layer. Our experiments on various transformer models and language tasks demonstrate the consistent effectiveness of alternating updates on a diverse set of benchmarks. Finally, we present extensions of AltUp to the sequence dimension, and demonstrate how AltUp can be synergistically combined with existing approaches, such as Sparse Mixture-of-Experts models, to obtain efficient models with even higher capacity.
Submission history
From: Nishanth Dikkala [view email][v1] Mon, 30 Jan 2023 22:06:05 GMT (2284kb,D)
Link back to: arXiv, form interface, contact.