SORTING ATTENTION NEURAL NETWORKS

    公开(公告)号:US20210248450A1

    公开(公告)日:2021-08-12

    申请号:US17169718

    申请日:2021-02-08

    Applicant: Google LLC

    Abstract: A system for performing a machine learning task on a network input is described. The system includes one or more computers and one or more storage devices storing instructions that, when executed by the one or more computers, cause the one or more computers to implement (i) multiple sorting networks in which each sorting network is configured to sort vector blocks in a sequence of vector blocks to generate a sorted sequence of vector blocks; and (ii) a sorting attention neural network configured to perform the machine learning task on the input sequence by executing multiple sorting attention mechanisms using the sorting networks.

Patent Agency Ranking