Ax icon indicating copy to clipboard operation
Ax copied to clipboard

[GENERAL SUPPORT]: Recommendations for "in-situ" changes to optimization configurations in closed-loop environments

Open sgbaird opened this issue 3 months ago • 5 comments

Question

I imagine Meta AE team has run into this situation in practice at various points:

  • a closed-loop optimization campaign is running (for clarity, we'll say it's a hyperparameter tuning problem)
  • it is decided that hyperparameter X should also be included in the search space
  • now, there's a mismatch between how the prior data and the current data is represented

Please provide any relevant code snippet if applicable.


Code of Conduct

  • [x] I agree to follow this Ax's Code of Conduct

sgbaird avatar Sep 15 '25 14:09 sgbaird

This is AFAIK a fundamental problem for any method of experimentation, not just Bayesian Optimisation.

Unless you can somehow recover, estimate or guesstimate the values of the new hyperparamter X for each historical sample, you are strictly speaking looking at a different experimental setup, and as such are starting a new campaign with 0 samples.

In the world of Bayesian Optimisation, the closest work I am aware of to tackle this challenges is available here: https://openreview.net/forum?id=FSqIx6FO0O

jakobzeitler avatar Sep 15 '25 16:09 jakobzeitler

Nice, but just noting that I'm talking from a plumbing / architecture perspective, not a modeling perspective.

sgbaird avatar Sep 15 '25 16:09 sgbaird

Yes, this is a pretty common occurrence in practice. As @jakobzeitler indicates above, we do consider this a problem of transfer learning with heterogeneous search spaces. We do have models and logic for this but I don't think it's fully available in OSS - @saitcakmak can you confirm?

Balandat avatar Sep 16 '25 00:09 Balandat

The BoTorch models for heterogeneous search space TL is available in this repo (https://github.com/saitcakmak/heterogeneous_botl) but the Ax integration for transfer learning methods is not available in OSS.

saitcakmak avatar Sep 16 '25 14:09 saitcakmak

Right, though what about from a plumbing/API perspective, nothing model-based, and assuming it's just a different representation of the search space, etc. For example, someone wanting to change bounds on a hyperparameter or including an ingredient into the search space that they originally didn't include (i.e., zero for the rest).

sgbaird avatar Sep 16 '25 14:09 sgbaird