aifollow.news
Back Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · Original publication time AI score54

Meta and MIT study looping Transformer layers to increase effective depth with few extra parametersMachine translation

AI introduction

A Meta and MIT paper studies running Transformer layers repeatedly to increase model depth without a proportional rise in parameters. Tests on dense models with up to 1 billion parameters trained on up to 155 billion tokens found that looping the full stack gave the best quality at a fixed size, while looping middle layers worked best when training compute was tight.

Original post · Original

The article text is unavailable in this language; an existing version is shown.

@rohanpaul_ai

– https://t.co/ALfO6njve4

Title: "How to Loop Your Transformer: Loop Scaling, Specialization, and Parallel Inference"

ReplyRohan Paul@rohanpaul_ai
New Meta and MIT paper finds reusing a transformer's layers in loops can beat a plain model. Looping runs the same layers several times, giving a model more depth without more parameters. But each loop costs compute and normally runs in sequence for every token. They trained dense models up to 1B parameters on up to 155B tokens, testing where to loop and how many times. They also added tiny per-loop adapters and repaired a parallel looping method so loops can run at once during decoding. Looping the whole stack gave the best quality at a fixed size, while looping middle layers paid off best when training compute was tight. Per-loop adapters added only 1.1 to 4.3% more parameters, versus 51.4% for doubling depth, and helped more as loops increased.
View this post on X ↗

Source: Rohan Paul · x.com

Research
Found an error? Send a correction