In recent years, diffusion models have emerged as a novel paradigm in generative modeling, achieving breakthroughs in image, audio-video, and text generation tasks. However, in the domain of text generation, existing continuous diffusion-based models struggle to efficiently capture temporal dependencies in long sequences and exhibit limited robustness in complex noisy environments, which significantly affects the quality and diversity of the generated text. To address these challenges, we propose MM-Diffusion, a novel continuous diffusion-based text generation model. MM-Diffusion incorporates a parallel architecture that combines Mamba and Transformer during the reverse denoising process, leveraging Mamba’s efficiency in long-sequence modeling and the Transformer’s ability to capture complex semantics, thereby facilitating effective modeling of long-text temporal dependencies. Furthermore, we introduce an adaptive noise adjustment module based on a mixture-of-experts framework. The dynamic routing mechanism within this module adaptively activates a variable number of expert networks based on token complexity, enabling fine-grained noise control and enhancing text diversity. Experimental results on four benchmark datasets demonstrate that MM-Diffusion achieves effective improvements across multiple performance metric.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Continuous Diffusion Text Generation Model with a New Denoising Structure

  • Sihui Li,
  • Guoyong Cai,
  • Yuan Jin,
  • Yunxian Shang

摘要

In recent years, diffusion models have emerged as a novel paradigm in generative modeling, achieving breakthroughs in image, audio-video, and text generation tasks. However, in the domain of text generation, existing continuous diffusion-based models struggle to efficiently capture temporal dependencies in long sequences and exhibit limited robustness in complex noisy environments, which significantly affects the quality and diversity of the generated text. To address these challenges, we propose MM-Diffusion, a novel continuous diffusion-based text generation model. MM-Diffusion incorporates a parallel architecture that combines Mamba and Transformer during the reverse denoising process, leveraging Mamba’s efficiency in long-sequence modeling and the Transformer’s ability to capture complex semantics, thereby facilitating effective modeling of long-text temporal dependencies. Furthermore, we introduce an adaptive noise adjustment module based on a mixture-of-experts framework. The dynamic routing mechanism within this module adaptively activates a variable number of expert networks based on token complexity, enabling fine-grained noise control and enhancing text diversity. Experimental results on four benchmark datasets demonstrate that MM-Diffusion achieves effective improvements across multiple performance metric.