Data-Driven Learning of Geometric Scattering Networks

Autor:	Tong, Alexander, Wenkel, Frederik, MacDonald, Kincaid, Krishnaswamy, Smita, Wolf, Guy
Rok vydání:	2020
Předmět:	Computer Science - Machine Learning Statistics - Machine Learning
Druh dokumentu:	Working Paper
Popis:	We propose a new graph neural network (GNN) module, based on relaxations of recently proposed geometric scattering transforms, which consist of a cascade of graph wavelet filters. Our learnable geometric scattering (LEGS) module enables adaptive tuning of the wavelets to encourage band-pass features to emerge in learned representations. The incorporation of our LEGS-module in GNNs enables the learning of longer-range graph relations compared to many popular GNNs, which often rely on encoding graph structure via smoothness or similarity between neighbors. Further, its wavelet priors result in simplified architectures with significantly fewer learned parameters compared to competing GNNs. We demonstrate the predictive performance of LEGS-based networks on graph classification benchmarks, as well as the descriptive quality of their learned features in biochemical graph data exploration tasks. Comment: 6 pages, 2 figures, 3 tables, Presented at IEEE MLSP 2021
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2010.02415 Zobrazit plný text záznamu View this record from Arxiv