矩阵求导简介

来源:https://zhuanlan.zhihu.com/p/24709748
作者:长躯鬼侠
虽然说学过微积分理论上就能推导出矩阵求导的法则,但是矩阵求导的确是很麻烦,先不提张量相关,简单的来组合一下就很多可能:
image.png

向量对标量求导 (Vector-by-scalar)

向量 矩阵求导简介 - 图2 ,对标量 矩阵求导简介 - 图3 求导,一般写为:
矩阵求导简介 - 图4
常见例子是位移对时间求导, 速度对时间求导。

标量对向量求导(Scalar-by-vector)

标量 y 对 向量 矩阵求导简介 - 图5 求导写作:
矩阵求导简介 - 图6
这里可能会让人疑惑,为什么上面是列向量而这里则是行向量。
这是矩阵求导可能会比较麻烦的第二个地方,可能有不同的布局方式:

  • 分子布局(Numerator Layout):分子不变,分母转置
  • 分母布局(Denominator Layout):分母不变,分子转置

当然还有混合布局。
用分子布局因为 wikipedia 里就是这样做的,而且有些只能有分子布局表示。
这里可能会分子布局可能会有点让人不习惯,因为比如梯度,我们很经常会把它看成列向量:
矩阵求导简介 - 图7
但实质上有些地方也把梯度看成/写成行向量,重要的是比如你在使用的时候根据自己所默认的认为是行向量还是列向量的不同在做不同的操作的时候稍加注意就ok了,比如下面这个式子列向量无所谓,因为是点乘,但是如果看成矩阵之间的乘法当然我们就需要转置作为列向量的梯度了:
矩阵求导简介 - 图8

向量对向量求导(Vector-by-vector)

向量 矩阵求导简介 - 图9 对向量 矩阵求导简介 - 图10 求导,同样分子布局:
矩阵求导简介 - 图11
正好是雅克比矩阵:
矩阵求导简介 - 图12
这是使用分子布局的好处,否则就是雅克比的转置了。
这样的好处是写下这个式子也很自然:
矩阵求导简介 - 图13

矩阵对标量求导 (Matrix-by-scalar)

矩阵求导简介 - 图14

标量对矩阵求导(Scalar-by-matrix)

矩阵求导简介 - 图15

例子

之所以说矩阵求导本质上还是对单个变量求导,来看一个例子,比如:
矩阵求导简介 - 图16
先来计算 矩阵求导简介 - 图17 :
矩阵求导简介 - 图18
矩阵求导简介 - 图19, 利用上面的向量对向量求导,使用雅克比矩阵:
矩阵求导简介 - 图20
同时也可以尝试使用分母布局,求出来对应为 矩阵求导简介 - 图21

计算

wikipedia 上的表格可以用来速查。除此之外还有一个网站可以用来计算矩阵求导:
Matrix Calculuswww.matrixcalculus.org
试一下:
矩阵求导简介 - 图22
image.png

还可以导出为 Python code 或者 Latex , cool,再试一个:
矩阵求导简介 - 图24
image.png

增加条件, 矩阵求导简介 - 图26为对称矩阵:
image.png
looks good, cool!

参考:

矩阵求导的技术,在统计学、控制论、机器学习等领域有广泛的应用。鉴于我看过的一些资料或言之不详、或繁乱无绪,本文来做个科普,分作两篇,上篇讲标量对矩阵的求导术,下篇讲矩阵对矩阵的求导术。本文使用小写字母x表示标量,粗体小写字母矩阵求导简介 - 图28表示(列)向量,大写字母X表示矩阵。

首先来琢磨一下定义,标量f对矩阵X的导数,定义为矩阵求导简介 - 图29,即f对X逐元素求导排成与X尺寸相同的矩阵。然而,这个定义在计算中并不好用,实用上的原因是对函数较复杂的情形难以逐元素求导;哲理上的原因是逐元素求导破坏了整体性。试想,为何要将f看做矩阵X而不是各元素矩阵求导简介 - 图30的函数呢?答案是用矩阵运算更整洁。所以在求导时不宜拆开矩阵,而是要找一个从整体出发的算法。

为此,我们来回顾,一元微积分中的导数(标量对标量的导数)与微分有联系:矩阵求导简介 - 图31;多元微积分中的梯度(标量对向量的导数)也与微分有联系:矩阵求导简介 - 图32,这里第一个等号是全微分公式,第二个等号表达了梯度与微分的联系:全微分矩阵求导简介 - 图33是梯度向量矩阵求导简介 - 图34(n×1)与微分向量矩阵求导简介 - 图35(n×1)的内积;受此启发,我们将矩阵导数与微分建立联系:矩阵求导简介 - 图36。其中tr代表迹(trace)是方阵对角线元素之和,满足性质:对尺寸相同的矩阵A,B,矩阵求导简介 - 图37,即矩阵求导简介 - 图38是矩阵A,B的内积。与梯度相似,这里第一个等号是全微分公式,第二个等号表达了矩阵导数与微分的联系:全微分矩阵求导简介 - 图39是导数矩阵求导简介 - 图40(m×n)与微分矩阵矩阵求导简介 - 图41(m×n)的内积。

然后来建立运算法则。回想遇到较复杂的一元函数如矩阵求导简介 - 图42,我们是如何求导的呢?通常不是从定义开始求极限,而是先建立了初等函数求导和四则运算、复合等法则,再来运用这些法则。故而,我们来创立常用的矩阵微分的运算法则:

  1. 加减法:矩阵求导简介 - 图43;矩阵乘法:矩阵求导简介 - 图44;转置:矩阵求导简介 - 图45;迹:矩阵求导简介 - 图46
  2. 逆:矩阵求导简介 - 图47。此式可在矩阵求导简介 - 图48两侧求微分来证明。
  3. 行列式:矩阵求导简介 - 图49,其中矩阵求导简介 - 图50表示X的伴随矩阵,在X可逆时又可以写作矩阵求导简介 - 图51。此式可用Laplace展开来证明,详见张贤达《矩阵分析与应用》第279页。
  4. 逐元素乘法:矩阵求导简介 - 图52矩阵求导简介 - 图53表示尺寸相同的矩阵X,Y逐元素相乘。
  5. 逐元素函数:矩阵求导简介 - 图54矩阵求导简介 - 图55是逐元素标量函数运算, 矩阵求导简介 - 图56是逐元素求导数。例如矩阵求导简介 - 图57

我们试图利用矩阵导数与微分的联系矩阵求导简介 - 图58,在求出左侧的微分矩阵求导简介 - 图59后,该如何写成右侧的形式并得到导数呢?这需要一些迹技巧(trace trick):

  1. 标量套上迹:矩阵求导简介 - 图60
  2. 转置:矩阵求导简介 - 图61
  3. 线性:矩阵求导简介 - 图62
  4. 矩阵乘法交换:矩阵求导简介 - 图63,其中矩阵求导简介 - 图64矩阵求导简介 - 图65尺寸相同。两侧都等于矩阵求导简介 - 图66
  5. 矩阵乘法/逐元素乘法交换:矩阵求导简介 - 图67,其中矩阵求导简介 - 图68尺寸相同。两侧都等于矩阵求导简介 - 图69

观察一下可以断言,若标量函数f是矩阵X经加减乘法、逆、行列式、逐元素函数等运算构成,则使用相应的运算法则对f求微分,再使用迹技巧给df套上迹并将其它项交换至dX左侧,对照导数与微分的联系矩阵求导简介 - 图70,即能得到导数。
特别地,若矩阵退化为向量,对照导数与微分的联系矩阵求导简介 - 图71,即能得到导数。

在建立法则的最后,来谈一谈复合:假设已求得矩阵求导简介 - 图72,而Y是X的函数,如何求矩阵求导简介 - 图73呢?在微积分中有标量求导的链式法则矩阵求导简介 - 图74,但这里我们不能随意沿用标量的链式法则,因为矩阵对矩阵的导数矩阵求导简介 - 图75截至目前仍是未定义的。于是我们继续追本溯源,链式法则是从何而来?源头仍然是微分。我们直接从微分入手建立复合法则:先写出矩阵求导简介 - 图76,再将dY用dX表示出来代入,并使用迹技巧将其他项交换至dX左侧,即可得到矩阵求导简介 - 图77
最常见的情形是矩阵求导简介 - 图78,此时 矩阵求导简介 - 图79 ,可得到矩阵求导简介 - 图80。注意这里矩阵求导简介 - 图81,由于矩阵求导简介 - 图82是常量,矩阵求导简介 - 图83,以及我们使用矩阵乘法交换的迹技巧交换了矩阵求导简介 - 图84矩阵求导简介 - 图85

接下来演示一些算例。特别提醒要依据已经建立的运算法则来计算,不能随意套用微积分中标量导数的结论,比如认为AX对X的导数为A,这是没有根据、意义不明的。
例1:矩阵求导简介 - 图86,求矩阵求导简介 - 图87。其中矩阵求导简介 - 图88矩阵求导简介 - 图89列向量,矩阵求导简介 - 图90矩阵求导简介 - 图91矩阵,矩阵求导简介 - 图92矩阵求导简介 - 图93列向量,矩阵求导简介 - 图94是标量。
解:先使用矩阵乘法法则求微分,矩阵求导简介 - 图95,注意这里的矩阵求导简介 - 图96是常量,矩阵求导简介 - 图97。由于df是标量,它的迹等于自身,矩阵求导简介 - 图98,套上迹并做矩阵乘法交换:矩阵求导简介 - 图99,注意这里我们根据矩阵求导简介 - 图100交换了矩阵求导简介 - 图101矩阵求导简介 - 图102。对照导数与微分的联系矩阵求导简介 - 图103,得到矩阵求导简介 - 图104
注意:这里不能用矩阵求导简介 - 图105,导数与矩阵乘法的交换是不合法则的运算(而微分是合法的)。有些资料在计算矩阵导数时,会略过求微分这一步,这是逻辑上解释不通的。

例2:矩阵求导简介 - 图106,求矩阵求导简介 - 图107。其中矩阵求导简介 - 图108矩阵求导简介 - 图109列向量,矩阵求导简介 - 图110矩阵求导简介 - 图111矩阵,矩阵求导简介 - 图112矩阵求导简介 - 图113列向量,exp表示逐元素求指数,矩阵求导简介 - 图114是标量。
解:先使用矩阵乘法、逐元素函数法则求微分:矩阵求导简介 - 图115,再套上迹并做交换:矩阵求导简介 - 图116矩阵求导简介 - 图117,注意这里我们先根据矩阵求导简介 - 图118交换了矩阵求导简介 - 图119矩阵求导简介 - 图120矩阵求导简介 - 图121,再根据矩阵求导简介 - 图122交换了矩阵求导简介 - 图123矩阵求导简介 - 图124。对照导数与微分的联系矩阵求导简介 - 图125,得到矩阵求导简介 - 图126

例3:矩阵求导简介 - 图127,求矩阵求导简介 - 图128。其中矩阵求导简介 - 图129矩阵求导简介 - 图130矩阵,矩阵求导简介 - 图131矩阵求导简介 - 图132矩阵,矩阵求导简介 - 图133矩阵求导简介 - 图134矩阵,矩阵求导简介 - 图135矩阵求导简介 - 图136对称矩阵,矩阵求导简介 - 图137是逐元素函数,矩阵求导简介 - 图138是标量。
解:先求矩阵求导简介 - 图139,求微分,使用矩阵乘法、转置法则:矩阵求导简介 - 图140,对照导数与微分的联系,得到矩阵求导简介 - 图141,注意这里M是对称矩阵。为求矩阵求导简介 - 图142,写出矩阵求导简介 - 图143,再将dY用dX表示出来代入,并使用矩阵乘法/逐元素乘法交换:矩阵求导简介 - 图144,对照导数与微分的联系,得到矩阵求导简介 - 图145

例4【线性回归】:矩阵求导简介 - 图146, 求矩阵求导简介 - 图147的最小二乘估计,即求矩阵求导简介 - 图148的零点。其中矩阵求导简介 - 图149矩阵求导简介 - 图150列向量,矩阵求导简介 - 图151矩阵求导简介 - 图152矩阵,矩阵求导简介 - 图153矩阵求导简介 - 图154列向量,矩阵求导简介 - 图155是标量。
解:这是标量对向量的导数,不过可以把向量看做矩阵的特例。先将向量模平方改写成向量与自身的内积:矩阵求导简介 - 图156,求微分,使用矩阵乘法、转置等法则:矩阵求导简介 - 图157,注意这里矩阵求导简介 - 图158矩阵求导简介 - 图159是向量,两个向量的内积满足矩阵求导简介 - 图160。对照导数与微分的联系矩阵求导简介 - 图161,得到矩阵求导简介 - 图162矩阵求导简介 - 图163矩阵求导简介 - 图164,得到矩阵求导简介 - 图165的最小二乘估计为矩阵求导简介 - 图166

例5【方差的最大似然估计】:样本矩阵求导简介 - 图167,求方差矩阵求导简介 - 图168的最大似然估计。写成数学式是:矩阵求导简介 - 图169,求矩阵求导简介 - 图170的零点。其中矩阵求导简介 - 图171矩阵求导简介 - 图172列向量,矩阵求导简介 - 图173是样本均值,矩阵求导简介 - 图174矩阵求导简介 - 图175对称正定矩阵,矩阵求导简介 - 图176是标量,log表示自然对数。
解:首先求微分,使用矩阵乘法、行列式、逆等运算法则,第一项是矩阵求导简介 - 图177,第二项是矩阵求导简介 - 图178。再给第二项套上迹做交换:矩阵求导简介 - 图179矩阵求导简介 - 图180矩阵求导简介 - 图181,其中先交换迹与求和,然后将 矩阵求导简介 - 图182交换到左边,最后再交换迹与求和,并定义矩阵求导简介 - 图183为样本方差矩阵。得到矩阵求导简介 - 图184。对照导数与微分的联系,有矩阵求导简介 - 图185,其零点即矩阵求导简介 - 图186的最大似然估计为矩阵求导简介 - 图187

例6【多元logistic回归】:矩阵求导简介 - 图188,求矩阵求导简介 - 图189。其中矩阵求导简介 - 图190是除一个元素为1外其它元素为0的矩阵求导简介 - 图191列向量,矩阵求导简介 - 图192矩阵求导简介 - 图193矩阵,矩阵求导简介 - 图194矩阵求导简介 - 图195列向量,矩阵求导简介 - 图196是标量;log表示自然对数,矩阵求导简介 - 图197,其中矩阵求导简介 - 图198表示逐元素求指数,代表全1向量。
解1:首先将softmax函数代入并写成,这里要注意逐元素log满足等式,以及满足。求微分,使用矩阵乘法、逐元素函数等法则:。再套上迹并做交换,注意可化简,这是根据等式,故。对照导数与微分的联系,得到。
解2:定义,则,先同上求出,再利用复合法则:,得到。

最后一例留给经典的神经网络。神经网络的求导术是学术史上的重要成果,还有个专门的名字叫做BP算法,我相信如今很多人在初次推导BP算法时也会颇费一番脑筋,事实上使用矩阵求导术来推导并不复杂。为简化起见,我们推导二层神经网络的BP算法。
例7【二层神经网络】:,求和。其中是除一个元素为1外其它元素为0的的列向量,是矩阵,是矩阵,是列向量,是标量;log表示自然对数,同上,是逐元素sigmoid函数。
解:定义,,,则。在前例中已求出。使用复合法则,,使用矩阵乘法交换的迹技巧从第一项得到,从第二项得到。接下来对第二项继续使用复合法则来求,并利用矩阵乘法和逐元素乘法交换的迹技巧:,得到。为求,再用一次复合法则:,得到。
推广:样本,,其中是列向量,是列向量,其余定义同上。
解1:定义,,,则。先同上可求出。使用复合法则,,从第一项得到得到,从第二项得到,从第三项得到到。接下来对第二项继续使用复合法则,得到。为求,再用一次复合法则:,得到,。
解2:可以用矩阵来表示N个样本,以简化形式。定义,,,,注意这里使用全1向量来扩展维度。先同上求出。使用复合法则, ,从第一项得到,从第二项得到,从第三项得到到。接下来对第二项继续使用复合法则,得到。为求,再用一次复合法则:,得到,。

下篇见https://zhuanlan.zhihu.com/p/24863977

本文承接上篇 https://zhuanlan.zhihu.com/p/24709748,来讲矩阵对矩阵的求导术。使用小写字母x表示标量,粗体小写字母矩阵求导简介 - 图199表示列向量,大写字母X表示矩阵。矩阵对矩阵的求导采用了向量化的思路,常应用于二阶方法中Hessian矩阵的分析。
首先来琢磨一下定义。矩阵对矩阵的导数,需要什么样的定义?第一,矩阵F(p×q)对矩阵X(m×n)的导数应包含所有mnpq个偏导数矩阵求导简介 - 图200,从而不损失信息;第二,导数与微分有简明的联系,因为在计算导数和应用中需要这个联系;第三,导数有简明的从整体出发的算法。我们先定义向量矩阵求导简介 - 图201(p×1)对向量矩阵求导简介 - 图202(m×1)的导数矩阵求导简介 - 图203(m×p),有矩阵求导简介 - 图204;再定义矩阵的(按列优先)向量化矩阵求导简介 - 图205(mn×1),并定义矩阵F对矩阵X的导数矩阵求导简介 - 图206(mn×pq)。导数与微分有联系矩阵求导简介 - 图207。几点说明如下:

  1. 按此定义,标量f对矩阵X(m×n)的导数矩阵求导简介 - 图208是mn×1向量,与上篇的定义不兼容,不过二者容易相互转换。为避免混淆,用记号矩阵求导简介 - 图209表示上篇定义的m×n矩阵,则有矩阵求导简介 - 图210。虽然本篇的技术可以用于标量对矩阵求导这种特殊情况,但使用上篇中的技术更方便。读者可以通过上篇中的算例试验两种方法的等价转换。

  2. 标量对矩阵的二阶导数,又称Hessian矩阵,定义为矩阵求导简介 - 图211(mn×mn),是对称矩阵。对向量矩阵求导简介 - 图212或矩阵矩阵求导简介 - 图213求导都可以得到Hessian矩阵,但从矩阵矩阵求导简介 - 图214出发更方便。

  3. 矩阵求导简介 - 图215,求导时矩阵被向量化,弊端是这在一定程度破坏了矩阵的结构,会导致结果变得形式复杂;好处是多元微积分中关于梯度、Hessian矩阵的结论可以沿用过来,只需将矩阵向量化。例如优化问题中,牛顿法的更新矩阵求导简介 - 图216,满足矩阵求导简介 - 图217
  4. 在资料中,矩阵对矩阵的导数还有其它定义,比如矩阵求导简介 - 图218(mp×nq),或是矩阵求导简介 - 图219(mp×nq),它能兼容上篇中的标量对矩阵导数的定义,但微分与导数的联系(dF等于矩阵求导简介 - 图220中逐个m×n子块分别与dX做内积)不够简明,不便于计算和应用。资料[5]综述了以上定义,并批判它们是坏的定义,能配合微分运算的才是好的定义。
  5. 在资料中,有分子布局和分母布局两种定义,其中向量对向量的导数的排布有所不同。本文使用的是分母布局,机器学习和优化中的梯度矩阵采用此定义。而控制论等领域中的Jacobian矩阵采用分子布局,向量矩阵求导简介 - 图221对向量矩阵求导简介 - 图222的导数定义是矩阵求导简介 - 图223,对应地导数与微分的联系是矩阵求导简介 - 图224;同样通过向量化定义矩阵F对矩阵X的导数矩阵求导简介 - 图225,有矩阵求导简介 - 图226。两种布局下的导数互为转置,二者求微分的步骤是相同的,仅在对照导数与微分的联系时有一个转置的区别,读者可根据所在领域的习惯选定一种布局。

然后来建立运算法则。仍然要利用导数与微分的联系矩阵求导简介 - 图227,求微分的方法与上篇相同,而从微分得到导数需要一些向量化的技巧:

  1. 线性:矩阵求导简介 - 图228
  2. 矩阵乘法:矩阵求导简介 - 图229,其中矩阵求导简介 - 图230表示Kronecker积,A(m×n)与B(p×q)的Kronecker积是矩阵求导简介 - 图231(mp×nq)。此式证明见张贤达《矩阵分析与应用》第107-108页。
  3. 转置:矩阵求导简介 - 图232,A是m×n矩阵,其中矩阵求导简介 - 图233(mn×mn)是交换矩阵(commutation matrix),将按列优先的向量化变为按行优先的向量化。例如矩阵求导简介 - 图234
  4. 逐元素乘法:矩阵求导简介 - 图235,其中矩阵求导简介 - 图236(mn×mn)是用A的元素(按列优先)排成的对角阵。

观察一下可以断言,若矩阵函数F是矩阵X经加减乘法、逆、行列式、逐元素函数等运算构成,则使用相应的运算法则对F求微分,再做向量化并使用技巧将其它项交换至vec(dX)左侧,对照导数与微分的联系矩阵求导简介 - 图237,即能得到导数。
特别地,若矩阵退化为向量,对照导数与微分的联系矩阵求导简介 - 图238,即能得到导数。

再谈一谈复合:假设已求得矩阵求导简介 - 图239,而Y是X的函数,如何求矩阵求导简介 - 图240呢?从导数与微分的联系入手,矩阵求导简介 - 图241,可以推出链式法则矩阵求导简介 - 图242
和标量对矩阵的导数相比,矩阵对矩阵的导数形式更加复杂,从不同角度出发常会得到形式不同的结果。有一些Kronecker积和交换矩阵相关的恒等式,可用来做等价变形:

  1. 矩阵求导简介 - 图243
  2. 矩阵求导简介 - 图244
  3. 矩阵求导简介 - 图245。可以对矩阵求导简介 - 图246求导来证明,一方面,直接求导得到矩阵求导简介 - 图247;另一方面,引入矩阵求导简介 - 图248,有矩阵求导简介 - 图249,用链式法则得到矩阵求导简介 - 图250
  4. 矩阵求导简介 - 图251
  5. 矩阵求导简介 - 图252,A是m×n矩阵,B是p×q矩阵。可以对矩阵求导简介 - 图253做向量化来证明,一方面,矩阵求导简介 - 图254;另一方面,矩阵求导简介 - 图255

接下来演示一些算例。
例1:矩阵求导简介 - 图256,X是m×n矩阵,求。
解:先求微分:,再做向量化,使用矩阵乘法的技巧,注意在dX右侧添加单位阵:,对照导数与微分的联系得到。
特例:如果X退化为向量,即,则根据向量的导数与微分的关系,得到。

例2:,X是n×n矩阵,求和。
解:使用上篇中的技术可求得。为求,先求微分:,再做向量化,使用转置和矩阵乘法的技巧,对照导数与微分的联系,得到,注意它是对称矩阵。在是对称矩阵时,可简化为。

例3:,A是l×m矩阵,X是m×n矩阵,B是n×p矩阵,exp为逐元素函数,求。
解:先求微分:,再做向量化,使用矩阵乘法的技巧:,再用逐元素乘法的技巧:,再用矩阵乘法的技巧:,对照导数与微分的联系得到。

例4【一元logistic回归】:,求和。其中是取值0或1的标量,是列向量。
解:使用上篇中的技术可求得,其中 为sigmoid函数。为求,先求微分:,其中为sigmoid函数的导数,对照导数与微分的联系,得到。
推广:样本,,求和。有两种方法,解1:先对每个样本求导,然后相加;解2:定义矩阵,向量,将写成矩阵形式,进而可以使用上篇中的技术求得。为求,先求微分,再用逐元素乘法的技巧:,对照导数与微分的联系,得到。

例5【多元logistic回归】:,求和。其中其中是除一个元素为1外其它元素为0的列向量,是矩阵,是列向量,是标量。
解:上篇中已求得。为求,先求微分:定义,,注意这里化简去掉逐元素乘法,第一项中,第二项中。定义矩阵,,做向量化并使用矩阵乘法的技巧,得到。

最后做个总结。我们发展了从整体出发的矩阵求导的技术,导数与微分的联系是计算的枢纽,标量对矩阵的导数与微分的联系是,先对f求微分,再使用迹技巧可求得导数,特别地,标量对向量的导数与微分的联系是;矩阵对矩阵的导数与微分的联系是,先对F求微分,再使用向量化的技巧可求得导数,特别地,向量对向量的导数与微分的联系是。

参考资料:

  1. 张贤达. 矩阵分析与应用. 清华大学出版社有限公司, 2004.
  2. Fackler, Paul L. “Notes on matrix calculus.” North Carolina State University(2005).
  3. Petersen, Kaare Brandt, and Michael Syskind Pedersen. “The matrix cookbook.” Technical University of Denmark 7 (2008): 15.
  4. HU, Pili. “Matrix Calculus: Derivation and Simple Application.” (2012).
  5. Magnus, Jan R., and Heinz Neudecker. “Matrix Differential Calculus with Applications in Statistics and Econometrics.” Wiley, 2019.