Unified Audio Generation and Editing via Joint Condition Modeling and Progressive Training

2026-06-15Sound

Sound
AI summary

The authors created AudioWeave, a single model that can both generate audio from text and edit audio, instead of having separate systems for each task. They designed a way for the model to handle different types of input using a special embedding technique and trained it carefully to avoid confusion between tasks. Their experiments showed this unified model performs about as well as individual task-specific models. This work helps simplify building audio generation and editing tools.

text-to-audioaudio generationaudio editingdiffusion transformerposition embeddingmultistage trainingtask interferencecatastrophic forgettingunified modelingcondition modeling
Authors
Haocheng Dong, Yuheng Lu, Cheng Gong, Shansong Liu, Xiao-Lei Zhang, Xuelong Li
Abstract
With the growing focus on audio in multimedia applications, numerous advanced works on audio generation have emerged. Existing studies typically treat text-to-audio (TTA) and other related audio generation tasks, such as instruction-based audio editing, as independent challenges, adopting task-specific architectures or modules. This absence of a unified modeling paradigm substantially increases the overhead and complexity of building a system for both audio generation and editing, while also leading to limited scalability. To address this issue, we introduce AudioWeave, a unified model for TTA and audio editing without additional task-specific components. Specifically, we propose a joint condition modeling approach with a factorized position embedding, enabling the diffusion transformer backbone to operate under heterogeneous inputs of TTA and audio editing. We further propose a progressive multistage training strategy to mitigate task competition and catastrophic forgetting caused by interference among multiple tasks. This in turn helps maintain the performance of each individual task and may even lead to improvements in certain aspects. Experimental results on TTA task and six audio editing tasks show that our unified model achieves competitive performance with task-specific models, laying a groundwork for further exploration of unified audio generation models.