5 papers
Variable-Length Generative Protein Design via Generalized Poisson Flow
Chaoran Cheng, Zhanghan Ni, Yanru Qu +4
The ability to generate variable-length proteins is crucial in protein design, where the optimal length is often unknown and tightly coupled to designability. Current diffusion- an…
Accelerating PDE Surrogates via RL-Guided Mesh Optimization
Yang Meng, Ruoxi Jiang, Zhuokai Zhao +3
Deep surrogate models for parametric partial differential equations (PDEs) can deliver high-fidelity approximations but remain prohibitively data-hungry: training often requires th…
Riemannian Consistency Model
Chaoran Cheng, Yusong Wang, Yuxin Chen +3
Consistency models are a class of generative models that enable few-step generation for diffusion and flow matching models. While consistency models have achieved promising results…
Adaptive Divergence Regularized Policy Optimization for Fine-tuning Generative Models
Jiajun Fan, Tong Wei, Chaoran Cheng +2
Balancing exploration and exploitation during reinforcement learning fine-tuning of generative models presents a critical challenge, as existing approaches rely on fixed divergence…
Online Reward-Weighted Fine-Tuning of Flow Matching with Wasserstein Regularization
Jiajun Fan, Shuaike Shen, Chaoran Cheng +3
Recent advancements in reinforcement learning (RL) have achieved great success in fine-tuning diffusion-based generative models. However, fine-tuning continuous flow-based generati…