Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP] CES implementation for CATKE #1

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

[WIP] CES implementation for CATKE #1

wants to merge 8 commits into from

Conversation

odunbar
Copy link

@odunbar odunbar commented Nov 22, 2024

Purpose

Runs a CES configuration on a large dataset

  • this problem isn't really a good Bayesian one, as we only have inputs & a loss function. Therefore we must estimate the noise level of the problem over the ensemble, and solve with respect to this, the noise level will dictate the posterior.

To-do

  • requires the 1GB data file catke-parameters.jld2

Content

  • script to run GP (slow) and scalar-RF (faster)-based CES algorithms. rather than emulating the forward map, here we emulate the loss function directly with a 23D->1D map.
  • plotting prior, posterior, and also before & after mlt hyperparameter training to see if it acheives benefit. (prior/untuned in grey on plots)
  • some L^2 errors also computed on a chunk of validation data (compared to an untuned fit) to assess the hyperparameter training

Current result: train with data from iterations 51:20:251, estimating noise from final 50 iterations

These are still rough-and-ready, and may be improved by further looking at the choice of kernel structure for the RF maps. Current finding increasing rank shrinks posterior spread before stabilizing (below = rank 7 kernel).
(Right: Zoom-in of Left, red bar = final EKI mean, grey bars, min/max of training data)

Grey bars indicate that some parameters: C^WwDelta, C^{hi}C, C^{hi}D, C^{hi}e, C^{lo}e, C^ec may accelerate EKI convergence and emulator sample production with more biased priors toward the respective min/max of the current training data


  • I have read and checked the items on the review checklist.

@odunbar odunbar changed the title CES implementation for CATKE [WIP] CES implementation for CATKE Nov 22, 2024
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