← Search

Ashkan Shahbazi

5 accepted papers

2026

Neural-Augmented Kelvinlet for Real-Time Soft Tissue Deformation Modeling

AAAI 2026technical

Accurate and efficient modeling of soft-tissue interactions is fundamental for advancing surgical simulation, surgical robotics, and model-based surgical automation. To achieve real-time latency, classical Finite Element Method (FEM) solvers are often replaced with neural approximations; however, na

Cited by 0SourcePDFScholar
2025

ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans

ICML 2025poster

While self-attention has been instrumental in the success of Transformers, it can lead to over-concentration on a few tokens during training, resulting in suboptimal information flow. Enforcing doubly-stochastic constraints in attention matrices has been shown to improve structure and balance in att…

2025

Expected Sliced Transport Plans

ICLR 2025poster

The optimal transport (OT) problem has gained significant traction in modern machine learning for its ability to: (1) provide versatile metrics, such as Wasserstein distances and their variants, and (2) determine optimal couplings between probability measures. To reduce the computational complexity…

Cited by 3SourcePDFScholar
2025

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

ICLR 2025spotlight

Efficient comparison of spherical probability distributions becomes important in fields such as computer vision, geosciences, and medicine. Sliced optimal transport distances, such as spherical and stereographic spherical sliced Wasserstein distances, have recently been developed to address this nee…

Cited by 0SourcePDFScholar
2024

Stereographic Spherical Sliced Wasserstein Distances

ICML 2024spotlight

Comparing spherical probability distributions is of great interest in various fields, including geology, medical domains, computer vision, and deep representation learning. The utility of optimal transport-based distances, such as the Wasserstein distance, for comparing probability measures has spur…