Equivariant vs. Invariant Layers: A Comparison of Backbone and Pooling for Point Cloud Classification. - Ashkan Shahbazi; Abihith Kothapalli; Xinran Liu; Robert Sheng; Soheil Kolouri (CoRR, 2023)

Stereographic Spherical Sliced Wasserstein Distances. - Huy Tran; Yikun Bai; Abihith Kothapalli; Ashkan Shahbazi; Xinran Liu; Rocio Diaz Martin; Soheil Kolouri (ICML, 2024)

Efficient Solvers for Partial Gromov-Wasserstein. - Yikun Bai; Rocio Diaz Martin; Hengrong Du; Ashkan Shahbazi; Soheil Kolouri (CoRR, 2024)

One Category One Prompt: Dataset Distillation using Diffusion Models. - Ali Abbasi 008; Ashkan Shahbazi; Hamed Pirsiavash; Soheil Kolouri (CoRR, 2024)

Understanding Learning with Sliced-Wasserstein Requires Rethinking Informative Slices. - Huy Tran; Yikun Bai; Ashkan Shahbazi; John R. Hershey; Soheil Kolouri (CoRR, 2024)

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 (ICLR, 2025)

Expected Sliced Transport Plans. - Xinran Liu; Rocio Diaz Martin; Yikun Bai; Ashkan Shahbazi; Matthew Thorpe; Akram Aldroubi; Soheil Kolouri (ICLR, 2025)

ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans. - Ashkan Shahbazi; Elaheh Akbari; Darian Salehi; Xinran Liu; Navid Naderializadeh; Soheil Kolouri (CoRR, 2025)

Neural-Augmented Kelvinlet: Real-Time Soft Tissue Deformation with Multiple Graspers. - Ashkan Shahbazi; Kyvia Pereira; Jon S. Heiselman; Elaheh Akbari; Annie C. Benson; Sepehr Seifi; Xinyuan Liu; Garrison L. H. Johnston; Erwin Terpstra; Anne Draaisma; Jan-Jaap Severes; Jie Ying Wu; et al. (CoRR, 2025)