Low-Rank Thinning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Carrell, Annabelle Michael, Gong, Albert, Shetty, Abhishek, Dwivedi, Raaz, Mackey, Lester
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915823769092096
author Carrell, Annabelle Michael
Gong, Albert
Shetty, Abhishek
Dwivedi, Raaz
Mackey, Lester
author_facet Carrell, Annabelle Michael
Gong, Albert
Shetty, Abhishek
Dwivedi, Raaz
Mackey, Lester
contents The goal in thinning is to summarize a dataset using a small set of representative points. Remarkably, sub-Gaussian thinning algorithms like Kernel Halving and Compress can match the quality of uniform subsampling while substantially reducing the number of summary points. However, existing guarantees cover only a restricted range of distributions and kernel-based quality measures and suffer from pessimistic dimension dependence. To address these deficiencies, we introduce a new low-rank analysis of sub-Gaussian thinning that applies to any distribution and any kernel, guaranteeing high-quality compression whenever the kernel or data matrix is approximately low-rank. To demonstrate the broad applicability of the techniques, we design practical sub-Gaussian thinning approaches that improve upon the best known guarantees for approximating attention in transformers, accelerating stochastic gradient training through reordering, and distinguishing distributions in near-linear time.
format Preprint
id arxiv_https___arxiv_org_abs_2502_12063
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Low-Rank Thinning
Carrell, Annabelle Michael
Gong, Albert
Shetty, Abhishek
Dwivedi, Raaz
Mackey, Lester
Machine Learning
Optimization and Control
Statistics Theory
Methodology
The goal in thinning is to summarize a dataset using a small set of representative points. Remarkably, sub-Gaussian thinning algorithms like Kernel Halving and Compress can match the quality of uniform subsampling while substantially reducing the number of summary points. However, existing guarantees cover only a restricted range of distributions and kernel-based quality measures and suffer from pessimistic dimension dependence. To address these deficiencies, we introduce a new low-rank analysis of sub-Gaussian thinning that applies to any distribution and any kernel, guaranteeing high-quality compression whenever the kernel or data matrix is approximately low-rank. To demonstrate the broad applicability of the techniques, we design practical sub-Gaussian thinning approaches that improve upon the best known guarantees for approximating attention in transformers, accelerating stochastic gradient training through reordering, and distinguishing distributions in near-linear time.
title Low-Rank Thinning
topic Machine Learning
Optimization and Control
Statistics Theory
Methodology
url https://arxiv.org/abs/2502.12063