CPC G06N 3/044 (2023.01) [G06N 3/048 (2023.01); G06N 3/08 (2013.01)] | 10 Claims |
1. A system comprising 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 an attention neural network that is configured to receive an input sequence and to process the input sequence to generate an output, the attention neural network comprising:
an attention block configured to receive a query input, a key input, and a value input that are derived from an attention block input, the attention block comprising:
an attention neural network layer configured to:
receive an attention layer input derived from the query input, the key input, and the value input, and
apply an attention mechanism to the attention layer input to generate an attention layer output for the attention neural network layer; and
a gating neural network layer configured to apply a gating mechanism to the attention block input and the attention layer output of the attention neural network layer to generate a gated attention output.
|