Dataset

A dataset is a curated collection of examples, labels, metadata, and splits used to train or evaluate models.

For chem-bio datasets, one row should usually preserve the Target-assay-label contract instead of collapsing the row into only molecule -> label.

A useful dataset note treats the dataset as an object with lineage:

where is source records, is the example-unit set, is model input, is labels or targets, is metadata, and is the split policy.

Dataset Object

PartQuestion
SourceWhere did the raw records come from, and can the source be cited publicly?
SchemaWhat fields, identifiers, units, and relationships exist?
Example unitWhat is exactly one example?
Label semanticsWhat does the target mean, and under what measurement process?
MetadataWhat context is required to interpret each example?
Split policyWhat grouping prevents leakage for the intended claim?
PreprocessingWhat transformations produce model-ready inputs?
Artifact boundaryWhich files, hashes, versions, or scripts define this dataset version?

Example Unit vs Split Unit

The row identifier is rarely enough. A dataset should state both:

If two examples share the same split unit, they should stay on the same side of train, validation, and test for that claim.

TaskExample UnitSplit Unit
molecule propertystandardized molecule or measurement rowscaffold, molecular cluster, source
target-conditioned activitymolecule-target-assay rowscaffold, protein family, assay/source
protein sequence tasksequence, domain, residue, or family recordsequence cluster or protein family
docking or pose taskreceptor-ligand pair, pose, or complexreceptor, ligand scaffold, complex, pocket family
genome sequence taskwindow, region, variant, or locuschromosome, locus, species, source batch

The split unit names the generalization claim. A random row split and a scaffold split are different scientific statements even if the model and metric are identical.

Label and Metadata Boundary

For supervised datasets, label meaning should not be inferred from a column name alone.

Label IssueRequired Context
binary activityendpoint, threshold, unit, source, missing-label policy
affinity or potencyunit, transformation, censoring, assay protocol
pose correctnessreference structure, tolerance, failed-pose handling
generated-object validityvalidity rule, invalid denominator, filtering policy
sequence annotationreference database, version, evidence level, region definition

For chem-bio rows, the label often belongs to a tuple:

where is the main entity and is context such as target, assay, pocket, source, time, construct, or measurement protocol.

Dataset Version

A dataset version is not just a file name. It should include enough information to reconstruct or audit the claim:

Version FieldExample
source snapshotpublic dataset release, paper supplement, database date
filtering ruleallowed records, excluded records, failure handling
standardizationmolecule normalization, sequence filtering, structure preparation
split rulegrouping key, validation policy, held-out claim
preprocessing artifactsvocabulary, scaler, featurizer, conformer generator, graph builder
checksums or manifestspublic-safe hashes or manifest shape
limitationscoverage gaps, bias, label noise, domain shift

Why It Matters

  • Splits, duplicate handling, and label provenance shape what a benchmark measures.
  • Molecular datasets often mix molecules, proteins, assays, and structures.
  • Clear dataset notes make leakage risks and task definitions easier to audit.
  • Dataset lineage explains how the dataset version was produced from source records.
  • Dataset cards keep model claims from drifting beyond the data’s supported scope.

Public Dataset Note Shape

Use this shape when writing about a public dataset:

SectionInclude
Purposetask family and intended use
Public sourcecitation, license, release, and to verify for missing metadata
Example unitrow/entity/pair/pose/region definition
Schemaidentifiers, fields, units, optional fields
Labelsmeaning, unit, transform, censoring, missingness, weak labels
Splitssplit unit, validation rule, leakage checks
Preprocessingtransformations fit on train only, versioned featurization
Evaluationsupported metrics and unsupported claims
Limitationssampling bias, label noise, assay/source drift, applicability domain

Do not publish private raw paths, collaborator-only datasets, internal labels, unpublished metrics, or unreleased split files. If a dataset is private, write the general dataset pattern rather than the private artifact.

Checks

  • What is one training example, and what is the target label?
  • Is the example unit different from the row identifier?
  • What split unit matches the intended generalization claim?
  • Are duplicate molecules, protein families, or near-identical structures separated?
  • Does the split match the intended generalization setting?
  • Are missing labels, censored measurements, and assay metadata handled explicitly?
  • Is the split key the same entity that defines the intended generalization claim?
  • Is label provenance kept with each row?
  • Does each row keep target, assay, unit, threshold, censoring, and source context when relevant?
  • Are preprocessing artifacts fit only on training data?
  • Can the public note support the claim without exposing private sources or unpublished results?