Reaction Prediction via Interaction Modeling of Symmetric Difference Shingle Sets
Runhan Shi, Letian Chen, Gufeng Yu, Yang Yang
Abstract
Chemical reaction prediction remains a fundamental challenge in organic chemistry, where existing machine learning models face two critical limitations: sensitivity to input permutations (molecule/atom orderings) and inadequate modeling of substructural interactions governing reactivity. These shortcomings lead to inconsistent predictions and poor generalization to real-world scenarios. To address these challenges, we propose ReaDISH, a novel reaction prediction model that learns permutation-invariant representations while incorporating interaction-aware features. It introduces two innovations: (1) symmetric difference shingle encoding, which extends the differential reaction fingerprint (DRFP) by representing shingles as continuous high-dimensional embeddings, capturing structural changes while eliminating order sensitivity; and (2) geometry-structure interaction attention, a mechanism that models intra- and inter-molecular interactions at the shingle level. Extensive experiments demonstrate that ReaDISH improves reaction prediction performance across diverse benchmarks. It shows enhanced robustness with an average improvement of 8.76% on R under permutation perturbations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on4
- SE(3)-Transformers: 3D Roto-Translation Equivariant Attention NetworksFabian Fuchs, Daniel E. Worrall, Volker Fischer, Max WellingNeurIPS 2020 · 1,025 citations
- Uni-Mol: A Universal 3D Molecular Representation Learning FrameworkGengmo Zhou, Zhifeng Gao, Qiankun Ding, Hang Zheng et al.ICLR 2023 · 254 citations
- Geometric Transformer with Interatomic Positional EncodingYusong Wang, Shaoning Li, Tong Wang, Bin Shao et al.NeurIPS 2023 · 25 citations
- Uncertainty-Aware Yield Prediction with Multimodal Molecular FeaturesJiayuan Chen, Kehan Guo, Zhen Liu, Olexandr Isayev et al.AAAI 2024 · 13 citations
Related papers
- Chemical-Reaction-Aware Molecule Representation LearningHongwei Wang, Weijiang Li, Xiaomeng Jin, Kyunghyun Cho et al.ICLR 2022 · 79 citations
- Non-Autoregressive Electron Redistribution Modeling for Reaction PredictionHangrui Bi, Hengyi Wang, Chence Shi, Connor W. Coley et al.ICML 2021 · 24 citations
- SubgDiff: A Subgraph Diffusion Model to Improve Molecular Representation LearningJiying Zhang, Zijing Liu, Yu Wang, Bin Feng et al.NeurIPS 2024 · 11 citations
- Learning Substructure Invariance for Out-of-Distribution Molecular RepresentationsNianzu Yang, Kaipeng Zeng, Qitian Wu, Xiaosong Jia et al.NeurIPS 2022 · 133 citations
- RetroOOD: Understanding Out-of-Distribution Generalization in Retrosynthesis PredictionYemin Yu, Luotian Yuan, Ying Wei, Hanyu Gao et al.AAAI 2024 · 5 citations
