条件随机场入门 条件随机场的训练

Posted 2020-08-05 ooon

本节讨论给定训练数据集估计条件随机场模型参数的问题，即条件随机场的学习问题。条件随机场模型实际上是定义在时序数据上的对数线形模型，其学习方法包括极大似然估计和正则化的极大似然估计。具体的优化实现算法有改进的迭代尺度法IIS、梯度下降法以及 L-BFGS 算法。（crf++ 采用了 L-BFGS 优化的方式，所以着重看这种训练方法即可）

L-BFGS算法

对于条件随机场模型：

\[P_w(y|x) = \frac{\exp \left \{ \sum_{k=1}^K w_kf_k(x,y)\right \}}{ \sum_y  \left \{ \exp \sum_{i=1}^n w_if_i(x,y)\right \}}\]

已知训练数据集，由此可知经验概率分布 $\widetilde{P}(X,Y)$ 可以通过极大化训练数据的对数似然函数来求模型参数，训练数据的对数似然函数为:

\[L(w) = L_{\widetilde{P}}(P_w) = \log \prod_{x,y}P_w(y|x)^{\widetilde{P}(x,y)} = \sum_{x,y}\widetilde{P}(x,y) \log P_w(y|x)\]

其中 P_w(y|x) 与 Z_w(x) 分别为：

\[P_w(y|x) = \frac{\exp \left\{ w \cdot F(y,x)\right\}}{Z_w(x)} , \ \ Z_w(x) = \sum_y \exp \left\{ w \cdot F(y,x)\right\}\]

接下来给出对数似然函数为

\begin{aligned}
L(w) &= \sum_{x,y} \widetilde{P}(x,y) \log P_w(y|x) \\
&= \sum_{x,y} \left \{ \widetilde{P}(x,y)\sum_{k=1}^Kw_kf_k(y,x)-\widetilde{P}(x,y) \log Z_w(x) \right \} \\
&= \sum_{x,y} \widetilde{P}(x,y) \sum_{k=1}^K w_kf_k(x,y) - \sum_x\widetilde{P}(x)\log\sum_y\exp\left \{ \sum_{i=1} ^nw_if_i(x,y)\right \}
\end{aligned}

对目标进行 MLE 等价于极小化以下优化目标函数：

\[\min_w f(w)  = \sum_x\widetilde{P}(x) \log \sum_y \exp \left \{  \sum_{i=1}^n w_if_i(x,y)\right \} -\sum_{x,y}\widetilde{P}(x,y)\sum_{k=1}^Kw_kf_k(x,y)\]

其梯度函数在 L-BFGS 的过程中有用到，形式如下：

\[g(w) = \sum_{x,y}\widetilde{P}(x)P_w(y|x)f(x,y)-E_{\widetilde{P}}(f)\]

这里 $E_{\widetilde{P}}$ 的计算已经在上一小节给出，总结一下变了得到求解 crf 的 L-BFGS 算法：

输入：特征函数 $f_1,f_2,…,f_n$；经验分布 $\widetilde{P}(X,Y)$；

输出：最优参数值 $\hat{w}$；最优模型 $P_{\hat{w}}(y|x)$。

(1)选定初始点 $w^{(0)}$，取 $B_0$ 为正定对称矩阵，置 $k = 0$；

(2)   计算 $g_k = g(w^{(k)})$。若 $g_k = 0$ ,则停止计算；否则转(3)

(3)由 $B_kp_k = –g_k$ 求出 $p_k$

(4)一维搜索：求 $\lambda_k$ 使得：

\[f(w^{(k)} + \lambda_kp_k) = min_{\lambda \ge 0}f(w^{(k)} + \lambda p_k)\]

(5)置 $w^{(k+1)} = w^{(k)} + \lambda_k p_k$

(6)计算 $g_{k+1} = g(w^{(k+1)})$，若 $g_k = 0$ ,则停止计算；否则，按下式求出 $B_{k+1}$:

\[B_{k+1} = B_k + \frac{y_ky_k^T}{y_k^T \delta_k} – \frac{B_k \delta_k \delta_k^TB_k}{\delta_k^TB_k\delta_k}\]

其中：\[y_k = g_{k+1}-g_k ,\delta_k = w^{(k+1)} - w^{(k)}\]

(7)置 k = k+1，转(3)。