跳到正文
原文
arXiv cs.AI· Shai Dickman, Mert Cemri, Landon Butler, Kannan Ramchandran·· 1 天前AI 评分37

Spectral Feedback:蛋白质扩散模型的测试时对齐方法

Spectral Feedback for Test-Time Alignment of Protein Diffusion Models

AI 导读

Spectral Feedback 通过反馈回路选择编辑位置、重新 mask 并重采样 token,让离散扩散模型迭代修正自身生成结果。该方法与模型无关,可用于预训练、测试时对齐与微调的扩散模型,且不修改底层生成过程。在蛋白质逆折叠任务上配合稳定性 reward oracle,稳定蛋白比例提升 32.3%,Best-of-10 提升 24.8%,SOTA RL 微调扩散模型提升 5.8%。

来源:arXiv cs.AI · arxiv.org