ard_num_dims configures a GPyTorch kernel to maintain one length scale for each specified input dimension. Setting it to the objective’s dimensionality enables automatic relevance determination (ARD): optimization can reveal which dimensions vary more strongly in the modeled function. section 3.4.3
Use the parameter to:
- Set ard_num_dims equal to the number of objective dimensions when initializing a covariance kernel. section 3.4.3
- Interpret optimized scales cautiously: a large length scale indicates lower variability and generally lower modeled relevance, while a small scale indicates greater variability. section 3.4.3
- Compare a shared-length-scale model with an ARD model by training both and inspecting their learned parameters. section 3.5
Two-dimensional objective
For a two-dimensional Ackley function, use gpytorch.kernels.RBFKernel(ard_num_dims=2). After gradient-based training, two learned scales such as tensor([[0.7175, 0.4117]]) indicate different variation across the dimensions. section 3.4.3
