works on

From the 1 of 12 linked papers with an AI index.

activity
20242026
collaborators

12 papers

cs.RO2026

GraspMeanFlow: SE(3)-Equivariant MeanFlow for Few-Step 6-DoF Grasp Generation

Jiyong Kwon, Yikun Bai, Amirhossein Mollaali +1

Recent data-driven methods for synthesizing 6-DoF grasp poses use generative models to learn complex grasp pose distributions and generate diverse candidate poses. In particular, S…

cs.LG2026

SE(3)-MeanFlow: Few-Step Protein Backbone Generation on Lie Groups

Yikun Bai, Binghang Lu, Yikai Liu +7

The paper presents SE(3)-MeanFlow, a generative model that creates protein backbone structures directly on the SE(3) Lie group using a few inference steps, avoiding costly ODE inte…

cs.LG2026

Generalized Discrete Diffusion with Self-Correction

Linxuan Wang, Ziyi Wang, Yikun Bai +3

Self-correction is an effective technique for maintaining parallel sampling in discrete diffusion models with minimal performance degradation. Prior work has explored self-correcti…

cs.LG2026

Sinkhorn-Drifting Generative Models

Ping He, Om Khangaonkar, Hamed Pirsiavash +2

We establish a theoretical link between the recently proposed "drifting" generative dynamics and gradient flows induced by the Sinkhorn divergence. In a particle discretization, th…

cs.LG2026

OT-MeanFlow3D: Bridging Optimal Transport and Meanflow for Efficient 3D Point Cloud Generation

Elaheh Akbari, Shansita Sharma, Ping He +5

Flow-matching models have recently emerged as a powerful framework for continuous generative modeling, including 3D point cloud synthesis. However, their deployment is limited by t…

cs.LG2026

LOTFormer: Doubly-Stochastic Linear Attention via Low-Rank Optimal Transport

Ashkan Shahbazi, Chayne Thrash, Yikun Bai +3

Transformers have proven highly effective across modalities, but standard softmax attention scales quadratically with sequence length, limiting long context modeling. Linear attent…