Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Ashkan Shahbazi

dblp:349/0217 · DBLP profile ↗
← Back
5ranked-venue papers
2as first author
5since 2021 · last 2026
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 5 · 2 first-author · 5 since 2021Graphics, computer vision, multimedia, augmented reality and games · 1 · 1 first-author · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
4 papers
Deep learning architectures and training · 49% Robot manipulation · 19% Learning theory · 16%
Theoretical computer science
3 papers
Mathematical optimization · 100%
Interdisciplinary, comprehensive, and emerging computing
1 paper
Medical and health informatics · 100%

Topics — the 11 heaviest of 11, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Mathematical optimization
optimal transport
2.532025
Expected Sliced Transport Plans · ICLR 2025
Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data · ICLR 2025
Stereographic Spherical Sliced Wasserstein Distances · ICML 2024
Mathematical optimization › optimal transport
sliced optimal transport
1.722025
Expected Sliced Transport Plans · ICLR 2025
Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data · ICLR 2025
Robotics › Robot manipulation › medical robotics
surgical robotics
1.012026
Neural-Augmented Kelvinlet for Real-Time Soft Tissue Deformation Modeling · AAAI 2026
Medical and health informatics › medical simulation › surgical simulation
soft tissue deformation
1.012026
Neural-Augmented Kelvinlet for Real-Time Soft Tissue Deformation Modeling · AAAI 2026
Machine learning › Deep learning architectures and training
attention mechanism
0.912025
ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans · ICML 2025
Machine learning › Deep learning architectures and training › regularization › attention regularization
doubly stochastic attention
0.912025
ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans · ICML 2025
Machine learning › Deep learning architectures and training › attention mechanism › structured attention
optimal transport attention
0.912025
ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans · ICML 2025
Machine learning › Learning theory
probability metric
0.912025
Expected Sliced Transport Plans · ICLR 2025
Machine learning › Optimization for machine learning › optimal transport
wasserstein distance
0.912025
Expected Sliced Transport Plans · ICLR 2025
Mathematical optimization › optimal transport › wasserstein distance
sliced wasserstein distance
0.812024
Stereographic Spherical Sliced Wasserstein Distances · ICML 2024
Computer animation and physical simulation
deformation simulation
0.312026
Neural-Augmented Kelvinlet for Real-Time Soft Tissue Deformation Modeling · AAAI 2026

Methods — techniques the papers use, named apart from their topics

optimal transport · 3.3neural approximation · 3.0kelvinlet priors · 3.0finite element method · 3.0sliced optimal transport · 1.7entropic regularization · 1.7stereographic projection · 1.5generalized radon transform · 1.5soft sorting · 0.9linear optimal transport · 0.9
YearPublicationVenuePosition
2026 Neural-Augmented Kelvinlet for Real-Time Soft Tissue Deformation Modeling
abstract
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, naively training such models in a fully data-driven manner without incorporating physical priors frequently leads to poor generalization and physically implausible predictions. We present a novel physics-informed neural simulation framework that enables real-time prediction of soft-tissue deformations under complex single- and multi-grasper interactions. Our approach integrates Kelvinlet-based analytical priors with large-scale FEM data, capturing both linear and nonlinear tissue responses. This hybrid design improves predictive accuracy and physical plausibility across diverse neural architectures while maintaining the low-latency performance required for interactive applications. We validate our method on challenging surgical manipulation tasks involving standard laparoscopic grasping tools, demonstrating substantial improvements in deformation fidelity and temporal stability over existing baselines. These results establish Kelvinlet-augmented learning as a principled and computationally efficient paradigm for real-time, physics-aware soft-tissue simulation in surgical AI.
Ashkan Shahbazi, Kyvia Pereira, Jon S. Heiselman, Elaheh Akbari, Annie C. Benson, Sepehr Seifi, Garrison L. H. Johnston, Jie Ying Wu, Nabil Simaan, Michael I. Miga, Soheil Kolouri
AAAI1
2025 Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data
abstract
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 need. These methods reduce the computational burden of optimal transport by slicing hyperspheres into one-dimensional projections, i.e., lines or circles. Concurrently, linear optimal transport has been proposed to embed distributions into $L^2$ spaces, where the $L^2$ distance approximates the optimal transport distance, thereby simplifying comparisons across multiple distributions. In this work, we introduce the Linear Spherical Sliced Optimal Transport (LSSOT) framework, which utilizes slicing to embed spherical distributions into $L^2$ spaces while preserving their intrinsic geometry, offering a computationally efficient metric for spherical probability measures. We establish the metricity of LSSOT and demonstrate its superior computational efficiency in applications such as cortical surface registration, 3D point cloud interpolation via gradient flow, and shape embedding. Our results demonstrate the significant computational benefits and high accuracy of LSSOT in these applications.
Yikun Bai, Rocio Diaz Martin, Ashkan Shahbazi, Bennett A. Landman, Catie Chang, Soheil Kolouri
ICLR5
2025 Expected Sliced Transport Plans
abstract
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 of OT solvers, methods like entropic regularization and sliced optimal transport have been proposed. The sliced OT framework improves efficiency by comparing one-dimensional projections (slices) of high-dimensional distributions. However, despite their computational efficiency, sliced-Wasserstein approaches lack a transportation plan between the input measures, limiting their use in scenarios requiring explicit coupling. In this paper, we address two key questions: Can a transportation plan be constructed between two probability measures using the sliced transport framework? If so, can this plan be used to define a metric between the measures? We propose a ‘lifting’ operation to extend one-dimensional optimal transport plans back to the original space of the measures. By computing the expectation of these lifted plans, we derive a new transportation plan, termed expected sliced transport (EST) plans. We further prove that using the EST plan to weight the sum of the individual Euclidean costs $\|x - y\|^p$ for moving from $x$ to $y$ results in a valid metric between the input discrete probability measures. Finally, we demonstrate the connection between our approach and the recently proposed min-SWGG, along with illustrative numerical examples that support our theoretical findings.
Rocio Diaz Martin, Yikun Bai, Ashkan Shahbazi, Matthew Thorpe, Akram Aldroubi, Soheil Kolouri
ICLR4
2025 ESPFormer: Doubly-Stochastic Attention with Expected Sliced Transport Plans
abstract
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 attention distributions. However, existing methods rely on iterative Sinkhorn normalization, which is computationally costly. In this paper, we introduce a novel, fully parallelizable doubly-stochastic attention mechanism based on sliced optimal transport, leveraging Expected Sliced Transport Plans (ESP). Unlike prior approaches, our method enforces doubly stochasticity without iterative Sinkhorn normalization, significantly enhancing efficiency. To ensure differentiability, we incorporate a temperature-based soft sorting technique, enabling seamless integration into deep learning models. Experiments across multiple benchmark datasets, including image classification, point cloud classification, sentiment analysis, and neural machine translation, demonstrate that our enhanced attention regularization consistently improves performance across diverse applications. Our implementation code can be found at \url{https://github.com/dariansal/ESPFormer}.
Ashkan Shahbazi, Elaheh Akbari, Darian Salehi, Navid NaderiAlizadeh, Soheil Kolouri
ICML1
2024 Stereographic Spherical Sliced Wasserstein Distances
abstract
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 spurred active research in developing computationally efficient variations of these distances for spherical probability measures. This paper introduces a high-speed and highly parallelizable distance for comparing spherical measures using the stereographic projection and the generalized Radon transform, which we refer to as the Stereographic Spherical Sliced Wasserstein (S3W) distance. We carefully address the distance distortion caused by the stereographic projection and provide an extensive theoretical analysis of our proposed metric and its rotationally invariant variation. Finally, we evaluate the performance of the proposed metrics and compare them with recent baselines in terms of both speed and accuracy through a wide range of numerical studies, including gradient flows and self-supervised learning. Our code is available at https://github.com/mint-vu/s3wd.
Yikun Bai, Abihith Kothapalli, Ashkan Shahbazi, Rocio Diaz Martin, Soheil Kolouri
ICML4