US 12,072,936 B2
Using graph queries to obtain results from machine learning models
Karthik Gurumoorthy Subramanya Bharathy, Sammamish, WA (US); Ankit Gupta, Rahuwas (IN); George Karypis, Bloomington, MN (US); Divyakala Vel, Redmond, WA (US); Bradley Bebee, Seattle, WA (US); Adesoji Adeshina, Vancouver (CA); and Xiang Song, Shanghai (CN)
Assigned to AMAZON TECHNOLOGIES, INC., Seattle, WA (US)
Filed by Amazon Technologies, Inc., Seattle, WA (US)
Filed on May 6, 2021, as Appl. No. 17/313,769.
Application 17/313,769 is a continuation of application No. PCT/CN2020/134603, filed on Dec. 8, 2020.
Prior Publication US 2022/0179910 A1, Jun. 9, 2022
Int. Cl. G06F 16/901 (2019.01); G06F 11/34 (2006.01); G06F 16/9032 (2019.01); G06F 16/9038 (2019.01); G06N 3/08 (2023.01); G06N 20/00 (2019.01)
CPC G06F 16/9024 (2019.01) [G06F 11/3419 (2013.01); G06F 16/90324 (2019.01); G06F 16/9038 (2019.01); G06N 3/08 (2013.01); G06N 20/00 (2019.01)] 20 Claims
OG exemplary drawing
 
1. A computer-implemented method, comprising:
obtaining, at a graph database, a graph query that identifies criteria to identify a graph data structure responsive to the graph query, wherein the graph query identifies input data to be used by a trained machine learning model to generate at least a portion of the graph data structure;
executing the trained machine learning model against the input data to obtain an inference result representing a first subset of graph data;
applying the graph query to the first subset of the graph data represented by the inference result to generate first query results corresponding to a first portion of the graph data structure that is responsive to the graph query;
applying the graph query to a second subset of the graph data stored in the graph database to generate second query results corresponding to a second portion of the graph data structure that is responsive to the graph query;
combining at least the first portion of the graph data structure generated by application of the graph query to the first subset of the graph data represented by the inference result and the second portion of the graph data structure generated by application of the graph query to the second subset of the graph data stored in the graph database to result in the graph data structure responsive to the graph query; and
providing, by the graph database the graph data structure in response to the graph query.