D-DOIT guides frozen discrete diffusion models to black-box rewards without training

新智元 · wechat · 2026-11-15

A NeurIPS 2026 paper introduces D-DOIT, a training-free sampler that adapts masked discrete diffusion models to arbitrary black-box rewards via the Doob h-transform, requiring no gradients or parameter updates. On HepG2 enhancer design it reaches 6.19 predicted activity, beating the fine-tuned DRAKES (5.61); on protein inverse folding it hits a 68.1% joint stability-structure success rate vs 63.6% for SMC and 62.9% for TDS. Key tricks: a prediction-head completion estimator (one forward pass per candidate) and finite-candidate Doob weighting, plus optional late-stage best-of-K branching to hedge irreversible unmasking decisions. Paper: arxiv.org/abs/2610.04938.

Original post →

More from Research

Research channel →