通过表示空间监督缓解文本到图像扩散模型的记忆问题
Mitigating Memorization In Text-To-Image Diffusion Models With Representation-Space Supervision
基于摘要分析。本文研究文本到图像扩散模型复现个别训练样本的记忆问题,提出 Representation-Space Supervision(RSS),将预测与目标的匹配从数据空间转移到预训练视觉表示模型的特征空间,以缓解涉及隐私与版权的训练样本复现。 核心机制:作者假设,直接在数据空间施加监督会鼓励模型恢复精确的训练样本,从而促成记忆。RSS 利用预训练视觉表示保留高层感知与语义信息、弱化实例特有细节的性质,改变扩散预测的监督空间。这一思路侧重训练目标,而非摘要所述既有方法常用的数据预处理或专门采样流程。具体特征提取器、匹配损失、监督位置,以及是否与原始扩散损失联合使用,尚未验证。 结果与证据边界:作者报告,在受控的文本到图像监督微调实验中,相较标准扩散训练,RSS 持续且显著地改善了生成性能与训练样本复现之间的权衡。摘要未给出数据集、模型规模、数据划分、生成性能与记忆的具体指标、效应量或统计检验,因而目前不能量化收益,也不能据此断言其提供隐私保证。实验协议、消融及统计信息尚未验证。 该研究的主要对象是图像生成,而非 EEG 或 BCI。机制上依赖能保留任务相关信息、同时弱化实例细节的预训练表示;这一条件在神经信号中是否成立尚不明确,现有材料不足以支持具体迁移方案。已有 EEG 相关工作尚未检索确认。
值得核对 RSS 是否能通过改变监督空间改善生成质量与训练样本复现之间的权衡,尤其需验证特征表示选择及记忆评估协议对结论的影响。
来源:OpenReview · State space models · openreview.net