Added timing and realised mdot can be faster as its almost always a diagonal matrix its multiplying with

This commit is contained in:
Alan Saul 2013-04-05 17:56:02 +01:00
parent 4a14a82dfb
commit 31d8faecf8
2 changed files with 21 additions and 13 deletions

View file

@ -8,11 +8,12 @@ from coxGP.python.likelihoods.likelihood_function import student_t
def timing(): def timing():
real_var = 0.1 real_var = 0.1
times = 1000 times = 1
deg_free = 10 deg_free = 10
real_sd = np.sqrt(real_var) real_sd = np.sqrt(real_var)
the_is = np.zeros(times) the_is = np.zeros(times)
X = np.linspace(0.0, 10.0, 30)[:, None] X = np.linspace(0.0, 10.0, 500)[:, None]
for a in xrange(times): for a in xrange(times):
Y = np.sin(X) + np.random.randn(*X.shape)*real_var Y = np.sin(X) + np.random.randn(*X.shape)*real_var
Yc = Y.copy() Yc = Y.copy()
@ -21,6 +22,8 @@ def timing():
Yc[25] += 10 Yc[25] += 10
Yc[23] += 10 Yc[23] += 10
Yc[24] += 10 Yc[24] += 10
Yc[300] += 10
Yc[400] += 10000
edited_real_sd = real_sd edited_real_sd = real_sd
kernel1 = GPy.kern.rbf(X.shape[1]) kernel1 = GPy.kern.rbf(X.shape[1])
@ -33,9 +36,9 @@ def timing():
m.optimize() m.optimize()
the_is[a] = m.likelihood.i the_is[a] = m.likelihood.i
import ipdb; ipdb.set_trace() ### XXX BREAKPOINT
print the_is print the_is
print np.mean(the_is) print np.mean(the_is)
import ipdb; ipdb.set_trace() ### XXX BREAKPOINT
def student_t_approx(): def student_t_approx():

View file

@ -128,7 +128,9 @@ class Laplace(likelihood):
:K: Covariance matrix :K: Covariance matrix
""" """
self.K = K.copy() self.K = K.copy()
self.Ki, _, _, log_Kdet = pdinv(K) print "Inverting K"
#self.Ki, _, _, log_Kdet = pdinv(K)
print "K inverted, optimising"
if self.rasm: if self.rasm:
self.f_hat = self.rasm_mode(K) self.f_hat = self.rasm_mode(K)
else: else:
@ -196,6 +198,7 @@ class Laplace(likelihood):
""" """
#W is diagnoal so its sqrt is just the sqrt of the diagonal elements #W is diagnoal so its sqrt is just the sqrt of the diagonal elements
W_12 = np.sqrt(W) W_12 = np.sqrt(W)
import ipdb; ipdb.set_trace() ### XXX BREAKPOINT
B = np.eye(K.shape[0]) + mdot(W_12, K, W_12) B = np.eye(K.shape[0]) + mdot(W_12, K, W_12)
L = jitchol(B) L = jitchol(B)
return (B, L, W_12) return (B, L, W_12)
@ -205,9 +208,7 @@ class Laplace(likelihood):
:K: Covariance matrix :K: Covariance matrix
:returns: f_mode :returns: f_mode
""" """
self.K = K.copy()
f = np.zeros((self.N, 1)) f = np.zeros((self.N, 1))
(self.Ki, _, _, self.log_Kdet) = pdinv(K)
LOG_K_CONST = -(0.5 * self.log_Kdet) LOG_K_CONST = -(0.5 * self.log_Kdet)
#FIXME: Can we get rid of this horrible reshaping? #FIXME: Can we get rid of this horrible reshaping?
@ -227,7 +228,7 @@ class Laplace(likelihood):
f_hat = sp.optimize.fmin_ncg(obj, f, fprime=obj_grad, fhess=obj_hess) f_hat = sp.optimize.fmin_ncg(obj, f, fprime=obj_grad, fhess=obj_hess)
return f_hat[:, None] return f_hat[:, None]
def rasm_mode(self, K, MAX_ITER=5000000000000000, MAX_RESTART=30): def rasm_mode(self, K, MAX_ITER=500000, MAX_RESTART=50):
""" """
Rasmussens numerically stable mode finding Rasmussens numerically stable mode finding
For nomenclature see Rasmussen & Williams 2006 For nomenclature see Rasmussen & Williams 2006
@ -249,6 +250,7 @@ class Laplace(likelihood):
rs = 0 rs = 0
i = 0 i = 0
while difference > epsilon:# and i < MAX_ITER and rs < MAX_RESTART: while difference > epsilon:# and i < MAX_ITER and rs < MAX_RESTART:
print "optimising"
f_old = f.copy() f_old = f.copy()
W = -np.diag(self.likelihood_function.link_hess(self.data, f)) W = -np.diag(self.likelihood_function.link_hess(self.data, f))
if not self.likelihood_function.log_concave: if not self.likelihood_function.log_concave:
@ -259,22 +261,25 @@ class Laplace(likelihood):
#If the likelihood is non-log-concave. We wan't to say that there is a negative variance #If the likelihood is non-log-concave. We wan't to say that there is a negative variance
#To cause the posterior to become less certain than the prior and likelihood, #To cause the posterior to become less certain than the prior and likelihood,
#This is a property only held by non-log-concave likelihoods #This is a property only held by non-log-concave likelihoods
print "Decomposing"
B, L, W_12 = self._compute_B_statistics(K, W) B, L, W_12 = self._compute_B_statistics(K, W)
print "Finding f"
W_f = np.dot(W, f) W_f = np.dot(W, f)#FIXME: Make this fast as W_12 is diagonal!
grad = self.likelihood_function.link_grad(self.data, f)[:, None] grad = self.likelihood_function.link_grad(self.data, f)[:, None]
#Find K_i_f #Find K_i_f
b = W_f + grad b = W_f + grad
#b = np.dot(W, f) + np.dot(self.Ki, f)*(1-step_size) + step_size*self.likelihood_function.link_grad(self.data, f)[:, None] #b = np.dot(W, f) + np.dot(self.Ki, f)*(1-step_size) + step_size*self.likelihood_function.link_grad(self.data, f)[:, None]
#TODO: Check L is lower #TODO: Check L is lower
solve_L = cho_solve((L, True), mdot(W_12, (K, b)))
a = b - mdot(W_12, solve_L) solve_L = cho_solve((L, True), mdot(W_12, (K, b)))#FIXME: Make this fast as W_12 is diagonal!
a = b - mdot(W_12, solve_L)#FIXME: Make this fast as W_12 is diagonal!
#f = np.dot(K, a) #f = np.dot(K, a)
#a should be equal to Ki*f now so should be able to use it #a should be equal to Ki*f now so should be able to use it
c = mdot(K, W_f) + f*(1-step_size) + step_size*np.dot(K, grad) c = mdot(K, W_f) + f*(1-step_size) + step_size*np.dot(K, grad)
solve_L = cho_solve((L, True), mdot(W_12, c)) solve_L = cho_solve((L, True), mdot(W_12, c))#FIXME: Make this fast as W_12 is diagonal!
f = c - mdot(K, W_12, solve_L) f = c - mdot(K, W_12, solve_L)#FIXME: Make this fast as W_12 is diagonal!
#K_w_f = mdot(K, (W, f)) #K_w_f = mdot(K, (W, f))
#c = step_size*mdot(K, self.likelihood_function.link_grad(self.data, f)[:, None]) - step_size*f #c = step_size*mdot(K, self.likelihood_function.link_grad(self.data, f)[:, None]) - step_size*f
@ -302,5 +307,5 @@ class Laplace(likelihood):
i += 1 i += 1
self.i = i self.i = i
print "{i} steps".format(i=i) #print "{i} steps".format(i=i)
return f return f