← Search

Aravind Gollakota

8 accepted papers

2025

Provable Uncertainty Decomposition via Higher-Order Calibration

ICLR 2025spotlight

We give a principled method for decomposing the predictive uncertainty of a model into aleatoric and epistemic components with explicit semantics relating them to the real-world data distribution. While many works in the literature have proposed such decompositions, they lack the type of formal guar…

Cited by 1SourcePDFScholar
2024

An Efficient Tester-Learner for Halfspaces

ICLR 2024poster

We give the first efficient algorithm for learning halfspaces in the testable learning model recently defined by Rubinfeld and Vasilyan [2022]. In this model, a learner certifies that the accuracy of its output hypothesis is near optimal whenever the training set passes an associated test, and train…

Cited by 14SourcePDFScholar
2023

Agnostically Learning Single-Index Models using Omnipredictors

NeurIPS 2023poster

We give the first result for agnostically learning Single-Index Models (SIMs) with arbitrary monotone and Lipschitz activations. All prior work either held only in the realizable setting or required the activation to be known. Moreover, we only require the marginal to have bounded second moments, wh…

Cited by 15SourcePDFScholar
2023

Ambient Diffusion: Learning Clean Distributions from Corrupted Data

NeurIPS 2023poster

We present the first diffusion-based framework that can learn an unknown distribution using only highly-corrupted samples. This problem arises in scientific applications where access to uncorrupted samples is impossible or expensive to acquire. Another benefit of our approach is the ability to train…

2023

Tester-Learners for Halfspaces: Universal Algorithms

NeurIPS 2023oral

We give the first tester-learner for halfspaces that succeeds universally over a wide class of structured distributions. Our universal tester-learner runs in fully polynomial time and has the following guarantee: the learner achieves error $O(\mathrm{opt}) + \epsilon$ on any labeled distribution tha…

Cited by 17SourcePDFScholar
2022

Hardness of Noise-Free Learning for Two-Hidden-Layer Neural Networks

NeurIPS 2022accept

We give superpolynomial statistical query (SQ) lower bounds for learning two-hidden-layer ReLU networks with respect to Gaussian inputs in the standard (noise-free) model. No general SQ lower bounds were known for learning ReLU networks of any depth in this setting: previous SQ lower bounds held onl…

Cited by 38SourcePDFScholar
2020

Superpolynomial Lower Bounds for Learning One-Layer Neural Networks using Gradient Descent

ICML 2020poster

We give the first superpolynomial lower bounds for learning one-layer neural networks with respect to the Gaussian distribution for a broad class of algorithms. In the regression setting, we prove that gradient descent run on any classifier with respect to square loss will fail to achieve small test…

Cited by 88SourcePDFScholar