Active 3D Shape Co-segmentation with Graph Convolutional Networks

We present a novel active learning approach for shape cosegmentation based on graph convolutional networks (GCNs). The premise of our approach is to represent the collections of three-dimensional shapes as graph-structured data, where each node in the graph corresponds to a primitive patch of an oversegmented shape, and is associated with a representation initialized by extracting features. Then, the GCN operates directly on the graph to update the representation of each node based on a layer-wise propagation rule, which aggregates information from its neighbors, and predicts the labels for unlabeled nodes. Additionally, we further suggest an active learning strategy that queries the most informative samples to extend the initial training samples of GCN to generate more accurate predictions of our method. Our experimental results on the Shape COSEG dataset demonstrate the effectiveness of our approach.