Demo Data from EDBT 2026 / reviewers in the wild / expert
Shashanka Ubaru
dblp:164/7307
· DBLP profile ↗
25ranked-venue papers
11as first author
12since 2021 · last 2026
0000-0001-6942-7158ORCID · corroborated
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 14 · 7 first-author · 6 since 2021Graphics, computer vision, multimedia, augmented reality and games · 4 · 3 since 2021Databases, data management, data science and information retrieval · 2 · 1 first-author · 1 since 2021Theory of computation · 2 · 1 first-authorApplied, interdisciplinary, general and emerging computing · 2 · 2 first-author · 1 since 2021Systems, architecture and hardware · 1 · 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.
| Theoretical computer science
8 papers |
Algorithms and data structures · 39% Computational geometry · 21% Coding theory · 19% | |
| Artificial intelligence
5 papers |
Learning paradigms · 43% Graph learning · 19% Learning theory · 16% | |
| Databases, data mining, and information retrieval
2 papers |
Information retrieval · 54% Data mining · 46% | |
| Computer architecture, parallel and distributed computing, and storage systems
2 papers |
Emerging computing paradigms · 60% GPUs and heterogeneous computing · 20% Distributed systems · 20% |
Topics — the 27 heaviest of 32, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Algorithms and data structures
numerical linear algebra |
1.0 | 2 | 2021 | Projection techniques to update the truncated SVD of evolving matrices with applications · ICML 2021 Analysis of stochastic Lanczos quadrature for spectrum approximation · ICML 2021 |
Coding theory
error-correcting codes |
0.8 | 3 | 2017 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting Codes · IEEE Trans. Inf. Theory 2017 Multilabel Classification with Group Testing and Codes · ICML 2017 Low Rank Approximation using Error Correcting Coding Matrices · ICML 2015 |
Computational geometry › topological data analysis
persistent homology |
0.8 | 1 | 2024 | Topological data analysis on noisy quantum computers · ICLR 2024 |
Quantum computing and quantum information
quantum machine learning |
0.8 | 1 | 2024 | Topological data analysis on noisy quantum computers · ICLR 2024 |
Computational geometry
topological data analysis |
0.8 | 1 | 2024 | Topological data analysis on noisy quantum computers · ICLR 2024 |
Machine learning › Learning paradigms
multi-label classification |
0.7 | 2 | 2020 | Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping · NeurIPS 2020 Multilabel Classification with Group Testing and Codes · ICML 2017 |
Algorithms and data structures › matrix approximation
low-rank approximation |
0.5 | 2 | 2017 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting Codes · IEEE Trans. Inf. Theory 2017 Low Rank Approximation using Error Correcting Coding Matrices · ICML 2015 |
Machine learning › Graph learning › graph neural network
dynamic graph neural network |
0.5 | 1 | 2021 | Efficient scaling of dynamic graph neural networks · SC 2021 |
Information retrieval › retrieval models › latent semantic models
latent semantic indexing |
0.5 | 1 | 2021 | Projection techniques to update the truncated SVD of evolving matrices with applications · ICML 2021 |
Algorithms and data structures › matrix approximation
spectral approximation |
0.5 | 1 | 2021 | Analysis of stochastic Lanczos quadrature for spectrum approximation · ICML 2021 |
Machine learning › Learning paradigms › multi-label classification
hierarchical multi-label classification |
0.4 | 1 | 2020 | Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping · NeurIPS 2020 |
Data mining › representation learning
label embedding |
0.4 | 1 | 2020 | Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping · NeurIPS 2020 |
Algorithms and data structures › numerical linear algebra
matrix factorization |
0.4 | 2 | 2020 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting Codes · IEEE Trans. Inf. Theory 2017 Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping · NeurIPS 2020 |
Machine learning › Learning theory › classification › multiclass classification
extreme classification |
0.3 | 1 | 2017 | Multilabel Classification with Group Testing and Codes · ICML 2017 |
Coding theory › error-correcting codes
reed-solomon codes |
0.3 | 1 | 2017 | Multilabel Classification with Group Testing and Codes · ICML 2017 |
Mathematical optimization › numerical analysis
eigenvalue problem |
0.2 | 1 | 2016 | Fast methods for estimating the Numerical rank of large matrices · ICML 2016 |
Information theory › signal processing › spectral estimation
spectral density estimation |
0.2 | 1 | 2016 | Fast methods for estimating the Numerical rank of large matrices · ICML 2016 |
Emerging computing paradigms › quantum computing
NISQ |
0.2 | 1 | 2024 | Topological data analysis on noisy quantum computers · ICLR 2024 |
Emerging computing paradigms
quantum computer architecture |
0.2 | 1 | 2024 | Topological data analysis on noisy quantum computers · ICLR 2024 |
Machine learning › Learning theory › approximation theory
approximation error bound |
0.1 | 1 | 2021 | Analysis of stochastic Lanczos quadrature for spectrum approximation · ICML 2021 |
Machine learning › Efficient and distributed learning › distributed training › communication-efficient training
communication optimization |
0.1 | 1 | 2021 | Efficient scaling of dynamic graph neural networks · SC 2021 |
Machine learning › Efficient and distributed learning
distributed training |
0.1 | 1 | 2021 | Efficient scaling of dynamic graph neural networks · SC 2021 |
Distributed systems › distributed machine learning › distributed training
distributed GNN training |
0.1 | 1 | 2021 | Efficient scaling of dynamic graph neural networks · SC 2021 |
GPUs and heterogeneous computing
multi-GPU computing |
0.1 | 1 | 2021 | Efficient scaling of dynamic graph neural networks · SC 2021 |
Algorithms and data structures › numerical linear algebra › matrix factorization › low-rank matrix factorization
nonnegative matrix factorization |
0.1 | 1 | 2020 | Multilabel Classification by Hierarchical Partitioning and Data-dependent Grouping · NeurIPS 2020 |
Mathematical optimization
least squares |
0.1 | 1 | 2017 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting Codes · IEEE Trans. Inf. Theory 2017 |
Mathematical optimization › least squares
linear least squares |
0.1 | 1 | 2017 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting Codes · IEEE Trans. Inf. Theory 2017 |
Methods — techniques the papers use, named apart from their topics
group testing · 1.9topological data analysis · 1.5quantum circuit · 1.5hierarchical partitioning · 1.3wasserstein distance · 1.0subspace approximation · 1.0stochastic trace estimation · 1.0projection techniques · 1.0lanczos algorithm · 1.0graph difference-based strategy · 1.0data distribution technique · 1.0non-negative matrix factorization · 0.9nonnegative matrix factorization · 0.4logistic regression · 0.3intersection and union operations · 0.3hamming loss bound · 0.3binary classifier · 0.3LASSO · 0.3
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Capacity Analysis of Vector Symbolic ArchitecturesabstractHyperdimensional computing (HDC) is a biologically-inspired framework which represents symbols with high-dimensional vectors, and uses vector operations to manipulate them. The ensemble of a particular vector space and a prescribed set of vector operations (e.g., addition-like for “bundling” and outer-product-like for “binding”), as targeted to HDC, forms a vector symbolic architecture (VSA). While VSAs have been employed in numerous learning applications and have been studied empirically, many theoretical questions about VSAs remain open. In this paper, we analyze the representation capacities of four common VSAs: MAP-I (vectors take integer values), MAP-B (binary vectors and operations), and two VSAs based on sparse binary vectors. “Representation capacity” here refers to bounds on the dimensions of the VSA vectors required to perform certain symbolic tasks, such as testing for set membership and estimating set intersection sizes for two sets of symbols, to a given degree of accuracy. We also propose a novel variant of a Hopfield network (a simple model of associative memory), and analyze its ability to perform some of the same tasks that are typically asked of VSAs. Our analyses establish and leverage connections between VSAs, matrix sketching algorithms, and Bloom filters. In particular, some of our analyses amount to showing that certain random projections with less than full randomness have length-preserving properties, and we give novel analyses of Bloom filters and Counting Bloom filters, with regard to rapid estimation of the size of set intersections. Kenneth L. Clarkson, Shashanka Ubaru, Elizabeth Yang |
J. Artif. Intell. Res. | 2 |
| 2025 | Transformers Learn Faster with Semantic FocusabstractVarious 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 of attention mechanisms, we find that input-dependent sparse attention models appear to converge faster and generalize better than standard attention models, while input-agnostic sparse attention models show no such benefits -- a phenomenon that is robust across architectural and optimization hyperparameter choices. This can be interpreted as demonstrating that concentrating a model's "semantic focus" with respect to the tokens currently being considered (in the form of input-dependent sparse attention) accelerates learning. We develop a theoretical characterization of the conditions that explain this behavior. We establish a connection between the stability of the standard softmax and the loss function's Lipschitz properties, then show how sparsity affects the stability of the softmax and the subsequent convergence and generalization guarantees resulting from the attention mechanism. This allows us to theoretically establish that input-agnostic sparse attention does not provide any benefits. We also characterize conditions when semantic focus (input-dependent sparse attention) can provide improved guarantees, and we validate that these conditions are in fact met in our empirical evaluations. Parikshit Ram, Kenneth L. Clarkson, Tim Klinger, Shashanka Ubaru, Alexander G. Gray |
NeurIPS | 4 |
| 2024 | Asynchronous Randomized Trace EstimationabstractRandomized 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 environments where the quadratic form $x’Ax$ is computed partially by observing only a random row subset of $A$ for each sample of the random vector $x$. Our asynchronous framework treats the number of rows, as well as the row subset observed for each sample, as random variables, and our theoretical analysis establishes the variance of the randomized estimator for Rademacher and Gaussian samples. We also present error analysis and sampling complexity bounds for the proposed asynchronous randomized trace estimator. Our numerical experiments illustrate that the asynchronous variant can be competitive even when a small number of rows is updated per each sample. Vasileios Kalantzis, Shashanka Ubaru, Chai Wah Wu, Georgios Kollias, Lior Horesh |
AISTATS | 2 |
| 2024 | Topological data analysis on noisy quantum computersabstractTopological 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. Quantum computers offer the potential of achieving significant speedup for certain computational problems. Indeed, TDA has been purported to be one such problem, yet, quantum computing algorithms proposed for the problem, such as the original Quantum TDA (QTDA) formulation by Lloyd, Garnerone and Zanardi, require fault-tolerance qualifications that are currently unavailable. In this study, we present NISQ-TDA, a fully implemented end-to-end quantum machine learning algorithm needing only a short circuit-depth, that is applicable to high-dimensional classical data, and with provable asymptotic speedup for certain classes of problems. The algorithm neither suffers from the data-loading problem nor does it need to store the input data on the quantum computer explicitly. The algorithm was successfully executed on quantum computing devices, as well as on noisy quantum simulators, applied to small datasets. Preliminary empirical results suggest that the algorithm is robust to noise. Ismail Yunus Akhalwaya, Shashanka Ubaru, Kenneth L. Clarkson, Mark S. Squillante, Vishnu Jejjala, Yang-Hui He, Kugendran Naidoo, Vasileios Kalantzis, Lior Horesh |
ICLR | 2 |
| 2023 | Accelerating Matrix Trace Estimation by Aitken's Δ2 ProcessabstractWe 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 converging sequence whose limit is the sought matrix trace, and we apply Aitken’s Δ2process to accelerate its convergence to the trace limit. Numerical experiments performed on covariance matrices demonstrate the competitiveness of the proposed scheme versus probing and randomized trace estimators. Vassilis Kalantzis, Georgios Kollias, Shashanka Ubaru, Theodoros Salonidis |
ICASSP | 3 |
| 2023 | Quantum Graph TransformersabstractWe propose Quantum Graph Transformers (QGT), a novel approach for realizing the Transformer architecture for graph learning with quantum processors. QGT is built on top of the Graph Trans-former (GT) architecture and addresses the main challenge of mapping GT basic functions such as node encodings, graph structure, all-to-all connectivity, and message passing to quantum computing primitives and processors. We empirically demonstrate the training and inference efficacy of our proposed QGT architecture for the graph classification task on quantum devices over various graph datasets. Georgios Kollias, Vassilis Kalantzis, Theodoros Salonidis, Shashanka Ubaru |
ICASSP | 4 |
| 2021 | Sparse Graph Based Sketching for Fast Numerical Linear AlgebraabstractIn 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 sketching matrices is that of subspace embedding. In this paper, we study sketching matrices that are obtained from bipartite graphs that are sparse, i.e., have left degree s that is small. In particular, we explore two popular classes of sparse graphs, namely, expander graphs and magical graphs. For a given subspace ${\mathcal{U}} \subseteq {{\mathbb{R}}^n}$ of dimension k, we show that the magical graph with left degree s = 2 yields a (1 ± ϵ) ℓ2-subspace embedding for ${\mathcal{U}}$, if the number of right vertices (the sketch size) $m = {\mathcal{O}}\left( {{k^2}/{\varepsilon ^2}} \right)$. The expander graph with $s = {\mathcal{O}}(\log k/\varepsilon )$ yields a subspace embedding for $m = {\mathcal{O}}\left( {k\log k/{\varepsilon ^2}} \right)$. We also discuss the construction of sparse sketching matrices with reduced randomness using expanders based on error-correcting codes. Empirical results on various synthetic and real datasets show that these sparse graph sketching matrices work very well in practice. Shashanka Ubaru, Alex Gittens, Kenneth L. Clarkson, Lior Horesh, Vassilis Kalantzis |
ICASSP | 2 |
| 2021 | Analysis of stochastic Lanczos quadrature for spectrum approximationabstractThe 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{x1D7D9}}[ \lambda_i[\mathbf{A}]\leq x]$. Spectral sums $\operatorname{tr}(f[\mathbf{A}])$ can be computed as the Riemann–Stieltjes integral of $f$ against $\Phi[\mathbf{A}]$, so the task of estimating CESM arises frequently in a number of applications, including machine learning. We present an error analysis for stochastic Lanczos quadrature (SLQ). We show that SLQ obtains an approximation to the CESM within a Wasserstein distance of $t \: | \lambda_{\text{max}}[\mathbf{A}] - \lambda_{\text{min}}[\mathbf{A}] |$ with probability at least $1-\eta$, by applying the Lanczos algorithm for $\lceil 12 t^{-1} + \frac{1}{2} \rceil$ iterations to $\lceil 4 ( n+2 )^{-1}t^{-2} \ln(2n\eta^{-1}) \rceil$ vectors sampled independently and uniformly from the unit sphere. We additionally provide (matrix-dependent) a posteriori error bounds for the Wasserstein and Kolmogorov–Smirnov distances between the output of this algorithm and the true CESM. The quality of our bounds is demonstrated using numerical experiments. Tyler Chen, Thomas Trogdon, Shashanka Ubaru |
ICML | 3 |
| 2021 | Projection techniques to update the truncated SVD of evolving matrices with applicationsabstractThis 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 Recommender Systems. Nonetheless, the proposed framework is purely algebraic and targets general updating problems. The algorithm presented in this paper undertakes a projection viewpoint and focuses on building a pair of subspaces which approximate the linear span of the sought singular vectors of the updated matrix. We discuss and analyze two different choices to form the projection subspaces. Results on matrices from real applications suggest that the proposed algorithm can lead to higher accuracy, especially for the singular triplets associated with the largest modulus singular values. Several practical details and key differences with other approaches are also discussed. Vasileios Kalantzis, Georgios Kollias, Shashanka Ubaru, Athanasios N. Nikolakopoulos, Lior Horesh, Kenneth L. Clarkson |
ICML | 3 |
| 2021 | Efficient scaling of dynamic graph neural networksabstractWe present distributed algorithms for training dynamic Graph Neural Networks (GNN) on large scale graphs spanning multi-node, multi-GPU systems. To the best of our knowledge, this is the first scaling study on dynamic GNN. We devise mechanisms for reducing the GPU memory usage and identify two execution time bottlenecks: CPU-GPU data transfer; and communication volume. Exploiting properties of dynamic graphs, we design a graph difference-based strategy to significantly reduce the transfer time. We develop a simple, but effective data distribution technique under which the communication volume remains fixed and linear in the input size, for any number of GPUs. Our experiments using billion-size graphs on a system of 128 GPUs shows that: (i) the distribution scheme achieves up to 30x speedup on 128 GPUs; (ii) the graph-difference technique reduces the transfer time by a factor of up to 4.1x and the overall execution time by up to 40%. Venkatesan T. Chakaravarthy, Shivmaran S. Pandian, Saurabh Raje, Yogish Sabharwal, Toyotaro Suzumura, Shashanka Ubaru |
SC | 6 |
| 2021 | Dynamic Graph Convolutional Networks Using the Tensor M-ProductabstractMany irregular domains such as social networks, financial transactions, neuron connections, and natural language constructs are represented using graph structures. In recent years, a variety of graph neural networks (GNNs) have been successfully applied for representation learning and prediction on such graphs. In many of the real-world applications, the underlying graph changes over time, however, most of the existing GNNs are inadequate for handling such dynamic graphs. In this paper we propose a novel technique for learning embeddings of dynamic graphs using a tensor algebra framework. Our method extends the popular graph convolutional network (GCN) for learning representations of dynamic graphs using the recently proposed tensor M-product technique. Theoretical results presented establish a connection between the proposed tensor approach and spectral convolution of tensors. The proposed method TM-GCN is consistent with the Message Passing Neural Network (MPNN) framework, accounting for both spatial and temporal message passing. Numerical experiments on real-world datasets demonstrate the performance of the proposed method for edge classification and link prediction tasks on dynamic graphs. We also consider an application related to the COVID-19 pandemic, and show how our method can be used for early detection of infected individuals from contact tracing data. Osman Asif Malik, Shashanka Ubaru, Lior Horesh, Misha Elena Kilmer, Haim Avron |
SDM | 2 |
| 2021 | Dynamic graph and polynomial chaos based models for contact tracing data analysis and optimal testing prescription
Shashanka Ubaru, Lior Horesh, Guy Cohen |
J. Biomed. Informatics | 1 |
| 2020 | Multilabel Classification by Hierarchical Partitioning and Data-dependent GroupingabstractIn 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 (unknown) hierarchy. In this paper we exploit the sparsity of label vectors and the hierarchical structure to embed them in low-dimensional space using label groupings. Consequently, we solve the classification problem in a much lower dimensional space and then obtain labels in the original space using an appropriately defined lifting. Our method builds on the work of (Ubaru & Mazumdar, 2017), where the idea of group testing was also explored for multilabel classification. We first present a novel data-dependent grouping approach, where we use a group construction based on a low-rank Nonnegative Matrix Factorization (NMF) of the label matrix of training instances. The construction also allows us, using recent results, to develop a fast prediction algorithm that has a \emph{logarithmic runtime in the number of labels}. We then present a hierarchical partitioning approach that exploits the label hierarchy in large-scale problems to divide the large label space into smaller sub-problems, which can then be solved independently via the grouping approach. Numerical results on many benchmark datasets illustrate that, compared to other popular methods, our proposed methods achieve comparable accuracy with significantly lower computational costs. Shashanka Ubaru, Sanjeeb Dash, Arya Mazumdar, Oktay Günlük |
NeurIPS | 1 |
| 2019 | Spectrum-adapted Polynomial Approximation for Matrix FunctionsabstractWe propose and investigate two new methods to approximate f(A)b for large, sparse, Hermitian matrices A. Computations of this form play an important role in numerous signal processing and machine learning tasks. The main idea behind both methods is to first estimate the spectral density of A, and then find polynomials of a fixed order that better approximate the function f on areas of the spectrum with a higher density of eigenvalues. Compared to state-of-the-art methods such as the Lanczos method and truncated Chebyshev expansion, the proposed methods tend to provide more accurate approximations of f(A)b at lower polynomial orders, and for matrices A with a large number of distinct interior eigenvalues and a small spectral width. David I. Shuman, Shashanka Ubaru, Yousef Saad |
ICASSP | 3 |
| 2019 | Find the dimension that counts: Fast dimension estimation and Krylov PCAabstractHigh dimensional data and systems with many degrees of freedom are often characterized by covariance matrices. In this paper, we consider the problem of simultaneously estimating the dimension of the principal (dominant) subspace of these covariance matrices and obtaining an approximation to the subspace. This problem arises in the popular principal component analysis (PCA), and in many applications of machine learning, data analysis, signal and image processing, and others. We first present a novel method for estimating the dimension of the principal subspace. We then show how this method can be coupled with a Krylov subspace method to simultaneously estimate the dimension and obtain an approximation to the subspace. The dimension estimation is achieved at no additional cost. The proposed method operates on a model selection framework, where the novel selection criterion is derived based on random matrix perturbation theory ideas. We present theoretical analyses which (a) show that the proposed method achieves strong consistency (i.e., yields optimal solution as the number of data-points n → ∞), and (b) analyze conditions for exact dimension estimation in the finite n case. Using recent results, we show that our algorithm also yields near optimal PCA. The proposed method avoids forming the sample covariance matrix (associated with the data) explicitly and computing the complete eigen-decomposition. Therefore, the method is inexpensive, which is particularly advantageous in modern data applications where the covariance matrices can be very large. Numerical experiments illustrate the performance of the proposed method in various applications. Shashanka Ubaru, Abd-Krim Seghouane, Yousef Saad |
SDM | 1 |
| 2018 | Spectrum Approximation Beyond Fast Matrix Multiplication: Algorithms and HardnessabstractUnderstanding the singular value spectrum of a matrix $A \in \mathbb{R}^{n \times n}$ is a fundamental task in countless applications. In matrix multiplication time, it is possible to perform a full SVD and directly compute the singular values $σ_1,...,σ_n$. However, little is known about algorithms that break this runtime barrier. Using tools from stochastic trace estimation, polynomial approximation, and fast system solvers, we show how to efficiently isolate different ranges of $A$'s spectrum and approximate the number of singular values in these ranges. We thus effectively compute a histogram of the spectrum, which can stand in for the true singular values in many applications. We use this primitive to give the first algorithms for approximating a wide class of symmetric matrix norms in faster than matrix multiplication time. For example, we give a $(1 + ε)$ approximation algorithm for the Schatten-$1$ norm (the nuclear norm) running in just $\tilde O((nnz(A)n^{1/3} + n^2)ε^{-3})$ time for $A$ with uniform row sparsity or $\tilde O(n^{2.18} ε^{-3})$ time for dense matrices. The runtime scales smoothly for general Schatten-$p$ norms, notably becoming $\tilde O (p \cdot nnz(A) ε^{-3})$ for any $p \ge 2$. At the same time, we show that the complexity of spectrum approximation is inherently tied to fast matrix multiplication in the small $ε$ regime. We prove that achieving milder $ε$ dependencies in our algorithms would imply faster than matrix multiplication time triangle detection for general graphs. This further implies that highly accurate algorithms running in subcubic time yield subcubic time matrix multiplication. As an application of our bounds, we show that precisely computing all effective resistances in a graph in less than matrix multiplication time is likely difficult, barring a major algorithmic breakthrough. Cameron Musco, Praneeth Netrapalli, Aaron Sidford, Shashanka Ubaru, David P. Woodruff |
ITCS | 4 |
| 2017 | Multilabel Classification with Group Testing and CodesabstractIn recent years, the multiclass and mutlilabel classification problems we encounter in many applications have very large ($10^3$–$10^6$) number of classes. However, each instance belongs to only one or few classes, i.e., the label vectors are sparse. In this work, we propose a novel approach based on group testing to solve such large multilabel classification problems with sparse label vectors. We describe various group testing constructions, and advocate the use of concatenated Reed Solomon codes and unbalanced bipartite expander graphs for extreme classification problems. The proposed approach has several advantages theoretically and practically over existing popular methods. Our method operates on the binary alphabet and can utilize the well-established binary classifiers for learning. The error correction capabilities of the codes are leveraged for the first time in the learning problem to correct prediction errors. Even if a linearly growing number of classifiers mis-classify, these errors are fully corrected. We establish Hamming loss error bounds for the approach. More importantly, our method utilizes a simple prediction algorithm and does not require matrix inversion or solving optimization problems making the algorithm very inexpensive. Numerical experiments with various datasets illustrate the superior performance of our method. Shashanka Ubaru, Arya Mazumdar |
ICML | 1 |
| 2017 | UoI-NMF Cluster: A Robust Nonnegative Matrix Factorization Algorithm for Improved Parts-Based Decomposition and Reconstruction of Noisy DataabstractWith the ever growing collection of large volumes of scientific data, development of interpretable machine learning tools to analyze such data is becoming more important. However, robust, interpretable machine learning tools are lacking, threatening extraction of scientific insight and discovery. Nonnegative Matrix Factorization (NMF) algorithms decompose an m × n nonnegative data matrix A into a k × n basis matrix H and an m × k weight matrix W, such that A ≈ WH, where k is the desired rank. In this paper, we present a novel two stage algorithm, UoI-NMFclusterfor NMF, which is based on three innovations: (i) completely separate bases learning from weight estimation, (ii) learn bases by clustering NMF results across bootstrap resamples of the data, and (iii) use the recently introduced Union of Intersections (UoI) framework to estimate ultra-sparse weights that maximize data reconstruction accuracy. We deploy our algorithm on various synthetic and scientific data to illustrate its performance, with a focus on neuroscience data. Compared to other NMF algorithms, UoI-NMFclusteryields: a) more accurate parts-based decompositions of noisy data, b) a sparse and accurate weight matrix, and c) high accuracy reconstructions of the de-noised data. Together, these improvements enhance the performance and interpretability of NMF application to noisy data, and suggest similar approaches may benefit other matrix decomposition algorithms. Shashanka Ubaru, Kesheng Wu, Kristofer E. Bouchard |
ICMLA | 1 |
| 2017 | Union of Intersections (UoI) for Interpretable Data Driven Discovery and PredictionabstractThe 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 and predictive. We introduce the Union of Intersections (UoI) method, a flexible, modular, and scalable framework for enhanced model selection and estimation. The method performs model selection and model estimation through intersection and union operations, respectively. We show that UoI can satisfy the bi-criteria of low-variance and nearly unbiased estimation of a small number of interpretable features, while maintaining high-quality prediction accuracy. We perform extensive numerical investigation to evaluate a UoI algorithm ($UoI_{Lasso}$) on synthetic and real data. In doing so, we demonstrate the extraction of interpretable functional networks from human electrophysiology recordings as well as the accurate prediction of phenotypes from genotype-phenotype data with reduced features. We also show (with the $UoI_{L1Logistic}$ and $UoI_{CUR}$ variants of the basic framework) improved prediction parsimony for classification and matrix factorization on several benchmark biomedical data sets. These results suggest that methods based on UoI framework could improve interpretation and prediction in data-driven discovery across scientific fields. Kristofer E. Bouchard, Alejandro F. Bujan, Fred (Farbod) Roosta, Shashanka Ubaru, Prabhat, Antoine Snijders, Jian-Hua Mao, Edward F. Chang, Michael W. Mahoney, Sharmodeep Bhattacharyya |
NIPS | 4 |
| 2017 | Improving the Incoherence of a Learned Dictionary via Rank ShrinkageabstractThis letter considers the problem of dictionary learning for sparse signal representation whose atoms have low mutual coherence. To learn such dictionaries, at each step, we first update the dictionary using the method of optimal directions (MOD) and then apply a dictionary rank shrinkage step to decrease its mutual coherence. In the rank shrinkage step, we first compute a rank 1 decomposition of the column-normalized least squares estimate of the dictionary obtained from the MOD step. We then shrink the rank of this learned dictionary by transforming the problem of reducing the rank to a nonnegative garrotte estimation problem and solving it using a path-wise coordinate descent approach. We establish theoretical results that show that the rank shrinkage step included will reduce the coherence of the dictionary, which is further validated by experimental results. Numerical experiments illustrating the performance of the proposed algorithm in comparison to various other well-known dictionary learning algorithms are also presented. Shashanka Ubaru, Abd-Krim Seghouane, Yousef Saad |
Neural Comput. | 1 |
| 2017 | Fast Estimation of Approximate Matrix Ranks Using Spectral DensitiesabstractMany machine learning and data-related applications require the knowledge of approximate ranks of large data matrices at hand. This letter presents two computationally inexpensive techniques to estimate the approximate ranks of such matrices. These techniques exploit approximate spectral densities, popular in physics, which are probability density distributions that measure the likelihood of finding eigenvalues of the matrix at a given point on the real line. Integrating the spectral density over an interval gives the eigenvalue count of the matrix in that interval. Therefore, the rank can be approximated by integrating the spectral density over a carefully selected interval. Two different approaches are discussed to estimate the approximate rank, one based on Chebyshev polynomials and the other based on the Lanczos algorithm. In order to obtain the appropriate interval, it is necessary to locate a gap between the eigenvalues that correspond to noise and the relevant eigenvalues that contribute to the matrix rank. A method for locating this gap and selecting the interval of integration is proposed based on the plot of the spectral density. Numerical experiments illustrate the performance of these techniques on matrices from typical applications. Shashanka Ubaru, Yousef Saad, Abd-Krim Seghouane |
Neural Comput. | 1 |
| 2017 | Low Rank Approximation and Decomposition of Large Matrices Using Error Correcting CodesabstractLow rank approximation is an important tool used in many applications of signal processing and machine learning. Recently, randomized sketching algorithms were proposed to effectively construct low rank approximations and obtain approximate singular value decompositions of large matrices. Similar ideas were used to solve least squares regression problems. In this paper, we show how matrices from error correcting codes can be used to find such low rank approximations and matrix decompositions, and extend the framework to linear least squares regression problems. The benefits of using these code matrices are the following. First, they are easy to generate and they reduce randomness significantly. Second, code matrices, with mild restrictions, satisfy the subspace embedding property, and have a better chance of preserving the geometry of an large subspace of vectors. Third, for parallel and distributed applications, code matrices have significant advantages over structured random matrices and Gaussian random matrices. Fourth, unlike Fourier or Hadamard transform matrices, which require sampling O(k log k) columns for a rank-k approximation, the log factor is not necessary for certain types of code matrices. In particular, (1 + ϵ) optimal Frobenius norm error can be achieved for a rank-k approximation with O(k/ϵ) samples. Fifth, fast multiplication is possible with structured code matrices, so fast approximations can be achieved for general dense input matrices. Sixth, for least squares regression problem min ∥Ax - b∥2where A ∈ Rn×d, the (1 + E) relative error approximation can be achieved with O(d/ϵ) samples, with high probability, when certain code matrices are used. Shashanka Ubaru, Arya Mazumdar, Yousef Saad |
IEEE Trans. Inf. Theory | 1 |
| 2016 | Fast methods for estimating the Numerical rank of large matricesabstractWe present two computationally inexpensive techniques for estimating the numerical rank of a matrix, combining powerful tools from computational linear algebra. These techniques exploit three key ingredients. The first is to approximate the projector on the non-null invariant subspace of the matrix by using a polynomial filter. Two types of filters are discussed, one based on Hermite interpolation and the other based on Chebyshev expansions. The second ingredient employs stochastic trace estimators to compute the rank of this wanted eigen-projector, which yields the desired rank of the matrix. In order to obtain a good filter, it is necessary to detect a gap between the eigenvalues that correspond to noise and the relevant eigenvalues that correspond to the non-null invariant subspace. The third ingredient of the proposed approaches exploits the idea of spectral density, popular in physics, and the Lanczos spectroscopic method to locate this gap. Shashanka Ubaru, Yousef Saad |
ICML | 1 |
| 2016 | Group testing schemes from low-weight codewords of BCH codesabstractDespite a large volume of research in group testing, explicit small-size group testing schemes are still difficult to construct, and the parameters of known combinatorial schemes are limited by the constraints of the problem. Relaxing the worst-case identification requirements to probabilistic localization of defectives enables one to expand the range of parameters, and yet the small-size practical constructions are sparse. Motivated by this question, we perform an experimental study of almost disjunct matrices constructed from low-weight codewords of binary BCH codes, and evaluate their performance in nonadaptive group testing. We observe that identification of defectives is much more stable in these schemes compared to the schemes constructed from random binary matrices. We derive an estimate of the error probability of identification in the constructed schemes which provides a partial explanation of their performance. Shashanka Ubaru, Arya Mazumdar, Alexander Barg |
ISIT | 1 |
| 2015 | Low Rank Approximation using Error Correcting Coding MatricesabstractLow-rank matrix approximation is an integral component of tools such as principal component analysis (PCA), as well as is an important instrument used in applications like web search models, text mining and computer vision, e.g., face recognition. Recently, randomized algorithms were proposed to effectively construct low rank approximations of large matrices. In this paper, we show how matrices from error correcting codes can be used to find such low rank approximations. The benefits of using these code matrices are the following: (i) They are easy to generate and they reduce randomness significantly. (ii) Code matrices have low coherence and have a better chance of preserving the geometry of an entire subspace of vectors; (iii) Unlike Fourier transforms or Hadamard matrices, which require sampling O(k\log k) columns for a rank-k approximation, the log factor is not necessary in the case of code matrices. (iv) Under certain conditions, the approximation errors can be better and the singular values obtained can be more accurate, than those obtained using Gaussian random matrices and other structured random matrices. Shashanka Ubaru, Arya Mazumdar, Yousef Saad |
ICML | 1 |