Approximate Coordinate Exchange (ACE) Algorithm
- The Approximate Coordinate Exchange (ACE) algorithm is a stochastic optimization method that sequentially optimizes individual design variables using Monte Carlo and Gaussian process surrogates.
- It iteratively builds one-dimensional surrogate models to approximate conditional expected utilities and leverages statistical tests to accept improved coordinate updates.
- ACE has demonstrated improved expected utility in diverse applications such as pharmacokinetic modeling and logistic regression, outperforming conventional pseudo‐Bayesian designs.
The Approximate Coordinate Exchange (ACE) algorithm is a general-purpose stochastic optimization scheme for constructing Bayesian optimal experimental designs when the expected utility function is analytically intractable and the design space is high-dimensional. ACE applies coordinate-exchange methodology in conjunction with Gaussian process (GP) regression to sequentially optimize each element (“coordinate”) of the design, iteratively building surrogate models and leveraging conditional expected utility evaluations via Monte Carlo approximation. The methodology is suitable for a wide variety of nonlinear and generalized linear models, utility functions, and experimental settings—substantially extending the practical scope of fully Bayesian design (Overstall et al., 2015, Overstall et al., 2017).
1. Bayesian Optimal Design and Intractability
The general Bayesian design problem seeks a design (e.g., a collection of settings for experimental runs) that maximizes the expected utility,
where is a user-specified utility function, are model parameters, are potential future data, and is the joint prior predictive for under design (Overstall et al., 2017). In practical applications, is typically analytically intractable due to the complexity or dimensionality of the integrals, stochastic utility structure, and possibly the need for nested Monte Carlo for information-type utilities. The design space is often high-dimensional; for 0 runs with 1 variables, 2 comprises 3 continuous elements.
2. Structure of the ACE Algorithm
ACE reduces the curse of dimensionality by decomposing 4 maximization into a cyclic sequence of one-dimensional optimization subproblems, each corresponding to a single coordinate 5 of the design vector. The algorithm’s key components are:
- Monte Carlo evaluation: At each coordinate update, 6 is approximated at a small grid of values along the chosen dimension by averaging 7 over 8 prior-predictive samples 9.
- Gaussian process surrogate modeling: The noisy conditional evaluations are modeled as a realization of a one-dimensional GP with mean function 0 and covariance 1. The surrogate 2 is then optimized on a dense grid to propose a new coordinate value.
- Stochastic acceptance: The move is accepted by comparing the current and candidate utilities through a two-sample t-test, controlling for Monte Carlo error.
- Iterative cycling and restarts: The process is repeated for all coordinates over 3 cycles, with optional random restarts to mitigate local mode trapping. A point-exchange phase may follow to consolidate duplicate or close runs (Overstall et al., 2015, Overstall et al., 2017).
3. Mathematical Details and Pseudocode
The algorithm can be formally structured as:
Phase I: Coordinate Exchange
- For 4 (with 5), hold all coordinates except 6 fixed.
- For a set 7 (8):
- Compute 9, where 0 replaces coordinate 1 with 2.
- Fit a GP to 3 data.
- Maximize the surrogate 4 on a fine grid to yield 5.
- Propose 6 with 7 replaced by 8.
- Compute acceptance probability 9 via a two-sample t-test between MC samples for 0 and 1. Accept with probability 2.
The process repeats for a user-specified number of cycles and independent starts. A consolidation phase (Phase II) further improves final design structure via point exchanges (Overstall et al., 2015, Overstall et al., 2017).
| Phase | Description | Output |
|---|---|---|
| Coordinate | Sequential 1-D surrogate optimization | Updated design vector |
| Point-exchange | Merges/consolidates near-duplicate runs | Replicated/improved design |
4. Gaussian Process Surrogates in ACE
At each coordinate update, the conditional expected utilities 3 are standardized:
4
A GP prior 5, typically with squared-exponential covariance,
6
is fit to these points. The emulator’s posterior predictive mean is
7
with 8 and 9 derived from the covariance function and chosen grid (Overstall et al., 2015). Surrogate prediction is computationally efficient: inversion of an 0 matrix per update, with 1 typically 10–30.
5. Computational Properties and Practical Implementation
ACE’s computational costs are dominated by Monte Carlo utility evaluations, required for surrogate fitting and acceptance steps. At each coordinate update:
- 2 conditional utilities approximated, each requiring 3 likelihood/utility calls.
- GP fit per coordinate costs 4 due to matrix inversion.
- For 5 design parameters, 6 cycles, cost scales as 7 (Overstall et al., 2015).
Practical guidelines:
- 8, 9–0.
- 1, 2 (for optional consolidation).
- 3 random restarts advised to avoid local optima.
- Convergence is monitored via trace plots of 4, with stabilization indicating sufficient cycling.
The cyclic coordinate-exchange, constrained to one-dimensional surrogates, enables ACE to scale to design spaces previously inaccessible to direct global search or brute-force Monte Carlo (Overstall et al., 2015, Overstall et al., 2017).
6. Illustrative Applications
The methodology has been demonstrated on a range of experiments, including:
- Pharmacokinetic compartment models: ACE produced unrestricted designs with 3–5% higher expected utility than Beta-DRS and pseudo-Bayesian 5-optimal competitors.
- Mixed models (hierarchical logistic regression): Fully-Bayesian SIG and NSEL designs found by ACE outperform pseudo-Bayesian 6-optimal alternatives by 20–25% for small sample sizes 7; differences diminish for larger 8 as expected.
- Model uncertainty (binomial regression for beetle mortality): Model-averaged NSEL designs concentrated runs near the posterior LD50, quickly reducing posterior variance (Overstall et al., 2015).
The ACE method is implemented in the R package acebayes, which provides end-user tools and tutorials for Bayesian design in generalized linear and nonlinear settings (Overstall et al., 2017).
7. Algorithmic Significance and Scope
ACE, by combining stochastic coordinate-wise optimization with adaptive one-dimensional GP emulation, enables computation of fully Bayesian designs for experiments with large numbers of variables, runs, and randomization restrictions. It avoids reliance on asymptotic approximations to posterior distributions or expected utilities, and is applicable to arbitrary user-specified utility functions and statistical models. Its flexibility, scalability, and practical effectiveness position it as the most general solution to high-dimensional Bayesian design-of-experiments problems to date (Overstall et al., 2015, Overstall et al., 2017).