Matrix calculus

We can generalise the high school calculus, which is about scalar functions of a scalar argument, in various ways, to handle matrix-valued functions or matrix-valued arguments. One could generalise this further, by to full tensor calculus. But it happens that specifically matrix/vector operations are at a useful point of complexity for lots of algorithms, kind of a MVP. (I usually want this for higher order gradient descent.)

I will mention two convenient and popular formalisms for doing that here. In practice a mix of each is often useful.

Matrix differentials

🏗 I need to return to this and tidy it up with some examples.

A special case of tensor calculus that happens to be handy for some common cases; where the rank of the argument and value of the function is not too big. Fun pain point: agreeing upon layout of derivatives, numerator vs denominator.

If our problem is nice, this often gets us a low-fuss, compact, tidy solution even for some surprising cases where it seems that more general tensors would be more natural. (for which, see below)

Indexed tensor calculus

Keywords: Ricci calculus, Einstein summation notation, index notation, subscript notation.

If we crack open a tensor textbook we get a lot of guff about general relativity and tensor fields and such, which is all very nice but not germane to typical machine learning applications. We want to start with the immediately-needed thing, which is some tidy notation conventions for dealing with multilinear operations without too many squiggles in our notation.

Soeren Laue, Matthias Mitterreiter, Joachim Giesen and Jens K. Mueller have been popularising such an approach recently. In their paper [LaueComputing2018], they argue that derivation of matrix differential results results can be greatly simplified with Ricci calculus, and P.S. it often induces faster code.

They have a website. which showcases this trick to do symbolic matrix calculus online (not the accelerated code generation bit.)

Here’s tasty readings on relevant bits of tensor machinery.

Alexander Graham. 1981. Kronecker Products and Matrix Calculus: With Applications. Horwood.

Darrell A. Turkington. 2001. Matrix Calculus Zero-One Matrices. Cambridge University Press.

Gene H. Golub, and Charles F. van Loan. 1983. Matrix Computations. JHU Press.

George A. F. Seber. 2007. A Matrix Handbook for Statisticians. Wiley.

Giles, Mike B. 2008. “Collected Matrix Derivative Results for Forward and Reverse Mode Algorithmic Differentiation.” In Advances in Automatic Differentiation, edited by Christian H. Bischof, H. Martin Bücker, Paul Hovland, Uwe Naumann, and Jean Utke, 64:35–44. Berlin, Heidelberg: Springer Berlin Heidelberg.

Laue, Soeren, Matthias Mitterreiter, and Joachim Giesen. 2018. “Computing Higher Order Derivatives of Matrix and Tensor Expressions.” In Advances in Neural Information Processing Systems 31, edited by S. Bengio, H. Wallach, H. Larochelle, K. Grauman, N. Cesa-Bianchi, and R. Garnett, 2750–9. Curran Associates, Inc.

Magnus, Jan R., and Heinz Neudecker. 1999. Matrix Differential Calculus with Applications in Statistics and Econometrics. Rev. ed. New York: John Wiley.

Minka, Thomas P. 2000. “Old and New Matrix Algebra Useful for Statistics.”

Parr, Terence, and Jeremy Howard. 2018. “The Matrix Calculus You Need for Deep Learning,” February.

Petersen, Kaare Brandt, and Michael Syskind Pedersen. 2012. “The Matrix Cookbook.”

Willi-Hans Steeb. 2006. Problems and Solutions in Introductory and Advanced Matrix Calculus. World Scientific.