Active 3D Shape Co-segmentation with Graph Convolutional Networks

We present a novel active learning approach for shape co-segmentation based on graph convolutional networks (GCN). The premise of our approach is to represent the collections of 3D shapes as graph-structured data, where each node in the graph corresponds to a primitive patch of an over-segmented 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.