← Search

Shashanka Ubaru

14 accepted papers

2025

Transformers Learn Faster with Semantic Focus

NeurIPS 2025poster

Various forms of sparse attention have been explored to mitigate the quadratic computational and memory cost of the attention mechanism in transformers. We study sparse transformers not through a lens of efficiency but rather in terms of learnability and generalization. Empirically studying a range…

Cited by 0SourceScholar
2024

Asynchronous Randomized Trace Estimation

AISTATS 2024poster

Randomized trace estimation is a popular technique to approximate the trace of an implicitly-defined matrix $A$ by averaging the quadratic form $x’Ax$ across several samples of a random vector $x$. This paper focuses on the application of randomized trace estimators on asynchronous computing environ…

Cited by 3SourcePDFScholar
2024

Topological data analysis on noisy quantum computers

ICLR 2024oral

Topological data analysis (TDA) is a powerful technique for extracting complex and valuable shape-related summaries of high-dimensional data. However, the computational demands of classical algorithms for computing TDA are exorbitant, and quickly become impractical for high-order characteristics. Qu…

Cited by 6SourcePDFScholar
2023

Accelerating Matrix Trace Estimation by Aitken's Δ2 Process

ICASSP 2023accepted

We present an algorithm to estimate the trace of symmetric matrices that are available only via Matrix-Vector multiplication. The proposed algorithm constructs a series of trace estimates by applying the probing technique with an increasing number of vectors. These estimates are then treated as a co…

Cited by 0SourceScholar
2021

Analysis of stochastic Lanczos quadrature for spectrum approximation

ICML 2021oral

The cumulative empirical spectral measure (CESM) $\Phi[\mathbf{A}] : \mathbb{R} \to [0,1]$ of a $n\times n$ symmetric matrix $\mathbf{A}$ is defined as the fraction of eigenvalues of $\mathbf{A}$ less than a given threshold, i.e., $\Phi[\mathbf{A}](x) := \sum_{i=1}^{n} \frac{1}{n} {\large\unicode{x1…

2021

Projection techniques to update the truncated SVD of evolving matrices with applications

ICML 2021spotlight

This submission considers the problem of updating the rank-$k$ truncated Singular Value Decomposition (SVD) of matrices subject to the addition of new rows and/or columns over time. Such matrix problems represent an important computational kernel in applications such as Latent Semantic Indexing and…

2021

Sparse Graph Based Sketching for Fast Numerical Linear Algebra

ICASSP 2021accepted

In recent years, a variety of randomized constructions of sketching matrices have been devised, that have been used in fast algorithms for numerical linear algebra problems, such as least squares regression, low-rank approximation, and the approximation of leverage scores. A key property of sketchin…

Cited by 0SourceScholar
2020

Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping

NeurIPS 2020poster

In modern multilabel classification problems, each data instance belongs to a small number of classes among a large set of classes. In other words, these problems involve learning very sparse binary label vectors. Moreover, in the large-scale problems, the labels typically have certain (unkno…

2017

Union of Intersections (UoI) for Interpretable Data Driven Discovery and Prediction

NeurIPS 2017poster

The increasing size and complexity of scientific data could dramatically enhance discovery and prediction for basic scientific applications, e.g., neuroscience, genetics, systems biology, etc. Realizing this potential, however, requires novel statistical analysis methods that are both interpretable…

Cited by 26SourcePDFScholar