Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Towards Interpretable Sparse Graph Representation Learning with Laplacian Pooling
(Submitted on 28 May 2019 (v1), last revised 2 Apr 2020 (this version, v4))
Abstract: Recent work in graph neural networks (GNNs) has led to improvements in molecular activity and property prediction tasks. Unfortunately, GNNs often fail to capture the relative importance of interactions between molecular substructures, in part due to the absence of efficient intermediate pooling steps. To address these issues, we propose LaPool (Laplacian Pooling), a novel, data-driven, and interpretable hierarchical graph pooling method that takes into account both node features and graph structure to improve molecular representation. We benchmark LaPool on molecular graph prediction and understanding tasks and show that it outperforms recent GNNs. Interestingly, LaPool also remains competitive on non-molecular tasks. Both quantitative and qualitative assessments are done to demonstrate LaPool's improved interpretability and highlight its potential benefits in drug design. Finally, we demonstrate LaPool's utility for the generation of valid and novel molecules by incorporating it into an adversarial autoencoder.
Submission history
From: Dominique Beaini [view email][v1] Tue, 28 May 2019 02:38:16 GMT (1698kb,D)
[v2] Thu, 30 Jan 2020 19:41:13 GMT (3343kb,D)
[v3] Thu, 19 Mar 2020 13:10:47 GMT (1752kb,D)
[v4] Thu, 2 Apr 2020 14:01:17 GMT (1752kb,D)
Link back to: arXiv, form interface, contact.