Covariant Compositional Networks For Learning Graphs

  • 2018-01-07 06:14:55
  • Risi Kondor, Hy Truong Son, Horace Pan, Brandon Anderson, Shubhendu Trivedi
  • 19


Most existing neural networks for learning graphs address permutationinvariance by conceiving of the network as a message passing scheme, where eachnode sums the feature vectors coming from its neighbors. We argue that thisimposes a limitation on their representation power, and instead propose a newgeneral architecture for representing objects consisting of a hierarchy ofparts, which we call Covariant Compositional Networks (CCNs). Here, covariancemeans that the activation of each neuron must transform in a specific way underpermutations, similarly to steerability in CNNs. We achieve covariance bymaking each activation transform according to a tensor representation of thepermutation group, and derive the corresponding tensor aggregation rules thateach neuron must implement. Experiments show that CCNs can outperform competingmethods on standard graph learning benchmarks.


Introduction (beta)



Conclusion (beta)