Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphs
  Procuring expressive molecular representations underpins AI-driven molecule
design and scientific discovery. The research to date mainly focuses on
atom-level homogeneous molecular graphs, ignoring the rich information in
subgraphs or motifs. However, it has been widely accepted that substructures
play a dominant role in the identification and determination of molecular
properties. To address such issues, we formulate heterogeneous molecular graphs
(HMGs), and introduce Molformer to exploit both molecular motifs and 3D
geometry. Specifically, we extract functional groups as motifs for small
molecules and resort to the reinforcement learning to adaptively select
quaternary amino acids as motifs for proteins. Then HMGs are constructed with
both atom-level and motif-level nodes. To better accommodate those HMGs, we
introduce a variant of Transformer named Molformer, which adopts a
heterogeneous self-attention layer to distinguish the interactions between
multi-level nodes. Besides, it is also coupled with a multi-scale mechanism to
capture local fine-grained patterns with increasing contextual scales. An
attentive farthest point sampling algorithm is also proposed to obtain the
molecular representations. We validate Molformer across a few domains including
quantum chemistry, physiology, and biophysics. Experiments show that Molformer
outperforms state-of-the-art baselines. Our work provides a promising way to
utilize informative motifs from the perspective of multi-level graph
construction.
