arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.02130cs.SDcs.CVeess.AS

MDSGen:用于开放域声音生成的快速高效的掩码扩散时间感知Transformer

MDSGen: Fast and Efficient Masked Diffusion Temporal-Aware Transformers for Open-Domain Sound Generation

Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院(KAIST))

Trung X. Pham, Tri Ton, Chang D. Yoo

AI总结:

提出MDSGen框架,通过冗余视频特征移除模块和时间感知掩码策略,结合去噪掩码扩散Transformer,在VGGSound数据集上实现高效视觉引导开放域声音生成,小模型参数少172倍、内存少371%、推理快36倍且精度更高。

摘要:

我们提出MDSGen,这是一种针对模型参数规模、内存消耗和推理速度优化的视觉引导开放域声音生成新框架。该框架包含两项关键创新:(1)冗余视频特征移除模块,用于过滤不必要的视觉信息;(2)时间感知掩码策略,利用时间上下文提高音频生成精度。与现有资源密集型的基于Unet的模型不同,MDSGen采用去噪掩码扩散Transformer,无需依赖预训练扩散模型即可实现高效生成。在基准数据集VGGSound上评估,我们的最小模型(5M参数)达到97.9%的对齐精度,与当前860M参数的最先进模型(93.9%精度)相比,参数少172倍,内存消耗少371%,推理速度快36倍。较大模型(131M参数)精度接近99%,同时参数少6.5倍。这些结果突显了我们方法的可扩展性和有效性。代码可在https://bit.ly/mdsgen获取。

英文摘要:

We introduce MDSGen, a novel framework for vision-guided open-domain sound generation optimized for model parameter size, memory consumption, and inference speed. This framework incorporates two key innovations: (1) a redundant video feature removal module that filters out unnecessary visual information, and (2) a temporal-aware masking strategy that leverages temporal context for enhanced audio generation accuracy. In contrast to existing resource-heavy Unet-based models, \texttt{MDSGen} employs denoising masked diffusion transformers, facilitating efficient generation without reliance on pre-trained diffusion models. Evaluated on the benchmark VGGSound dataset, our smallest model (5M parameters) achieves $97.9$% alignment accuracy, using $172\times$ fewer parameters, $371$% less memory, and offering $36\times$ faster inference than the current 860M-parameter state-of-the-art model ($93.9$% accuracy). The larger model (131M parameters) reaches nearly $99$% accuracy while requiring $6.5\times$ fewer parameters. These results highlight the scalability and effectiveness of our approach. The code is available at https://bit.ly/mdsgen.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑