← Search

Léon Zheng

3 accepted papers

2025

Fast Inference with Kronecker-Sparse Matrices

ICML 2025poster

Kronecker-sparse (KS) matrices—whose supports are Kronecker products of identity and all-ones blocks—underpin the structure of Butterfly and Monarch matrices and offer the promise of more efficient models. However, existing GPU kernels for KS matrix multiplication suffer from high data movement cost…

2023

Self-supervised learning with rotation-invariant kernels

ICLR 2023top-25%

We introduce a regularization loss based on kernel mean embeddings with rotation-invariant kernels on the hypersphere (also known as dot-product kernels) for self-supervised learning of image representations. Besides being fully competitive with the state of the art, our method significantly reduces…