Graph Prolongation Convolutional Networks: Explicitly Multiscale Machine
  Learning on Graphs with Applications to Modeling of Cytoskeleton
  We define a novel type of ensemble Graph Convolutional Network (GCN) model.
Using optimized linear projection operators to map between spatial scales of
graph, this ensemble model learns to aggregate information from each scale for
its final prediction. We calculate these linear projection operators as the
infima of an objective function relating the structure matrices used for each
GCN. Equipped with these projections, our model (a Graph
Prolongation-Convolutional Network) outperforms other GCN ensemble models at
predicting the potential energy of monomer subunits in a coarse-grained
mechanochemical simulation of microtubule bending. We demonstrate these
performance gains by measuring an estimate of the FLOPs spent to train each
model, as well as wall-clock time. Because our model learns at multiple scales,
it is possible to train at each scale according to a predetermined schedule of
coarse vs. fine training. We examine several such schedules adapted from the
Algebraic Multigrid (AMG) literature, and quantify the computational benefit of
each. We also compare this model to another model which features an optimized
coarsening of the input graph. Finally, we derive backpropagation rules for the
input of our network model with respect to its output, and discuss how our
method may be extended to very large graphs.
