most citedOnline Reward-Weighted Fine-Tuning of Flow Matching with Wasserstein Regularization

1 citations · 1 across the 4 of their papers we have counts for

collaborators

6 papers

cs.LG2026

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…

cs.LG2026

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…

cs.LG2025

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…

cs.LG2025

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…

cs.AI2025

MindOmni: Unleashing Reasoning Generation in Vision Language Models with RGPO

Yicheng Xiao, Lin Song, Yukang Chen +7

Recent text-to-image systems face limitations in handling multimodal inputs and complex reasoning tasks. We introduce MindOmni, a unified multimodal large language model that addre…

cs.LG2025★ 1 cited

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…