Non-autoregressive electron flow generation for reaction prediction [article]

Hangrui Bi, Hengyi Wang, Chence Shi, Jian Tang
<span title="2021-02-05">2021</span> <i > arXiv </i> &nbsp; <span class="release-stage" >pre-print</span>
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
more &raquo; ... emistry 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.
<span class="external-identifiers"> <a target="_blank" rel="external noopener" href="https://arxiv.org/abs/2012.12124v2">arXiv:2012.12124v2</a> <a target="_blank" rel="external noopener" href="https://fatcat.wiki/release/m4kqoilwavgjhm5jysfsvmnizy">fatcat:m4kqoilwavgjhm5jysfsvmnizy</a> </span>
<a target="_blank" rel="noopener" href="https://web.archive.org/web/20201227130804/https://arxiv.org/pdf/2012.12124v1.pdf" title="fulltext PDF download [not primary version]" data-goatcounter-click="serp-fulltext" data-goatcounter-title="serp-fulltext"> <button class="ui simple right pointing dropdown compact black labeled icon button serp-button"> <i class="icon ia-icon"></i> Web Archive [PDF] <span style="color: #f43e3e;">&#10033;</span> <div class="menu fulltext-thumbnail"> <img src="https://blobs.fatcat.wiki/thumbnail/pdf/e9/7a/e97ac608b018d6d92e5470ad5ee2a86369bd86a9.180px.jpg" alt="fulltext thumbnail" loading="lazy"> </div> </button> </a> <a target="_blank" rel="external noopener" href="https://arxiv.org/abs/2012.12124v2" title="arxiv.org access"> <button class="ui compact blue labeled icon button serp-button"> <i class="file alternate outline icon"></i> arxiv.org </button> </a>