Skip to navigation

Benchmarks

View as Markdown

With BioNeMo GenMol, a benchmark for both performance and accuracy is built on two fragment completion tasks, motif-extension and scaffold-decoration, with 10 tests per task drawn from the SAFE-DRUGS dataset. These are derived by following tests from SAFE-GPT. For more information about these tasks, refer to Gotta be SAFE: A New Framework for Molecular Design.

For the task of motif-extension, parameters are chosen as:

  • mask_length = 17
  • temperature = 1.2
  • noise = 1.6

For the task of scaffold-decoration, parameters are chosen as:

  • mask_length = 17
  • temperature = 1.2
  • noise = 2.0

Performance

Average wall-time (in seconds) for generating 1000 molecules across 10 tests on each supported GPU.

GPUmotif-extensionscaffold-decoration
A10G3.8933.051
A1001.5981.379
L40S1.6251.458
RTX6000 Ada1.6801.423
RTX6000 Blackwell0.9090.811
H1001.0510.961
H2000.8960.804
B2000.6840.653
B3001.6760.947
GH2000.9110.901
GB10 (DGX Spark)3.9382.811
GB2000.9410.870
GB3001.4830.924

Accuracy

Accuracy is evaluated from generating 100 molecules and computing the following metrics:

  • validity: fraction of generated SMILES that are valid.
  • uniqueness: ratio of unique molecules from all valid molecules.
  • diversity: average pair-wise distances in molecular fingerprints of generated molecules.
  • novelty: average distances of molecular fingerprints from the input molecule to generated molecules.
  • quality: fraction of generated molecules with a QED score greater than 0.6 and an SA score less than 4.

Reference values measured on H100:

metricmotif-extensionscaffold-decoration
validity0.8890.995
uniqueness0.6700.756
diversity0.6740.564
novelty0.6910.624
quality0.1880.354

Values are single-run measurements and standard deviations across runs are not reported. Accuracy is broadly consistent across the validated GPU matrix, while validity stays within approximately ±2 percentage points and quality within approximately ±4 percentage points across all 13 supported SKUs.

v1 Compared to v2 on H100

The following table compares v1 and v2 measurements on the same H100 reference SKU, using the same benchmark harness and methodology in both runs.

Metricv1 H100v2 H100Δ
Performance (wall-time, seconds)
motif-extension1.8241.051−42.4%
scaffold-decoration0.9720.961−1.1%
Accuracy — motif-extension
validity0.9150.889−0.026
uniqueness0.6710.670−0.001
diversity0.6040.674+0.070
novelty0.6840.691+0.007
quality0.2730.188−0.085
Accuracy — scaffold-decoration
validity0.9670.995+0.028
uniqueness0.7630.756−0.007
diversity0.5550.564+0.009
novelty0.6550.624−0.031
quality0.3460.354+0.008

Motif-extension quality is sensitive to hyperparameter selection (temperature, noise, and mask-length). The values reported above use the defaults listed in the test parameters at the top of this page; customer applications can achieve different accuracy/quality trade-offs by tuning these parameters.