Skip to content

Add a GPU-PSO benchmark on BioPreDyn-bench B5 - #1953

Open
AdityaPandeyCN wants to merge 1 commit into
SciML:masterfrom
AdityaPandeyCN:pso-biopredyn-b5
Open

AdityaPandeyCN wants to merge 1 commit into
SciML:masterfrom
AdityaPandeyCN:pso-biopredyn-b5

Conversation

@AdityaPandeyCN

@AdityaPandeyCN AdityaPandeyCN commented Oct 21, 2026 •

Copy link
Copy Markdown
Contributor

Adds benchmarks/PSOBioPreDyn/pso_biopredyn_b5.jmd, a GPU-PSO version of the BioPreDyn-bench B5 parameter estimation problem (86 parameters, 26 states, 10 simulated experiments).

It compares PSOKernel and SyncPSOKernel (GPU, 50,000 particles) and SerialPSO (CPU, 64 particles) with the global optimizers from the CPU B5 benchmark, BBO adaptive DE and NLopt GN_CRS2_LM. Each optimizer gets 300 s per run and 3 runs, after one untimed warm-up. The page reports the best cost at 300 s, how many runs reach the eSS reference (J = 3072.5), and a best cost vs time plot.

The cost uses fixed-step RK4 (dt = 0.1 h) instead of Rodas5P so it can run inside GPU kernels. The page checks it against Rodas5P at tight tolerances; they agree to about 0.1%.

🤖 Generated with Claude Code

Fits the 86 parameters of BioPreDyn-bench B5 with the ParallelParticleSwarms
optimizers and compares them with the global optimizers from the CPU B5
benchmark (BBO adaptive DE, NLopt GN_CRS2_LM) on the same wall-clock budget,
300 s per run, 3 runs each. PSOKernel and SyncPSOKernel run on the GPU with
50,000 particles, SerialPSO on the CPU with 64.

The cost uses fixed-step RK4 (dt = 0.1 h) so it can run inside GPU kernels.
It is checked against Rodas5P at tight tolerances and agrees to about 0.1%.
Model, bounds, start point and data are copied from the CPU B5 benchmark.

HybridPSO is left out. Its per-particle L-BFGS phase took 2000-2900 s with
Enzyme (64 particles, limited by the tape) and ~7600 s with chunked forward
mode, against the 300 s budget.

ParallelParticleSwarms is pinned to b27e809 on my fork for the scalar-cost
best-particle reduction and chunked ForwardDiff gradients.

This branch has not been deployed

No deployments
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