Classifying linear infrastructure elements using a graph neural network

Abstract

In example embodiments, improved techniques are provided for classifying elements of an infrastructure model that represents linear infrastructure (e.g., roads). The techniques may extract a set of cross sections perpendicular to a centerline of the linear infrastructure from the infrastructure model, generate a graph representation of each cross section to produce a set of graphs having nodes that represent elements and edges that represent contextual relationships, provide the set of graphs to a trained graph neural network (GNN) model, and produce therefrom class predictions for the elements. The class predictions may include one or more predicted classes for each element with a respective confidence. A best predicted class for each element may be selected and assigned to the element, thereby creating a new version of the infrastructure model. For elements that extend through multiple cross sections, the selection may involve aggregating predicted classes originating from the different graphs.