Authors

Jiajun He

Publications 3

Preprint Open access

Fixed-point neural samplers on discrete spaces

Sampling from discrete, unnormalized distributions without access to data is a challenging problem. Neural samplers offer a promising approach by training generative models from density evaluations directly. Despite recent progress, existing discrete neural samplers are prone to mode collapse, come without convergence …

Preprint Open access

ENCORE: Exact Non-equilibrium COntrol with Replica Exchange for Diffusion Generation

Inference-time control steers a pretrained generative model towards a target distribution without retraining. We study tilted targets $π_0\propto G_0\,p_0$, where $p_0$ is the sampler output distribution and $G_0$ is an evaluable reweighting function. Existing approaches rely on sequential annealing with sequential Mon …

Co-authors