arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.02447cs.LG

扩散模型的频谱正则化

Spectral Regularization for Diffusion Models

ucr_math(加州大学河滨分校数学系);ucr_cs(加州大学河滨分校计算机科学系)

Satish Chandran, Nicolas Roque dos Santos, Yunshu Wu, Greg Ver Steeg, Evangelos Papalexakis

AI总结:

本文提出了一种基于频谱正则化的扩散模型训练方法,通过引入可微的傅里叶和小波域损失,提升生成样本的频率平衡和多尺度结构质量。

摘要:

扩散模型通常使用点wise重建目标进行训练,这些目标对自然信号的频谱和多尺度结构漠不关心。我们提出了一种损失层面的频谱正则化框架,通过可微的傅里叶域和小波域损失增强标准扩散训练,而无需修改扩散过程、模型架构或采样过程。所提出的正则化器作为软归纳偏置,鼓励生成样本中适当的频率平衡和相干的多尺度结构。我们的方法与DDPM、DDIM和EDM公式兼容,并引入了 negligible 的计算开销。在图像和音频生成上的实验表明,样本质量有持续改进,其中在高分辨率无条件数据集上获得的最大收益发生在细尺度结构最难以建模的情况下。

英文摘要:

Diffusion models are typically trained using pointwise reconstruction objectives that are agnostic to the spectral and multi-scale structure of natural signals. We propose a loss-level spectral regularization framework that augments standard diffusion training with differentiable Fourier- and wavelet-domain losses, without modifying the diffusion process, model architecture, or sampling procedure. The proposed regularizers act as soft inductive biases that encourage appropriate frequency balance and coherent multi-scale structure in generated samples. Our approach is compatible with DDPM, DDIM, and EDM formulations and introduces negligible computational overhead. Experiments on image and audio generation demonstrate consistent improvements in sample quality, with the largest gains observed on higher-resolution, unconditional datasets where fine-scale structure is most challenging to model.

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

↑