Freivalds' algorithm

Freivalds' algorithm (named after Rūsiņš Mārtiņš Freivalds) is a probabilistic randomized algorithm used to verify matrix multiplication. Given three n × n matrices $A$, $B$, and $C$, a general problem is to verify whether $A\times B=C$. A naïve algorithm would compute the product $A\times B$ explicitly and compare term by term whether this product equals $C$. However, the best known matrix multiplication algorithm runs in $O(n^{2.372})$ time. Freivalds' algorithm utilizes randomization in order to reduce this time bound to $O(n^2)$ with high probability. In $O(k{n}^2)$ time the algorithm can verify a matrix product with probability of failure less than $2^{-k}$.

The algorithm

Input

Three n × n matrices $A$, $B$, and $C$.

Output

Yes, if $A\times B=C$; No, otherwise.

Procedure

1. Generate an n × 1 random 0/1 vector $\overrightarrow{r}$.
2. Compute $\overrightarrow{P}=A\times (B\overrightarrow{r})-C\overrightarrow{r}$.
3. Output "Yes" if $\overrightarrow{P}=(0,0,\dots ,0{)}^T$; "No," otherwise.

Error

If $A\times B=C$, then the algorithm always returns "Yes". If $A\times B\ne C$, then the probability that the algorithm returns "Yes" is less than or equal to one half. This is called one-sided error.

By iterating the algorithm k times and returning "Yes" only if all iterations yield "Yes", a runtime of $O(k{n}^2)$ and error probability of $\le 1/2^k$ is achieved.

Example

Suppose one wished to determine whether:

$AB=\left[\begin{array}{cc}2& 3\\ 3& 4\end{array}\right]\left[\begin{array}{cc}1& 0\\ 1& 2\end{array}\right]\stackrel{?}{=}\left[\begin{array}{cc}6& 5\\ 8& 7\end{array}\right]=C.$

A random two-element vector with entries equal to 0 or 1 is selected – say $\overrightarrow{r}=\left[\begin{array}{c}1\\ 1\end{array}\right]$ – and used to compute:

$\begin{array}{cc}A\times (B\overrightarrow{r})-C\overrightarrow{r}& =\left[\begin{array}{cc}2& 3\\ 3& 4\end{array}\right]\left(\left[\begin{array}{cc}1& 0\\ 1& 2\end{array}\right]\left[\begin{array}{c}1\\ 1\end{array}\right]\right)-\left[\begin{array}{cc}6& 5\\ 8& 7\end{array}\right]\left[\begin{array}{c}1\\ 1\end{array}\right]\\ & =\left[\begin{array}{cc}2& 3\\ 3& 4\end{array}\right]\left[\begin{array}{c}1\\ 3\end{array}\right]-\left[\begin{array}{c}11\\ 15\end{array}\right]\\ & =\left[\begin{array}{c}11\\ 15\end{array}\right]-\left[\begin{array}{c}11\\ 15\end{array}\right]\\ & =\left[\begin{array}{c}0\\ 0\end{array}\right].\end{array}$

This yields the zero vector, suggesting the possibility that AB = C. However, if in a second trial the vector $\overrightarrow{r}=\left[\begin{array}{c}1\\ 0\end{array}\right]$ is selected, the result becomes:

$A\times (B\overrightarrow{r})-C\overrightarrow{r}=\left[\begin{array}{cc}2& 3\\ 3& 4\end{array}\right]\left(\left[\begin{array}{cc}1& 0\\ 1& 2\end{array}\right]\left[\begin{array}{c}1\\ 0\end{array}\right]\right)-\left[\begin{array}{cc}6& 5\\ 8& 7\end{array}\right]\left[\begin{array}{c}1\\ 0\end{array}\right]=\left[\begin{array}{c}-1\\ -1\end{array}\right].$

The result is nonzero, proving that in fact AB ≠ C.

There are four two-element 0/1 vectors, and half of them give the zero vector in this case ($\overrightarrow{r}=\left[\begin{array}{c}0\\ 0\end{array}\right]$ and $\overrightarrow{r}=\left[\begin{array}{c}1\\ 1\end{array}\right]$), so the chance of randomly selecting these in two trials (and falsely concluding that AB=C) is 1/2² or 1/4. In the general case, the proportion of r yielding the zero vector may be less than 1/2, and a larger number of trials (such as 20) would be used, rendering the probability of error very small.

Error analysis

Let p equal the probability of error. We claim that if A × B = C, then p = 0, and if A × B ≠ C, then p ≤ 1/2.

Case A × B = C

$\begin{array}{cc}\overrightarrow{P}& =A\times (B\overrightarrow{r})-C\overrightarrow{r}\\ & =(A\times B)\overrightarrow{r}-C\overrightarrow{r}\\ & =(A\times B-C)\overrightarrow{r}\\ & =\overrightarrow{0}\end{array}$

This is regardless of the value of $\overrightarrow{r}$, since it uses only that $A\times B-C=0$. Hence the probability for error in this case is:

$\mathrm{Pr}[\overrightarrow{P}\ne 0]=0$

Case A × B ≠ C

Let $D$ such that

$\overrightarrow{P}=D\times \overrightarrow{r}=(p_1,p_2,\dots ,p_n{)}^T$

Where

$D=A\times B-C=(d_{ij})$.

Since $A\times B\ne C$, we have that some element of $D$ is nonzero. Suppose that the element $d_{ij}\ne 0$. By the definition of matrix multiplication, we have:

$p_i=\sum _{k=1}^n{d}_{ik}{r}_k=d_{i1}{r}_1+\cdots +d_{ij}{r}_j+\cdots +d_{in}{r}_n=d_{ij}{r}_j+y$.

For some constant $y$. Using Bayes' theorem, we can partition over $y$:

$\mathrm{Pr}[p_i=0]=\mathrm{Pr}[p_i=0|y=0]\cdot \mathrm{Pr}[y=0]+\mathrm{Pr}[p_i=0|y\ne 0]\cdot \mathrm{Pr}[y\ne 0]$ 1

We use that:

$\mathrm{Pr}[p_i=0|y=0]=\mathrm{Pr}[r_j=0]=\frac{1}{2}$

$\mathrm{Pr}[p_i=0|y\ne 0]=\mathrm{Pr}[r_j=1\wedge d_{ij}=-y]\le \mathrm{Pr}[r_j=1]=\frac{1}{2}$

Plugging these in the equation (1), we get:

$\begin{array}{cc}\mathrm{Pr}[p_i=0]& \le \frac{1}{2}\cdot \mathrm{Pr}[y=0]+\frac{1}{2}\cdot \mathrm{Pr}[y\ne 0]\\ & =\frac{1}{2}\cdot \mathrm{Pr}[y=0]+\frac{1}{2}\cdot (1-\mathrm{Pr}[y=0])\\ & =\frac{1}{2}\end{array}$

Therefore,

$\mathrm{Pr}[\overrightarrow{P}=0]=\mathrm{Pr}[p_1=0\wedge \dots \wedge p_i=0\wedge \dots \wedge p_n=0]\le \mathrm{Pr}[p_i=0]\le \frac{1}{2}.$

This completes the proof.

Ramifications

Simple algorithmic analysis shows that the running time of this algorithm is $O(n^2)$ (in big O notation). This beats the classical deterministic algorithm's runtime of $O(n^3)$ (or $O(n^{2.372})$ if using fast matrix multiplication). The error analysis also shows that if the algorithm is run $k$ times, an error bound of less than $1/2^k$ can be achieved, an exponentially small quantity. The algorithm is also fast in practice due to wide availability of fast implementations for matrix-vector products. Therefore, utilization of randomized algorithms can speed up a very slow deterministic algorithm.

Freivalds' algorithm frequently arises in introductions to probabilistic algorithms because of its simplicity and how it illustrates the superiority of probabilistic algorithms in practice for some problems.

See also

• Schwartz–Zippel lemma