Sub-linear RACE Sketches for Approximate Kernel Density Estimation on Streaming Data
Benjamin Coleman, Anshumali Shrivastava
Abstract
Kernel density estimation is a simple and effective method that lies at the heart of many important machine learning applications. Unfortunately, kernel methods scale poorly for large, high dimensional datasets. Approximate kernel density estimation has a prohibitively high memory and computation cost, especially in the streaming setting. Recent sampling algorithms for high dimensional densities can reduce the computation cost but cannot operate online, while streaming algorithms cannot handle high dimensional datasets due to the curse of dimensionality. We propose RACE, an efficient sketching algorithm for kernel density estimation on high-dimensional streaming data. RACE compresses a set of N high dimensional vectors into a small array of integer counters. This array is sufficient to estimate the kernel density for a large class of kernels. Our sketch is practical to implement and comes with strong theoretical guarantees. We evaluate our method on real-world highdimensional datasets and show that our sketch achieves 10x better compression compared to competing methods. CCS CONCEPTS • Theory of computation → Sketching and sampling.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 5696b435-8093-4cb8-97eb-75cbbe8104ccCited by top-tier papers12
- How to train data-efficient LLMsNoveen Sachdeva, Benjamin Coleman, Wang-Cheng Kang, Jianmo Ni et al.ICLR 2026 · 106 citations
- DESSERT: An Efficient Algorithm for Vector Set Search with Vector Set QueriesJoshua Engels, Benjamin Coleman, Vihan Lakshman, Anshumali ShrivastavaNeurIPS 2023 · 27 citations
- Sub-linear Memory Sketches for Near Neighbor Search on Streaming DataBenjamin Coleman, Richard G. Baraniuk, Anshumali ShrivastavaICML 2020 · 21 citations
- One-Pass Distribution Sketch for Measuring Data Heterogeneity in Federated LearningZichang Liu, Zhaozhuo Xu, Benjamin Coleman, Anshumali ShrivastavaNeurIPS 2023 · 19 citations
- Locality Sensitive TeachingZhaozhuo Xu, Beidi Chen, Chaojian Li, Weiyang Liu et al.NeurIPS 2021 · 18 citations
Related papers
- Fast Rotation Kernel Density Estimation over Data StreamsRunze Lei, Pinghui Wang, Rundong Li, Peng Jia et al.KDD 2021 · 9 citations
- A One-Pass Distributed and Private Sketch for Kernel Sums with Applications to Machine Learning at ScaleBenjamin Coleman, Anshumali ShrivastavaCCS 2021 · 1 citation
- Bayesian Sketches for Volume Estimation in Data StreamsFrancesco Da Dalt, Simon Scherrer, Adrian PerrigVLDB 2023 · 5 citations
- HistSketch: A Compact Data Structure for Accurate Per-Key Distribution MonitoringJintao He, Jiaqi Zhu, Qun HuangICDE 2023 · 21 citations
- Fast Private Kernel Density Estimation via Locality Sensitive QuantizationTal Wagner, Yonatan Naamad, Nina MishraICML 2023 · 11 citations
