Yikun Bai

According to our database, Yikun Bai authored at least 23 papers between 2020 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

Sinkhorn-Drifting Generative Models.

Ping He, Om Khangaonkar, Hamed Pirsiavash, Yikun Bai, Soheil Kolouri

CoRR, March, 2026

Generalized Discrete Diffusion with Self-Correction.

Linxuan Wang, Ziyi Wang, Yikun Bai, Wei Deng, Guang Lin, Qifan Song

CoRR, March, 2026

A Case for Vanilla SWD: New Perspectives on Informative Slices, Sliced-Wasserstein Distances, and Learning Rates.

Huy Tran, Yikun Bai, Ashkan Shahbazi, John R. Hershey, David Hyde, Soheil Kolouri

Trans. Mach. Learn. Res., 2026

2025

LUNA: Linear Universal Neural Attention with Generalization Guarantees.

Ashkan Shahbazi, Ping He, Ali Abbasi, Yikun Bai, Xinran Liu, Elaheh Akbari, Darian Salehi, Navid Naderializadeh, Soheil Kolouri

CoRR, December, 2025

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

Ashkan Shahbazi, Chayne Thrash, Yikun Bai, Keaton Hamm, Navid Naderializadeh, Soheil Kolouri

CoRR, September, 2025

Transport Based Mean Flows for Generative Modeling.

Elaheh Akbari, Ping He, Ahmadreza Moradipari, Yikun Bai, Soheil Kolouri

CoRR, September, 2025

Fused Partial Gromov-Wasserstein for Structured Objects.

Yikun Bai, Huy Tran, Hengrong Du, Xinran Liu, Soheil Kolouri

CoRR, February, 2025

Wasserstein task embedding for measuring task similarities.

Xinran Liu, Yikun Bai, Yuzhe Lu, Andrea Soltoggio, Soheil Kolouri

Neural Networks, 2025

Expected Sliced Transport Plans.

Xinran Liu, Rocio Diaz Martin, Yikun Bai, Ashkan Shahbazi, Matthew Thorpe, Akram Aldroubi, Soheil Kolouri

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data.

Xinran Liu, Yikun Bai, Rocio Diaz Martin, Kaiwen Shi, Ashkan Shahbazi, Bennett Allan Landman, Catie Chang, Soheil Kolouri

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

Partial Gromov-Wasserstein Metric.

Yikun Bai, Rocio Diaz Martin, Abihith Kothapalli, Hengrong Du, Xinran Liu, Soheil Kolouri

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

Linear Partial Gromov-Wasserstein Embedding.

Yikun Bai, Abihith Kothapalli, Hengrong Du, Rocio Diaz Martin, Soheil Kolouri

Proceedings of the Thirteenth International Conference on Learning Representations, 2025

2024

Understanding Learning with Sliced-Wasserstein Requires Rethinking Informative Slices.

Huy Tran, Yikun Bai, Ashkan Shahbazi, John R. Hershey, Soheil Kolouri

CoRR, 2024

Sinkhorn algorithms and linear programming solvers for optimal partial transport problems.

Yikun Bai

CoRR, 2024

Efficient Solvers for Partial Gromov-Wasserstein.

Yikun Bai, Rocio Diaz Martin, Hengrong Du, Ashkan Shahbazi, Soheil Kolouri

CoRR, 2024

Stereographic Spherical Sliced Wasserstein Distances.

Huy Tran, Yikun Bai, Abihith Kothapalli, Ashkan Shahbazi, Xinran Liu, Rocio Diaz Martin, Soheil Kolouri

Proceedings of the Forty-first International Conference on Machine Learning, 2024

LCOT: Linear Circular Optimal Transport.

Rocio Diaz Martin, Ivan Vladimir Medri, Yikun Bai, Xinran Liu, Kangbai Yan, Gustavo K. Rohde, Soheil Kolouri

Proceedings of the Twelfth International Conference on Learning Representations, 2024

2023

Partial Transport for Point-Cloud Registration.

Yikun Bai, Huy Tran, Steven B. Damelin, Soheil Kolouri

CoRR, 2023

PTL^p: Partial Transport L^p Distances.

Xinran Liu, Yikun Bai, Huy Tran, Zhanqi Zhu, Matthew Thorpe, Soheil Kolouri

CoRR, 2023

Linear optimal partial transport embedding.

Yikun Bai, Ivan Vladimir Medri, Rocio Diaz Martin, Rana Muhammad Shahroz Khan, Soheil Kolouri

Proceedings of the International Conference on Machine Learning, 2023

Sliced Optimal Partial Transport.

Yikun Bai, Bernhard Schmitzer, Matthew Thorpe, Soheil Kolouri

Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023

2021

Understanding Entropic Regularization in GANs.

Daria Reshetova, Yikun Bai, Xiugang Wu, Ayfer Özgür

Proceedings of the IEEE International Symposium on Information Theory, 2021

2020

Information Constrained Optimal Transport: From Talagrand, to Marton, to Cover.

Yikun Bai, Xiugang Wu, Ayfer Özgür

Proceedings of the IEEE International Symposium on Information Theory, 2020