Videos

Xavier Bresson - Graph MLP-Mixer for Molecular Analysis - IPAM at UCLA

Presenter
January 25, 2023
Abstract
Recorded 25 January 2023. Xavier Bresson of the National University of Singapore presents "Graph MLP-Mixer for Molecular Analysis" at IPAM's Learning and Emergence in Molecular Systems Workshop. Abstract: Graph Neural Networks (GNNs) have shown great potential in the field of graph representation learning. Standard GNNs define a local message-passing mechanism which propagates information over the whole graph domain by stacking multiple layers. This paradigm suffers from two major limitations, over-squashing and poor long-range dependencies, that can be solved using global attention but significantly increases the computational cost to quadratic complexity. In this work, we propose an alternative approach to overcome these structural limitations by leveraging the ViT/MLP-Mixer architectures introduced in computer vision. We introduce a new class of GNNs, called Graph MLP-Mixer, that holds three key properties. First, they capture long-range dependency as demonstrated on the long-range LRGB datasets and mitigate the over-squashing issue on the TreeNeighbour dataset. Second, they offer memory and speed efficiency, surpassing related techniques. Third, they show high expressivity in terms of graph isomorphism as they can distinguish at least 3-WL isomorphic graphs. As a result, this novel architecture provides significantly better results over standard message-passing GNNs for molecular datasets. Learn more online at: http://www.ipam.ucla.edu/programs/workshops/learning-and-emergence-in-molecular-systems/