Abstract: A graph with a plurality of nodes, a plurality of edges, and a plurality of node features is obtained and node representations for the node features are generated. A plurality of structure learning scores is generated based on the node representations, each structure learning score corresponding to one of the plurality of edges. A subset of the plurality of edges that identify a subgraph is selected, each edge of the subset having a structure learning score that is greater than a given threshold. The subgraph is inputted to a representation learner and an inferencing operation is performed using the representation learner based on the subgraph.