Learning Geometric Invariant Features for Classification of Vector Polygons with Graph Message-passing Neural Network

Zexian Huang,Kourosh Khoshelham,Martin Tomko

2024-07-05

Abstract:Geometric shape classification of vector polygons remains a non-trivial learning task in spatial analysis. Previous studies mainly focus on devising deep learning approaches for representation learning of rasterized vector polygons, whereas the study of discrete representations of polygons and subsequent deep learning approaches have not been fully investigated. In this study, we investigate a graph representation of vector polygons and propose a novel graph message-passing neural network (PolyMP) to learn the geometric-invariant features for shape classification of polygons. Through extensive experiments, we show that the graph representation of polygons combined with a permutation-invariant graph message-passing neural network achieves highly robust performances on benchmark datasets (i.e., synthetic glyph and real-world building footprint datasets) as compared to baseline methods. We demonstrate that the proposed graph-based PolyMP network enables the learning of expressive geometric features invariant to geometric transformations of polygons (i.e., translation, rotation, scaling and shearing) and is robust to trivial vertex removals of polygons. We further show the strong generalizability of PolyMP, which enables generalizing the learned geometric features from the synthetic glyph polygons to the real-world building footprints.

Computer Vision and Pattern Recognition,Machine Learning

What problem does this paper attempt to address?

The problem that this paper attempts to solve is the challenge of invariant feature learning for vector polygons in geometric shape classification. Specifically, the research aims to develop a method that can learn geometric features invariant to geometric transformations (such as rotation, scaling, and shearing) from vector polygons. Traditional methods mainly focus on rasterizing vector polygons and then using deep - learning techniques for feature representation learning, while the research on discretely represented polygons and their subsequent deep - learning methods is still insufficient. In this paper, this problem is solved by introducing graph representation and proposing a new graph message - passing neural network (PolyMP). The PolyMP network can learn the geometric features of polygons while maintaining geometric transformation invariance and is robust to small changes in polygon vertices. In addition, the research also verifies the generalization ability of PolyMP on different datasets, including synthetic character datasets and real - world building contour datasets.