arXiv · 2010.15552
Successive Halving Top-k Operator
Abstract
We propose a differentiable successive halving method of relaxing the top-k operator, rendering gradient-based optimization possible. The need to perform softmax iteratively on the entire vector of scores is avoided by using a tournament-style selection. As a result, a much better approximation of top-k with lower computational cost is achieved compared to the previous approach.
Explore related subjects
Keep this discovery
Michał Pietruszka, Łukasz Borchmann, Filip Graliński. 2020-10-08. Successive Halving Top-k Operator. https://arxiv.org/abs/2010.15552
Cite the original work for its findings. Save a collection to share your selection of sources.