Non-autoregressive electron flow generation for reaction prediction release_eaiv7ifhy5dm5gfbfsedko2nbi

by Hangrui Bi, Hengyi Wang, Chence Shi, Jian Tang

Released as a article .

2020  

Abstract

Reaction prediction is a fundamental problem in computational chemistry. Existing approaches typically generate a chemical reaction by sampling tokens or graph edits sequentially, conditioning on previously generated outputs. These autoregressive generating methods impose an arbitrary ordering of outputs and prevent parallel decoding during inference. We devise a novel decoder that avoids such sequential generating and predicts the reaction in a Non-Autoregressive manner. Inspired by physical-chemistry insights, we represent edge edits in a molecule graph as electron flows, which can then be predicted in parallel. To capture the uncertainty of reactions, we introduce latent variables to generate multi-modal outputs. Following previous works, we evaluate our model on USPTO MIT dataset. Our model achieves both an order of magnitude lower inference latency, with state-of-the-art top-1 accuracy and comparable performance on Top-K sampling.
In text/plain format

Archived Files and Locations

application/pdf  351.4 kB
file_drnxwubdanh2bgssvcahc5yksi
arxiv.org (repository)
web.archive.org (webarchive)
Read Archived PDF
Preserved and Accessible
Type  article
Stage   submitted
Date   2020-12-16
Version   v1
Language   en ?
arXiv  2012.12124v1
Work Entity
access all versions, variants, and formats of this works (eg, pre-prints)
Catalog Record
Revision: f0dd4de5-0f8f-4166-ad19-8e5645020b98
API URL: JSON