Learning the Positions in CountSketch

Published in International Conference on Learning Representations (ICLR), 2023, 2023

We show that the hash positions in CountSketch can be learned from data to improve sketch performance beyond random constructions, and provide theoretical guarantees for when learned sketches outperform random ones. This work received an oral presentation at ICLR 2023 (top 25%).