Simplex Diffusion Models
单纯形扩散模型:离散扩散的概率单纯形框架
🧭 离散扩散 · 概率单纯形 ★ 深度解读
Google DeepMind · EPFL · UCL Gatsby
提出在概率单纯形上扩散的 SDM,保留中间不确定性,用 DDIM 式采样统一离散与连续扩散
📌 工作参考价值(4 点)
- 生成式推荐/ Semantic ID 生成中,离散扩散每一步仅采样一个 token 会丢失分布,SDM 用单纯形上的分布作为中间状态,可保留候选 item 的不确定性,减少误差累积;无需 Self-Conditioning 的额外前向。
- 大词表场景(如 item ID 词表很大)可以用 argmax embedding 替代期望 embedding,避免 O(V*d) 矩阵乘法,同时保持推理连续状态;这是工程上可落地的 trick。
- DDIM-like 采样器有 churn 参数,可调节随机性(实验中 κ=1 通常最好),且支持蒸馏到极低步数(8 步),有利于在线生成式推荐的低延迟部署。
- 用 normalized variance ν_t 调度浓度比直接调 c_t 更直观,便于在不同任务上快速调参;自适应时间采样也能提升性能。