A General-Purpose Framework for Chemical Reaction Representation with Atomic Correspondence and Flexible Condition Adaptation

Abstract

Motivation: Organic synthesis is fundamental to the chemical industry, particularly in domains such as pharmaceutical development. While artificial intelligence offers powerful tools for modeling chemical reactions, current approaches are primarily limited to two paradigms: those that rely on hand-crafted, domain-specific features, and those that apply generic deep learning models through simplistic concatenation or aggregation of reaction components. The former often struggles to scale effectively with increasing data volumes, while the latter relies on simple input- or feature-level concatenation to combine different reaction components. Such simplistic aggregation prevents these models from directly capturing the structural transformations between reactants and products, and also makes it difficult to adapt or extend them to datasets that include non-molecular reaction conditions without modifying their model architectures. Results: This paper introduces Align-React, a novel chemical reaction representation learning framework designed for a wide range of organic reaction tasks. Our approach integrates atomic correspondence between reactants and products to discern precise molecular transformations, thereby improving the model's comprehension of molecular transformation patterns. We incorporate an adapter structure to embed reaction conditions into the representation, enhancing adaptability across varied datasets and tasks. Furthermore, a Reaction-Center-Aware attention mechanism is proposed to enable the model to focus on critical functional groups, yielding more powerful and informative representations. Evaluated across multiple downstream tasks, our model demonstrates superior performance, significantly outperforming existing chemical reaction representation learning architectures on most benchmark datasets.