Awesome research! Highly recommend for anyone interested in looping models to read it.
It suggests that the main bottleneck holding back loop models against a fully unlooped model is the memorization.
๐ฅ
๐๐๐ฐ ๐๐ฅ๐จ๐ : ๐๐จ๐ฐ๐๐ซ๐๐ฌ ๐๐จ๐จ๐ฉ๐๐ ๐๐จ๐๐๐ฅ๐ฌ ๐๐จ๐ง๐ ๐๐ข๐ ๐ก๐ญ โ ๐๐๐ซ๐ญ ๐
Looped models reuse the same weights across depth, promising a better computeโparameter trade-off, especially for reasoning.
๐๐ฎ๐ญ ๐) ๐๐จ ๐ญ๐ก๐ ๐ ๐๐ข๐ง๐ฌ ๐ฌ๐ฎ๐ซ๐ฏ๐ข๐ฏ๐ ๐ฐ๐ก๐๐ง ๐๐จ๐ญ๐ก ๐ญ๐ซ๐๐ข๐ง๐ข๐ง๐ ๐๐ง๐ ๐ข๐ง๐๐๐ซ๐๐ง๐๐ ๐ ๐๐๐๐ฌ ๐๐ซ๐ ๐ฆ๐๐ญ๐๐ก๐๐? ๐) ๐๐ง๐ ๐ฐ๐ก๐ข๐๐ก ๐๐ซ๐๐ก๐ข๐ญ๐๐๐ญ๐ฎ๐ซ๐๐ฅ ๐๐ก๐จ๐ข๐๐๐ฌ ๐๐๐ญ๐ฎ๐๐ฅ๐ฅ๐ฒ ๐ฆ๐๐ญ๐ญ๐๐ซ?
We run ๐๐ฉ๐ฉ๐ฅ๐๐ฌ-๐ญ๐จ-๐๐ฉ๐ฉ๐ฅ๐๐ฌ ablations spanning Ouro to Huginn. Huginn performs better overall, with the largest gains coming from the loop-in-the-middle (sandwich) design and input injection, though they provide different benefits.
Trained on ๐๐๐๐ tokens, an ๐๐-๐๐.๐๐ Huginn MoE approaches or surpasses a ๐๐๐-๐๐.๐๐ feedforward MoE on several reasoning benchmarks, including GSM8K (83.6% vs. 80.8%), while using ๐๐% ๐๐๐ฐ๐๐ซ resident parameters under ๐ฆ๐๐ญ๐๐ก๐๐ ๐ญ๐ซ๐๐ข๐ง๐ข๐ง๐ ๐๐ง๐ ๐ข๐ง๐๐๐ซ๐๐ง๐๐ FLOPs.
More details and the blog link in the thread โ
@RidgerZhu Hey nice work! I was reading the paper and I found this part interesting. Does the "Ouro-Thinking" having better score than the original model means we could still add more loops? (Assuming you trained with reasoning traces)
@tensorqt Im also exploring looped transformers and I don't the improvements I was expecting from them (probably skill issue ๐ ), Energy transformers seems a good alternative but their efficiency is still a issue.
@tensorqt Cool post!! Would love to see the impact of position embeddings like ROPE, my intuition is that they reduce the contribution of distance tokens (so initial tokens contribute less) but probably they aren't enough to reduce this bias. ๐ค
๐กLxMLS 2025 is almost here!
From July 19 to 25, @istecnico will once again host the Lisbon Machine Learning Summer School (LxMLS), a flagship event in the ML community. IT is among the event supporters.
๐ https://t.co/D0rWLmCloj
#SummerSchool
@tensorqt I am very bullish on ttc on latent space, would love to fine tune these model https://t.co/P2icyjBRCX on math dataset to know how they compare with reasoning models.
หหห bop and it's live! หห-
In a world full of cookie-cutter design, we're bringing character to products.
And, we just launched our first website!
Link below โ