Alibaba Group Holding Limited
OBTAINING DYNAMIC EMBEDDING VECTORS OF NODES IN RELATIONSHIP GRAPHS

Abstract:

Implementations of this disclosure provide for obtaining dynamic embedding vectors of nodes in relationship graphs. An example method includes determining N neighboring nodes of a first node of a plurality of nodes; obtaining respective input embedding vectors of the first node and the N neighboring nodes, the input embedding vector of each node being determined based on a respective static embedding vector and a respective positional embedding vector of the node; inputting the respective input embedding vectors of the first node and the N neighboring nodes into a pre-trained embedding model that includes one or more sequentially connected computing blocks, each computing block including a corresponding self-attention layer that outputs N+1 output vectors corresponding to N+1 input vectors; and receiving respective dynamic embedding vectors of the first node and the N neighboring nodes output by the pre-trained embedding model.