Electron flow matching for generative reaction mechanism prediction

Central to our understanding of chemical reactivity is the principle of mass conservation¹, which is fundamental for ensuring physical consistency, balancing equations and guiding reaction design. However, data-driven computational models^2-9 for tasks such as reaction product prediction rarely abide by this most basic constraint^10-13. Here we recast the problem of reaction prediction as a problem of electron redistribution using the modern deep generative framework of flow matching^14-16, explicitly conserving both mass and electrons through the bond-electron (BE) matrix representation^17,18. Our model, FlowER, overcomes limitations inherent in previous approaches by enforcing exact mass conservation, resolving hallucinatory failure modes, recovering mechanistic reaction sequences for unseen substrate scaffolds and generalizing effectively to out-of-domain reaction classes with extremely data-efficient fine-tuning. FlowER also enables downstream estimation of thermodynamic or kinetic feasibility and manifests a degree of chemical intuition in reaction prediction tasks. This inherently interpretable framework represents an important step in bridging the gap between predictive accuracy and mechanistic understanding in data-driven reaction outcome prediction.