Video yükleniyor...
Video Yüklenemedi
🚀 Introducing Next-Concept Modeling (NCM): our new paradigm for fast, parallel, coarse-to-fine text generation. Compared to strong diffusion language models (DLMs) at similar parameter size, NCM achieves: ⚡ Up to 31× faster generation on downstream tasks like summarization and machine translation with comparable or better quality. 🏎️ 10–21× faster... show more
219,267 görüntüleme • 1 gün önce •via X (Twitter)
31 Yorum

Quality at the Fraction of the Cost: NCM pushes the non-autoregressive quality–efficiency frontier. We evaluate NCM across 3 groups of tasks and compare against strong diffusion language model baselines at similar parameter scales: 🏎️ Unconditional generation: TinyStories, LM1B, and OpenWebText. ⚡ Conditional generation: XSum summarization and WMT14 machine translation. 🧩 Constraint-based reasoning: Sudoku across three difficulty levels. We summarize some of our key findings in the figure below.

🧩 How does NCM work? NCM follows a simple two-stage approach: 1️⃣ Learn: A multiscale VQ-VAE learns a hierarchy of discrete text representations, from coarse concepts to fine details. Prior work (VAR) already does this in images but getting it to work well with text requires many substantial modeling changes. 2️⃣ Generate: A transformer predicts the hierarchy level by level, generating all codes within each level in parallel.

✨ Coarse-to-Fine Generation What does NCM's generation trajectory actually look like? To understand how text emerges across the hierarchy, we decode the intermediate latent representations into text at each generation step. NCM progressively adds finer details to the text as it traverses its hierarchy

⚡ Why is NCM so efficient? NCM's inference efficiency comes from two key advantages: 1️⃣ Fewer sampling steps: NCM achieves high generation quality in far fewer steps. 2️⃣ Cheaper steps: Unlike diffusion models (e.g Duo, MDLM etc) that process the full sequence at every step, NCM processes shorter representations at early levels, reducing the average computation per step. Fewer steps + cheaper steps = substantially faster generation! 🚀

📉 Strong Generation Quality with Fewer Training Tokens Across a wide range of tasks & datasets, NCM achieves strong quality while training on 3x-5x fewer tokens compared to its diffusion language baselines. One possible explanation for this is that NCM learns a single pre-set text ordering while many of the diffusion baselines have to learn way more different orderings of the text.

📈 Scaling We explore two directions for scaling NCM: 🧠 Model Scaling: Increasing NCM's model size consistently improves both generation quality and diversity, even at a fixed training-token budget. ⚡ Test-Time Scaling: NCM's fast generation allows us to sample more candidates while still being cheaper or at the same cost of diffusion baselines. We use this to further improving generation quality.

🧩Coarse-to-Fine Reasoning Across different difficulty levels on sudoku, NCM outperforms both AR and diffusion baselines.

🎁 Bonus NCM cranking a sudoku puzzle! It iteratively refines its solutions arriving at the correct solved board at the end.

🚀 NCM is open source! We're excited to share the paper, code, checkpoints, and datasets with the community. Paper: Blog 1: Blog 2: Code: Checkpoints/Datasets: alphaXiv:

Thanks to @modal and @LambdaAPI for compute support towards this project.

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora congrats on the release 🔥🔥🔥

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Looks cool! Just curious, did you get a chance to try it on tinyGSM?

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Thank you!! Not yet, but we will soon!

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Great, let me know what comes out!

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Will do

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora This is really cool research, congrats Hermann! ⚡️🔥

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora this is so weird I love it

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora incredible work! this is the future

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Looks very cool! Congratulations @KumbongHermann !!!!

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Thank you so much David!! I learned a lot of this from you 🙏🙏

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Haha I wish I could take credit! ;) This is truly awesome! Super proud of you!!!

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora 31 倍加速还不掉质量,这 coarse-to-fine 的思路是真漂亮😂

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Incredible release!

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora 先粗后细 31 倍速,这加速我是服气的⚡

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora I have been building something similar for looped AR models. This line of work is definitely the future of the field.

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora really cool!! might take a stab and try NCM on other puzzle tasks.

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Sounds great! Let me know if you need help setting up

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Thanks for the huge effort Hermann! One of the coolest projects to be a part of

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Real full in bio 🔥

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Very cool. Congrata @KumbongHermann

@IddahMlauzi @HazyResearch @Azaliamirh @simran_s_arora Thank you!

