← Search

Arya Mazumdar

29 accepted papers

2026

Fairness in Aggregation: Optimal Top-$k$ and Improved Full Ranking

ICML 2026poster

Ensuring fairness in algorithmic ranking systems is a critical challenge with significant societal implications for hiring, recommendations, web search, and data management. Standard methods for aggregating multiple preference orders into a consensus ranking may perpetuate and even amplify the lack …

Cited by 0SourceScholar
2026

On the Theory of Continual Learning with Gradient Descent for Neural Networks

ICML 2026poster

Continual learning, the ability of a model to adapt to an ongoing sequence of tasks without forgetting earlier ones, is a central goal of artificial intelligence. To better understand its underlying mechanisms, we study the limitations of continual learning in a tractable yet representative setting.…

Cited by 0SourceScholar
2025

Generalization Bound of Gradient Flow through Training Trajectory and Data-dependent Kernel

NeurIPS 2025poster

Gradient-based optimization methods have shown remarkable empirical success, yet their theoretical generalization properties remain only partially understood. In this paper, we establish a generalization bound for gradient flow that aligns with the classical Rademacher complexity bounds for kernel m…

Cited by 0SourceScholar
2025

Optimal Transfer Learning for Missing Not-at-Random Matrix Completion

ICML 2025poster

We study transfer learning for matrix completion in a Missing Not-at-Random (MNAR) setting that is motivated by biological problems. The target matrix $Q$ has entire rows and columns missing, making estimation impossible without side information. To address this, we use a noisy and incomplete sourc…

Cited by 0SourcePDFScholar
2025

Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methods

ICLR 2025poster

In this paper, we study the data-dependent convergence and generalization behavior of gradient methods for neural networks with smooth activation. Our first result is a novel bound on the excess risk of deep networks trained by the logistic loss via an alogirthmic stability analysis. Compared to pre…

Cited by 0SourcePDFScholar
2024

Transfer Learning for Latent Variable Network Models

NeurIPS 2024poster

We study transfer learning for estimation in latent variable network models. In our setting, the conditional edge probability matrices given the latent variables are represented by $P$ for the source and $Q$ for the target. We wish to estimate $Q$ given two kinds of data: (1) edge data from a subgra…

Cited by 2SourcePDFScholar
2021

Support Recovery of Sparse Signals from a Mixture of Linear Measurements

NeurIPS 2021poster

Recovery of support of a sparse vector from simple measurements is a widely studied problem, considered under the frameworks of compressed sensing, 1-bit compressed sensing, and more general single index models. We consider generalizations of this problem: mixtures of linear regressions, and mixture…

Cited by 13SourcePDFScholar
2021

vqSGD: Vector Quantized Stochastic Gradient Descent

AISTATS 2021poster

In this work, we present a family of vector quantization schemes vqSGD (Vector-Quantized Stochastic Gradient Descent) that provide an asymptotic reduction in the communication cost with convergence guarantees in first-order distributed optimization. In the process we derive the following fundamental…

Cited by 74SourcePDFScholar
2020

Distributed Newton Can Communicate Less and Resist Byzantine Workers

NeurIPS 2020poster

We develop a distributed second order optimization algorithm that is communication-efficient as well as robust against Byzantine failures of the worker machines. We propose an iterative approximate Newton-type algorithm, where the worker machines communicate \emph{only once} per iteration with the c…

Cited by 44SourcePDFScholar
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…

2020

Recovery of sparse linear classifiers from mixture of responses

NeurIPS 2020poster

In the problem of learning a mixture of linear classifiers, the aim is to learn a collection of hyperplanes from a sequence of binary responses. Each response is a result of querying with a vector and indicates the side of a randomly chosen hyperplane from the collection the query vector belong to.…

Cited by 14SourcePDFScholar
2019

Sample Complexity of Learning Mixture of Sparse Linear Regressions

NeurIPS 2019poster

In the problem of learning mixtures of linear regressions, the goal is to learn a col-lection of signal vectors from a sequence of (possibly noisy) linear measurements,where each measurement is evaluated on an unknown signal drawn uniformly fromthis collection. This setting is quite expressive and h…

Cited by 21SourcePDFScholar
2019

Superset Technique for Approximate Recovery in One-Bit Compressed Sensing

NeurIPS 2019poster

One-bit compressed sensing (1bCS) is a method of signal acquisition under extreme measurement quantization that gives important insights on the limits of signal compression and analog-to-digital conversion. The setting is also equivalent to the problem of learning a sparse hyperplane-classifier. In…