扩散模型的泛化源于正则化的表征空间
Generalization of Diffusion Models Arises from a Regularized Representation Space
基于摘要分析。本文研究扩散模型为何既能生成新样本,也可能在过度拟合训练目标时记忆训练数据,并从表征学习角度解释两者的区别。贡献包括基于去噪自编码器的理论分析、实际扩散模型中的表征考察,以及表征驱动的记忆检测与免训练编辑方法。 在理论部分,作者采用两层 ReLU 去噪自编码器(DAE)参数化,证明在其分析设定下,记忆对应于模型学习用于编码和解码的原始数据矩阵,并形成尖峰式表征;泛化则对应于捕获局部数据统计、形成较均衡的表征。这一结论的适用假设、表征均衡性的具体定义,以及标题中“正则化”对应的数学条件,尚需全文核对,不能据摘要推断作者引入了显式正则化损失。 作者报告,在实际无条件扩散模型与 text-to-image 扩散模型的表征空间中观察到了相同区分,并提出基于表征的记忆检测方法,以及通过 representation steering 实现精确控制的免训练编辑方法。摘要未提供具体模型、数据集、评估指标或数值;实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 扩散生成模型中也存在可区分样本记忆与局部统计学习的表征结构,该分析或可用于辅助检查训练样本复现风险。可尝试复用表征提取与记忆检测思路,但需针对 EEG 时序、通道结构及生理有效性改造特征和评价标准;低信噪比、跨被试差异和小数据可能使表征尖峰反映伪迹或个体特征,而非记忆。一个可证伪的小实验是在固定数据划分下比较不同拟合程度的 EEG 扩散模型,检验表征形态是否与生成样本对训练样本的近邻相似性同步变化,并用留出数据校验其区分能力。已有 EEG 相关工作尚未检索确认。
值得核对其将记忆与泛化联系到表征形态的理论条件,以及基于表征的记忆检测和免训练编辑是否在实际扩散模型中获得一致证据;其对 EEG 生成的适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net