Scalability and interpretability of graph neural networks for small molecules

In this thesis I examine the use of graph neural networks for prediction tasks in chemistry with an emphasis on interpretable and scalable methods. I propose a novel kernel-inspired graph neural network architecture, called a subgraph matching neural network (SMNN), which is designed to have all feature representations and weights be human interpretable. I show that this network can achieve competitive performance with common graph neural network baselines. I also show that the network is capable of learning features that allow for transfer learning to larger molecules with significantly better performance than some baselines. This provides evidence the network is learning chemically useful representations. I then propose a framework for defining graph pooling operations to improve the scalability of graph neural networks with molecule size. I empirically examine some examples of these graph pooling layers and show that they can provide a significant speed-up without hurting accuracy, and even improving accuracy in some cases. Finally an instance of the SMNN network with a pooling layer is shown to achieve state-of-the-art accuracy on the Harvard Clean Energy Project dataset.

Name thesis:degree_name

Master

Department dc:contributor.department

Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science

Grantor dc:publisher

Massachusetts Institute of Technology

Year dc:date.issued

2018

Author dc:creator

• Sangha, Manjot.

Advisor dc:contributor.advisor

• Joseph Jacobson.

• Electrical Engineering and Computer Science.

Statement dc:rights

• MIT theses are protected by copyright. They may be viewed, downloaded, or printed from this source but further reproduction or distribution in any format is prohibited without written permission.

Licence dc:rights.uri

• http://dspace.mit.edu/handle/1721.1/7582

Language dc:language.iso

eng

Handle dc:identifier.uri

https://hdl.handle.net/1721.1/121639

OAI identifier oai:identifier

oai:dspace.mit.edu:1721.1/121639