Video yükleniyor...

Video Yüklenemedi

Ana Sayfaya Dön

🚨Coming to #ICML2026 🇰🇷 Last year, we introduced Block Diffusion LMs, adopted by dLLMs at Google, NVIDIA, Meta, etc. 🚀Today, we introduce Set Diffusion, a faster, more flexible alternative that combines AR and diffusion by interpolating generation orders. 🧵1/8

62,810 görüntüleme • 3 ay önce •via X (Twitter)

15 Yorum

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

🧱Block Diffusion runs diffusion in fixed blocks, limiting any-order decoding and delaying KV-cache updates until block completion. 🤝 Set Diffusion runs diffusion in flexible token sets, enabling any-order decoding and KV-cache updates after every step. 2/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

⚖️Set Diffusion interpolates between full diffusion and AR by tuning: (1) set size for speed (2) generation order for quality-flexibility trade-offs. It recovers AR, full diffusion (e.g., MDLM/LLaDA), and Block Diffusion as special cases. 3/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

We induce generation order via position-dependent priority schedules, avoiding direct parameterization over combinatorial orderings. We introduce SW-DLMs: Set Diffusion LMs with sliding-window decoding, an order-causal transformer architecture, and low-variance training. 4/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

⏩Efficient Training. We denoise all sets in 1 forward pass with a set-causal transformer, where each set conditions on itself and previous clean sets. For training, singleton sets reduce this to a causal mask over the permuted sequence, with likelihoods for every context token. 5/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

💪Our Set Diffusion LMs improve key weaknesses of prior diffusion LMs: ✅ Better speed-quality tradeoffs on math, summarization, unconditional generation with stronger infilling than Block Diffusion. 6/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

✅ KV Caching: Set Diffusion caches KVs after each generated set. This is more efficient than Block Diffusion, which waits until block completion. ✅ Arbitrary-Length Generation: Sequential set modeling enables flexible-length generation, unlike full-sequence diffusion that uses fixed bidirectional context. 7/8

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

Catch me at poster #303 Thurs. morning! 🇰🇷📍 🧑‍🎓I’m a final-year Cornell CS PhD on the job market! Please reach out if you’re hiring in AI research. This work was done @cornell_tech with @volokuleshov 📄 🌐 🧑‍💻 🤗

vik profil fotoğrafı
vik3 ay önce

very cool

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

thank you😁

Mark Miller profil fotoğrafı
Mark Miller3 ay önce

Nice

farzad profil fotoğrafı
farzad3 ay önce

available paper?

Shubham Parashar profil fotoğrafı
Shubham Parashar3 ay önce

Very cool work! @mariannearr Wondering if this helps in quality as well? Sometimes smaller blocks are better. Or are the benefits tied mostly to caching ?

Marianne Arriola profil fotoğrafı
Marianne Arriola3 ay önce

Thank you! Yes, similar to block diffusion, set diffusion also has a knob to tighten the likelihood bound for better quality. Smaller blocks in block diffusion make the NELBO tighter, but set diffusion exposes a different knob: the distribution over generation orderings. We increase the left-to-right generation bias by introducing position-dependent ordering schedules, which provably tighten the NELBO while better supporting flexible-position decoding. Empirically, this improves the speed–quality tradeoff relative to block diffusion. (more in Section 4

Zephyr profil fotoğrafı
Zephyr3 ay önce

Does this work with phrased reasoning iteratively

Gabe Astrobot profil fotoğrafı
Gabe Astrobot3 ay önce

KV-cache every step is the real unlock. Block partitioning was a workaround.

Benzer Videolar

DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Prior paper page: present DreamCraft3D, a hierarchical 3D content generation method that produces high-fidelity and coherent 3D objects. We tackle the problem by leveraging a 2D reference image to guide the stages of geometry sculpting and texture boosting. A central focus of this work is to address the consistency issue that existing works encounter. To sculpt geometries that render coherently, we perform score distillation sampling via a view-dependent diffusion model. This 3D prior, alongside several training strategies, prioritizes the geometry consistency but compromises the texture fidelity. We further propose Bootstrapped Score Distillation to specifically boost the texture. We train a personalized diffusion model, Dreambooth, on the augmented renderings of the scene, imbuing it with 3D knowledge of the scene being optimized. The score distillation from this 3D-aware diffusion prior provides view-consistent guidance for the scene. Notably, through an alternating optimization of the diffusion prior and 3D scene representation, we achieve mutually reinforcing improvements: the optimized 3D scene aids in training the scene-specific diffusion model, which offers increasingly view-consistent guidance for 3D optimization. The optimization is thus bootstrapped and leads to substantial texture boosting. With tailored 3D priors throughout the hierarchical generation, DreamCraft3D generates coherent 3D objects with photorealistic renderings, advancing the state-of-the-art in 3D content generation.

AK

161,530 görüntüleme • 3 yıl önce