在实验室调试一个蛋白质构象生成模型，损失函数震荡了三小时就是不收敛。盯着损失曲线突然意识到：我们习惯在二维图像上做扩散，但分子结构是三维的、有物理约束的、带化学键的——直接把图像那套搬过来，不崩才怪。分子扩散的核心挑战分子和蛋白质不是像素点。每个原子有类型、电荷、键能，原子间距离不是随便定的。早期尝试直接把原子坐标当点云做扩散，生成的结构化学上根本不合理——氢氧键长能差出两倍，键角扭曲得像个现代艺术雕塑。关键突破在于坐标和类型的分离建模。原子坐标是连续的，可以用标准扩散；原子类型是离散的，需要分类扩散。但这两者必须协同更新，不然坐标漂移了类型还对不上。# 分子扩散的典型前向过程defforward_diffusion(mol):# 坐标扩散：加高斯噪声coords_noisy=coords+sqrt