Retentive Network: A Successor to Transformer for Large Language Models (Paper Explained)
#ai #retnet #transformers
Retention is an alternative to Attention in Transformers that can both be written in a parallel and in a recurrent fashion. This means the architecture achieves training parallelism while maintaining low-cost inference. Experiments in the paper look very promising.
OUTLINE:
0:00 - Intro
2:40 - The impossible triangle
6:55 - Parallel vs sequential
15:35 - Retention mechanism
21:00 - Chunkwise and multi-scale retention
24:10 - Comparison to other architectures
26:30 - Experimental evaluation
Paper: https://arxiv.org/abs/2307.08621
Abstract:
In this work, we propose Retentive Network (RetNet) as a foundation architecture for large language models, simultaneously achieving training parallelism, low-cost inference, and good performance. We theoretically derive the connection between recurrence and attention. Then we propose the retention mechanism for sequence modeling, which supports three computation paradigms, i.e., parallel, recurrent, and chunkwise recurrent. Specifically, the parallel representation allows for training parallelism. The recurrent representation enables low-cost O(1) inference, which improves decoding throughput, latency, and GPU memory without sacrificing performance. The chunkwise recurrent representation facilitates efficient long-sequence modeling with linear complexity, where each chunk is encoded parallelly while recurrently summarizing the chunks. Experimental results on language modeling show that RetNet achieves favorable scaling results, parallel training, low-cost deployment, and efficient inference. The intriguing properties make RetNet a strong successor to Transformer for large language models. Code will be available at this https URL.
Authors: Yutao Sun, Li Dong, Shaohan Huang, Shuming Ma, Yuqing Xia, Jilong Xue, Jianyong Wang, Furu Wei
Links:
Homepage: https://ykilcher.com
Merch: https://ykilcher.com/merch
YouTube: https://www.youtube.com/c/yannickilcher
Twitter: https://twitter.com/ykilcher
Discord: https://ykilcher.com/discord
LinkedIn: https://www.linkedin.com/in/ykilcher
If you want to support me, the best thing to do is to share out the content :)
If you want to support me financially (completely optional and voluntary, but a lot of people have asked for this):
SubscribeStar: https://www.subscribestar.com/yannickilcher
Patreon: https://www.patreon.com/yannickilcher
Bitcoin (BTC): bc1q49lsw3q325tr58ygf8sudx2dqfguclvngvy2cq
Ethereum (ETH): 0x7ad3513E3B8f66799f507Aa7874b1B0eBC7F85e2
Litecoin (LTC): LQW2TRyKYetVC8WjFkhpPhtpbDM4Vw7r9m
Monero (XMR): 4ACL8AGrEo5hAir8A9CeVrW8pEauWvnp1WnSDZxW7tziCDLhZAGsgzhRQABDnFy8yuM9fWJDviJPHKRjV4FWt19CJZN9D4n
-
LIVE
Jim Jordan
2 hours agoOversight of the U.S. Department of Justice
2,247 watching -
2:10:07
The Kirk Minihane Show
4 hours agoKMS Live - June 4, 2024
7.42K3 -
LIVE
The Shannon Joy Show
2 hours ago🔥🔥LIVE EXCLUSIVE - Meet The NY Mom Who SURVIVED The Covid Hospital Kill Protocol Ayesha Kruetz🔥🔥
910 watching -
LIVE
Film Threat
13 hours agoTHE ACOLYTE SAVES STAR WARS! | Hollywood on the Rocks
357 watching -
53:07
Rethinking the Dollar
2 hours agoEconomic Hardship Hits Middle Class: 65% of Americans Struggling to Make Ends Meet
5K2 -
48:25
House Republicans
3 hours agoHouse Republicans Leadership Stakeout
8.31K5 -
1:35:52
The Podcast of the Lotus Eaters
7 hours agoThe Podcast of the Lotus Eaters #929
33.7K23 -
2:58:19
Wendy Bell Radio
7 hours agoThe REAL Virus
58.1K122 -
2:15:59
Matt Kohrs
5 hours agoGameStop Squeeze Saga, Breaking Market News & Live Trading || The MK Show
38.4K1 -
1:28:36
Game On!
13 hours agoJustin Jefferson Gets Paid More Than Crickett! | Sports Morning Espresso Shot
32.8K3