LOADING

加载过慢请开启缓存 浏览器默认开启

矩阵求导法则

创建:2026/5/30 科研 机器学习

矩阵求导的定义与常见公式。

一、矩阵求导的定义

设

X=[x11⋯x1n⋮⋮xm1⋯xmn]

共有 m×n 个变量。

那么定义:

∂f∂X=(∂f∂xij)

即

∂f∂X=[∂f∂x11⋯∂f∂x1n⋮⋮∂f∂xm1⋯∂f∂xmn]

结论

对于 f:Rm×n→R,即:输入是矩阵,输出是标量。

则:∂f∂X 与 X 形状相同。

即:X∈Rm×n,那么 ∂f∂X∈Rm×n。


AI 里常见的矩阵求导公式

1. 线性项

f=aTx

导数:

∂f∂x=af=xTa

导数:

∂f∂x=a

2. 二范数

f=xTx

导数:

∂f∂x=2x

3. 二次型

f=xTAx

导数:

∂f∂x=(A+AT)x

若 A=AT(对称),则:

∂f∂x=2Ax

4. Frobenius 范数

f=∥W∥F2

导数:

∂f∂W=2W

5. 线性变换

y=Wx

对 x 求导:

∂y∂x=W

对 W 求导(反向传播常用,其中 L=L(y)):

∂L∂W=∂L∂yxT

这是神经网络权重更新最核心的公式。

6. Trace 技巧

∂∂Xtr(AX)=AT∂∂Xtr(XTAX)=(A+AT)X

若 A 对称:

∂∂Xtr(XTAX)=2AX