diff options
Diffstat (limited to 'finale/sections/experiments.tex')
| -rw-r--r-- | finale/sections/experiments.tex | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/finale/sections/experiments.tex b/finale/sections/experiments.tex index 14c83f6..169554f 100644 --- a/finale/sections/experiments.tex +++ b/finale/sections/experiments.tex @@ -16,6 +16,12 @@ be learned ($2M$ in the product-prior in Eq.~\ref{eq:gaussianprior}). The maximum-likelihood estimator converges to an $l_\infty$-error of $.05$ for most graphs after having observed at least $100M$ distinct cascade-steps. +We therefore used the Blocks~\cite{blocks} framework, written on top of Theano +for highly efficient SGD routines with minimal memory overhead. By encoding a +cascade step as a tuple of two binary vectors, one for the infected nodes and +one for the susceptible nodes, the variational inference objective can be +written as a sum of two matrix multiplications, which Theano optimizes for +on GPU. baseline |