Skip to main content

Table 3 Wall time of the model’s inference on the USPTO MIT test set in reaction product prediction with beam search (BS) and speculative beam search (SBS). The number of generated sequences (denoted by “K”) is 5. “B” stands for batch size. The number of drafts and the draft length in the speculative beam search is different for every batch size. The average time and the standard deviation are estimated based on five runs

From: Accelerating the inference of string generation-based chemical reaction models for industrial applications

Decoding (K=5)

Time (B=1), min

Time (B=2), min

Time (B=3), min

Time (B=4), min

Beam search

297.9 ± 20.2

166.5 ± 32.0

151.8 ± 19.4

87.7 ± 13.8

SBS

140.4 ± 7.2

109.1 ± 3.6

93.3 ± 1.3

82.1 ± 0.6