公开(公告)号：US20240143696A1

公开(公告)日：2024-05-02

申请号：US18275737

申请日：2022-02-07

Applicant: DeepMind Technologies Limited

Inventor： Ali Taylan Cemgil ,  Krishnamurthy Dvijotham ,  Arnaud Doucet ,  Jamie Hayes

IPC: G06F17/18

CPC classification number: G06F17/18

Abstract: Methods, systems, and apparatus, including computer programs encoded on a computer storage medium, for generating one or more differentiable order statistics for a vector of scores. In one aspect, a method comprises: obtaining the vector of scores, wherein each position in the vector of scores is associated with a respective index from a set of indices; obtaining a plurality of pairs of indices; generating a respective swapping probability for each pair of indices based on the vector of scores; generating, for each pair of indices, a respective soft-swapping matrix for the pair of indices as a combination of: (i) an identity matrix, and (ii) an exchange matrix, wherein the exchange matrix is weighted in the combination by the swapping probability for the pair of indices; and generating the one or more differentiable order statistics for the vector of scores using the soft-swapping matrices.