paper-with-me

Papers

Chefs' Random Tables: Non-Trigonometric Random Features

2022-05-30 · Valerii Likhosherstov, Krzysztof Choromanski, Avinava Dubey, Frederick Liu, Tamas Sarlos, Adrian Weller

We introduce chefs' random tables (CRTs), a new class of non-trigonometric random features (RFs) to approximate Gaussian and softmax kernels. CRTs are an alternative to standard random kitchen sink (RKS) methods, which inherently rely on the trigonometric maps. We present variants of CRTs where RFs are positive, a key requirement for applications in recent low-rank Transformers. Further variance reduction is possible by leveraging statistics which are simple to compute. One instantiation of CRTs, the optimal positive random features (OPRFs), is to our knowledge the first RF method for unbiased softmax kernel estimation with positive and bounded RFs, resulting in exponentially small tails and much lower variance than its counterparts. As we show, orthogonal random features applied in OPRFs provide additional variance reduction for any dimensionality $d$ (not only asymptotically for sufficiently large $d$, as for RKS). We test CRTs on many tasks ranging from non-parametric classification to training Transformers for text, speech and image data, obtaining new state-of-the-art results for low-rank text Transformers, while providing linear space and time complexity.

📄 PDF Abstract BibTeX arXiv:2205.15317

Code (1)

tensorflow/models/blob/master/official/nlp/modeling/layers/kernel_attention.py 공식 구현 tf

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…

Similar Papers 제목 키워드 기반

FastLSQ: Solving PDEs in One Shot via Fourier Features with Exact Analytical Derivatives

2026-02-11 · Antonin Sulc arxiv

We present FastLSQ, a framework for PDE solving and inverse problems built on trigonometric random Fourier features with exact analytical derivatives. Trigonometric features admit closed-form derivatives of any order in …

Hybrid Random Features

2021-10-08 · ICLR 2022 4 · Krzysztof Choromanski, Haoxian Chen, Han Lin, Yuanzhe Ma 외

We propose a new class of random feature methods for linearizing softmax and Gaussian kernels called hybrid random features (HRFs) that automatically adapt the quality of kernel estimation to provide most accurate approx…

Benchmarking

Dense-Exponential Random Features: Sharp Positive Estimators of the Gaussian Kernel

2023-09-21 · NeurIPS 2023 11

The problem of efficient approximation of a linear operator induced by the Gaussian or softmax kernel is often addressed using random features (RFs) which yield an unbiased approximation of the operator's result. Such op…

FAVOR#: Sharp Attention Kernel Approximations via New Classes of Positive Random Features

2023-02-01 · Valerii Likhosherstov, Krzysztof Choromanski, Avinava Dubey, Frederick Liu 외

The problem of efficient approximation of a linear operator induced by the Gaussian or softmax kernel is often addressed using random features (RFs) which yield an unbiased approximation of the operator's result. Such op…

Learning to Prove Trigonometric Identities

2022-07-14 · Zhou Liu, YuJun Li, Zhengying Liu, Lin Li 외

Automatic theorem proving with deep learning methods has attracted attentions recently. In this paper, we construct an automatic proof system for trigonometric identities. We define the normalized form of trigonometric i…

Automated Theorem ProvingImitation Learning