Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspective

Michael E. Sander, Joan Puigcerver, Josip Djolonga +2

The top-k operator returns a sparse vector, where the non-zero values correspond to the k largest values of the input. Unfortunately, because it is a discontinuous function, it is…