Skip to content

Benchmark on real summarization data - #5

Merged
kholdrex merged 2 commits into
mainfrom
benchmarks-real-data
Oct 11, 2026
Merged

kholdrex merged 2 commits into
mainfrom
benchmarks-real-data

Conversation

@kholdrex

Copy link
Copy Markdown
Member

Replaces the synthetic Gutenberg benchmark with measurements on pinned test files from CNN/DailyMail, XSum and PubMed (2,000 pairs each, lead-sentence baselines as predictions, SHA-256-verified downloads kept outside the repository).

  • Every Rust result is checked for exact P/R/F parity with rouge-score 0.1.2; the run fails otherwise.
  • Serial and threaded (1/2/4/8) timings with and without stemming, Evaluate rouge vs SyntaxSpirits/rouge with default bootstrap aggregation, and peak RSS.
  • A like-for-like comparison with rouge-rust on the subset it supports (ROUGE-1/2/L, no stemming): its scores match rouge-score on this sample, and it is faster than this package there. rouge (pltrdy) is timed only as a reference for a different computation.
  • A feature comparison of rouge-score, rouge-score-rs, rouge-rust, rouge (pltrdy) and TorchMetrics, with sources.
  • benchmark.yml runs the same benchmark on Linux on manual dispatch only.

On an M1 Pro: 10.6–39.4× faster than rouge-score on one core, 39.7–201× with 8 threads, 2.3–63.5× through Evaluate.

The synthetic Gutenberg benchmark said little about real workloads. The
new benchmark scores lead-sentence baselines against references from
pinned CNN/DailyMail, XSum and PubMed test files (2,000 pairs each),
checks score parity exactly, and reports serial, threaded and Evaluate
timings plus a like-for-like comparison with rouge-rust, which is faster
on the subset it supports. A manual workflow collects the same numbers
on Linux.
@kholdrex
kholdrex merged commit 4a734a7 into main Oct 11, 2026
7 checks passed
@kholdrex
kholdrex deleted the benchmarks-real-data branch October 11, 2026 03:14
@kholdrex kholdrex mentioned this pull request Oct 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant