Conditional Graph Information Bottleneck for Molecular Relational Learning

by   Namkyeong Lee, et al.

Molecular relational learning, whose goal is to learn the interaction behavior between molecular pairs, got a surge of interest in molecular sciences due to its wide range of applications. Recently, graph neural networks have recently shown great success in molecular relational learning by modeling a molecule as a graph structure, and considering atom-level interactions between two molecules. Despite their success, existing molecular relational learning methods tend to overlook the nature of chemistry, i.e., a chemical compound is composed of multiple substructures such as functional groups that cause distinctive chemical reactions. In this work, we propose a novel relational learning framework, called CGIB, that predicts the interaction behavior between a pair of graphs by detecting core subgraphs therein. The main idea is, given a pair of graphs, to find a subgraph from a graph that contains the minimal sufficient information regarding the task at hand conditioned on the paired graph based on the principle of conditional graph information bottleneck. We argue that our proposed method mimics the nature of chemical reactions, i.e., the core substructure of a molecule varies depending on which other molecule it interacts with. Extensive experiments on various tasks with real-world datasets demonstrate the superiority of CGIB over state-of-the-art baselines. Our code is available at


page 1

page 2

page 3

page 4


Shift-Robust Molecular Relational Learning with Causal Substructure

Recently, molecular relational learning, whose goal is to predict the in...

Relational Self-Supervised Learning on Graphs

Over the past few years, graph representation learning (GRL) has been a ...

Atomic and Subgraph-aware Bilateral Aggregation for Molecular Representation Learning

Molecular representation learning is a crucial task in predicting molecu...

Deep Learning for Molecular Graphs with Tiered Graph Autoencoders and Graph Classification

Tiered graph autoencoders provide the architecture and mechanisms for le...

BayReL: Bayesian Relational Learning for Multi-omics Data Integration

High-throughput molecular profiling technologies have produced high-dime...

Graph neural networks to learn joint representations of disjoint molecular graphs

Graph neural networks are widely used to learn global representations of...

One Transformer Can Understand Both 2D 3D Molecular Data

Unlike vision and language data which usually has a unique format, molec...

Please sign up or login with your details

Forgot password? Click here to reset