Revision fc7e5a6dae3d8c1365e99ac5d64d8f696496e69f authored by vdutor on 11 October 2019, 17:54:54 UTC, committed by vdutor on 11 October 2019, 17:54:54 UTC
1 parent c23c6db
test_variational.py
# Copyright 2016 the GPflow authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import numpy as np
import pytest
from numpy.testing import assert_allclose
import gpflow
from gpflow.kernels import SquaredExponential
from gpflow.likelihoods import Gaussian
from tests.reference import ref_rbf_kernel
rng = np.random.RandomState(1)
# ------------------------------------------
# Helpers
# ------------------------------------------
def univariate_log_marginal_likelihood(y, K, noise_var):
return (-0.5 * y * y / (K + noise_var) - 0.5 * np.log(K + noise_var) -
0.5 * np.log(np.pi * 2.))
def univariate_posterior(y, K, noise_var):
mean = K * y / (K + noise_var)
variance = K - K / (K + noise_var)
return mean, variance
def univariate_prior_KL(meanA, meanB, varA, varB):
# KL[ qA | qB ] = E_{qA} \log [qA / qB] where qA and qB are univariate normal distributions.
return (0.5 * (np.log(varB) - np.log(varA) - 1. + varA / varB +
(meanB - meanA) * (meanB - meanA) / varB))
def multivariate_prior_KL(meanA, covA, meanB, covB):
# KL[ qA | qB ] = E_{qA} \log [qA / qB] where qA and aB are
# K dimensional multivariate normal distributions.
# Analytically tractable and equal to...
# 0.5 * (Tr(covB^{-1} covA) + (meanB - meanA)^T covB^{-1} (meanB - meanA)
# - K + log(det(covB)) - log (det(covA)))
K = covA.shape[0]
traceTerm = 0.5 * np.trace(np.linalg.solve(covB, covA))
delta = meanB - meanA
mahalanobisTerm = 0.5 * np.dot(delta.T, np.linalg.solve(covB, delta))
constantTerm = -0.5 * K
priorLogDeterminantTerm = 0.5 * np.linalg.slogdet(covB)[1]
variationalLogDeterminantTerm = -0.5 * np.linalg.slogdet(covA)[1]
return (traceTerm + mahalanobisTerm + constantTerm +
priorLogDeterminantTerm + variationalLogDeterminantTerm)
# ------------------------------------------
# Data classes: storing constants
# ------------------------------------------
class Datum:
num_latent = 1
y_data = 2.
X = np.atleast_2d(np.array([0.]))
Y = np.atleast_2d(np.array([y_data]))
Z = X.copy()
zero_mean = 0.
K = 1.
noise_var = 0.5
posterior_mean, posterior_var = univariate_posterior(y=y_data,
K=K,
noise_var=noise_var)
posterior_std = np.sqrt(posterior_var)
class MultiDatum:
dim = 3
num_latent = 1
Y = rng.randn(dim, 1)
X = rng.randn(dim, 1)
Z = X.copy()
noise_var = 0.5
signal_var = 1.5
ls = 1.7
q_mean = rng.randn(dim, num_latent)
q_sqrt_diag = rng.rand(dim, num_latent)
q_sqrt_full = np.tril(rng.rand(dim, dim))
def test_refrence_implementation_consistency():
q_mean = rng.rand(1, 1)
q_cov = rng.rand(1, 1)
p_mean = rng.rand(1, 1)
p_cov = rng.rand(1, 1)
multivariate_KL = multivariate_prior_KL(q_mean, p_mean, q_cov, p_cov)
univariate_KL = univariate_prior_KL(q_mean.reshape(-1), p_mean.reshape(-1),
q_cov.reshape(-1), p_cov.reshape(-1))
assert_allclose(univariate_KL - multivariate_KL, 0, atol=4)
@pytest.mark.parametrize('diag', [True, False])
@pytest.mark.parametrize('whiten', [True, False])
def test_variational_univariate_prior_KL(diag, whiten):
reference_kl = univariate_prior_KL(Datum.posterior_mean, Datum.zero_mean,
Datum.posterior_var, Datum.K)
q_mu = np.ones((1, Datum.num_latent)) * Datum.posterior_mean
ones = np.ones((1, Datum.num_latent)) if diag else np.ones(
(1, 1, Datum.num_latent))
q_sqrt = ones * Datum.posterior_std
model = gpflow.models.SVGP(kernel=SquaredExponential(variance=Datum.K),
likelihood=Gaussian(),
inducing_variable=Datum.Z,
num_latent=Datum.num_latent,
q_diag=diag,
whiten=whiten,
q_mu=q_mu,
q_sqrt=q_sqrt)
test_prior_KL = model.prior_kl()
assert_allclose(reference_kl - test_prior_KL, 0, atol=4)
@pytest.mark.parametrize('diag', [True, False])
@pytest.mark.parametrize('whiten', [True, False])
def test_variational_univariate_log_likelihood(diag, whiten):
# reference marginal likelihood estimate
reference_log_marginal_likelihood = univariate_log_marginal_likelihood(
y=Datum.y_data, K=Datum.K, noise_var=Datum.noise_var)
q_mu = np.ones((1, Datum.num_latent)) * Datum.posterior_mean
ones = np.ones((1, Datum.num_latent)) if diag else np.ones(
(1, 1, Datum.num_latent))
q_sqrt = ones * Datum.posterior_std
model = gpflow.models.SVGP(kernel=SquaredExponential(variance=Datum.K),
likelihood=Gaussian(),
inducing_variable=Datum.Z,
num_latent=Datum.num_latent,
q_diag=diag,
whiten=whiten,
q_mu=q_mu,
q_sqrt=q_sqrt)
model_likelihood = model.log_likelihood(X=Datum.X, Y=Datum.Y).numpy()
assert_allclose(model_likelihood - reference_log_marginal_likelihood,
0,
atol=4)
@pytest.mark.parametrize('diag', [True, False])
@pytest.mark.parametrize('whiten', [True, False])
def test_variational_univariate_conditionals(diag, whiten):
q_mu = np.ones((1, Datum.num_latent)) * Datum.posterior_mean
ones = np.ones((1, Datum.num_latent)) if diag else np.ones(
(1, 1, Datum.num_latent))
q_sqrt = ones * Datum.posterior_std
model = gpflow.models.SVGP(kernel=SquaredExponential(variance=Datum.K),
likelihood=Gaussian(),
inducing_variable=Datum.Z,
num_latent=Datum.num_latent,
q_diag=diag,
whiten=whiten,
q_mu=q_mu,
q_sqrt=q_sqrt)
fmean_func, fvar_func = gpflow.conditionals.conditional(
Datum.X,
Datum.Z,
model.kernel,
model.q_mu,
q_sqrt=model.q_sqrt,
white=whiten)
mean_value, var_value = fmean_func[0, 0], fvar_func[0, 0]
assert_allclose(mean_value - Datum.posterior_mean, 0, atol=4)
assert_allclose(var_value - Datum.posterior_var, 0, atol=4)
@pytest.mark.parametrize('whiten', [True, False])
def test_variational_multivariate_prior_KL_full_q(whiten):
cov_q = MultiDatum.q_sqrt_full @ MultiDatum.q_sqrt_full.T
mean_prior = np.zeros((MultiDatum.dim, 1))
cov_prior = np.eye(MultiDatum.dim) if whiten else ref_rbf_kernel(
MultiDatum.X, MultiDatum.ls, MultiDatum.signal_var)
reference_kl = multivariate_prior_KL(MultiDatum.q_mean, cov_q, mean_prior,
cov_prior)
q_sqrt = MultiDatum.q_sqrt_full[None, :, :]
model = gpflow.models.SVGP(kernel=SquaredExponential(variance=MultiDatum.signal_var,
lengthscale=MultiDatum.ls),
likelihood=Gaussian(MultiDatum.noise_var),
inducing_variable=MultiDatum.Z,
num_latent=MultiDatum.num_latent,
q_diag=False,
whiten=whiten,
q_mu=MultiDatum.q_mean,
q_sqrt=q_sqrt)
test_prior_kl = model.prior_kl()
assert_allclose(reference_kl - test_prior_kl, 0, atol=4)
Computing file changes ...