Access ModelBridge Transforms from Acquisition Function

I am currently working on implementing an acquisition function to find contours levels; in order to do so, the acquisition function takes in a set of “integration points” from within the search space and a set of “contour levels” from within the outcome space. This works fine in pure BoTorch, since transformations to the search and outcome space are handled directly in the GP model (i.e., in the Surrogate in Ax). However, as best I can tell, Ax handles its input and outcome transformations in the ModelBridge, not in the Surrogate. As a result, the transformations (which change over each iteration) are known neither to the acquisition function nor to the surrogate model when computing acquisition function values. As a result, there is no way to transform the “integration points” and “contour levels” to match the transformations that occurred during model fitting. Is there any way to easily access these transformations inside of an AcquisitionFunction subclass? Any help would be greatly appreciated.

[akern40]

Ah ya I like challenges but I’m not a sadist. I’ll wait until that PR is complete, and I can take the time to consider some design options to propose for this issue.

[lena-kashtelyan]

@akern40, the PR is now complete if you wanted to take this on! Let us know if you have any questions or need help. I’m going to put this as “needs repro or more info” for now as we’ll be waiting to hear from you on how this goes!