用稀疏注意力与专家混合模型加速长文本生成。
MoE-DiffuSeq: Enhancing Long-Document Diffusion Models with Sparse Attention and Mixture of Experts
- 结合稀疏注意力与专家路由,降低计算开销。
- 引入软吸收态加速生成,保持文本连贯性与质量。
- 适合科学写作、代码生成等长上下文场景。
我们提出 MoE-DiffuSeq,一种基于扩散模型的高效长文本生成框架,融合稀疏注意力与专家混合(MoE)架构。现有序列扩散模型在生成长文档时面临巨大的计算和内存开销,主要源于密集注意力机制和缓慢的迭代重构过程。MoE-DiffuSeq 通过专家路由与定制化的稀疏注意力机制相结合,显著降低注意力复杂度,同时保持全局连贯性和文本保真度。此外,我们在扩散过程中引入了软吸收态,重塑去噪阶段的注意力动态,实现更快的序列重构与更精确的词元优化。该设计在不牺牲生成质量的前提下,提升了训练与采样速度。在多个长文档基准上的实验表明,MoE-DiffuSeq 在训练效率、推理速度和生成质量上均优于先前的扩散模型及稀疏注意力基线。本方法特别适用于科学文档生成、大规模代码合成和长对话建模等长上下文任务,为扩散模型的长文本生成提供了可扩展且表达力强的解决方案。
原文摘要 · Abstract (English)
We propose \textbf{MoE-DiffuSeq}, a diffusion-based framework for efficient long-form text generation that integrates sparse attention with a Mixture-of-Experts (MoE) architecture. Existing sequence diffusion models suffer from prohibitive computational and memory costs when scaling to long documents, largely due to dense attention and slow iterative reconstruction. MoE-DiffuSeq addresses these limitations by combining expert routing with a tailored sparse attention mechanism, substantially reducing attention complexity while preserving global coherence and textual fidelity. In addition, we introduce a \emph{soft absorbing state} within the diffusion process that reshapes attention dynamics during denoising, enabling faster sequence reconstruction and more precise token refinement. This design accelerates both training and sampling without sacrificing generation quality. Extensive experiments on long-document benchmarks demonstrate that MoE-DiffuSeq consistently outperforms prior diffusion-based and sparse-attention baselines in training efficiency, inference speed, and generation quality. Our approach is particularly effective for long-context applications such as scientific document generation, large-scale code synthesis, and extended dialogue modeling, establishing a scalable and expressive solution for diffusion-based long-form text generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。