Nianwen Xue在《Chinese Word Segmentation as Character Tagging》中将中文分词视作为序列标注问题(sequence tagging problem),由此引入监督学习算法来解决分词问题。
1. HMM
首先,我们将简要地介绍HMM(主要参考了李航老师的《统计学习方法》)。HMM包含如下的五元组:
- 状态值集合(Q={q_1, q_2, cdots, q_N}),其中(N)为可能的状态数;
- 观测值集合(V={v_1, v_2, cdots, v_M}),其中(M)为可能的观测数;
- 转移概率矩阵(A=left[ a_{ij} ight]),其中(a_{ij})表示从状态(i)转移到状态(j)的概率;
- 发射概率矩阵(在[2]中称之为观测概率矩阵)(B=left[ b_{j}(k) ight]),其中(b_{j}(k))表示在状态(j)的条件下生成观测(v_k)的概率;
- 初始状态分布(pi).
一般地,将HMM表示为模型(lambda = (A, B, pi)),状态序列为(I),对应测观测序列为(O)。对于这三个基本参数,HMM有三个基本问题:
- 概率计算问题,在模型(lambda)下观测序列(O)出现的概率;
- 学习问题,已知观测序列(O),估计模型(lambda)的参数,使得在该模型下观测序列(P(O|lambda))最大;
- 解码(decoding)问题,已知模型(lambda)与观测序列(O),求解条件概率(P(I|O))最大的状态序列(I)。
2. 中文分词
将状态值集合(Q)置为({ B, E, M, S}),分别表示词的开始、结束、中间(begin、end、middle)及字符独立成词(single);观测序列即为中文句子。比如,“今天天气不错”通过HMM求解得到状态序列“B E B E B E”,则分词结果为“今天/天气/不错”。
通过上面例子,我们发现中文分词的任务对应于解码问题:对于字符串(C={ c_1, cdots, c_n }),求解最大条件概率
其中,(t_i)表示字符(c_i)对应的状态。应如何求解状态序列呢?解决的办法便是Viterbi算法;其实,Viterbi算法本质上是一个动态规划算法,利用到了状态序列的最优路径满足这样一个特性:最优路径的子路径也一定是最优的。定义在时刻(t)状态为(i)的概率最大值为(delta_t(i)),则有递推公式:
egin{equation}
delta_{t+1}(i) = max { [delta_t(j) a_{ji}] b_i(o_{t+1}) }
label{eq:hmm}
end{equation}
其中,(o_{t+1})即为字符(c_{t+1})。
3. 开源实现
以下的源码分析基于Jieba 0.36版本。
Jieba的jieba.finalseg实现HMM中文分词。prob_start.py
定义初始状态分布(pi):
P={'B': -0.26268660809250016,
'E': -3.14e+100,
'M': -3.14e+100,
'S': -1.4652633398537678}
prob_trans.py
转移概率矩阵(A):
P={'B': {'E': -0.510825623765990, 'M': -0.916290731874155},
'E': {'B': -0.5897149736854513, 'S': -0.8085250474669937},
'M': {'E': -0.33344856811948514, 'M': -1.2603623820268226},
'S': {'B': -0.7211965654669841, 'S': -0.6658631448798212}}
prob_emit.py
定义了发射概率矩阵(B),比如,P("和"|M)
表示状态为M的情况下出现“和”这个字的概率;
P={'B': {'一': -3.6544978750449433,
'丁': -8.125041941842026,
'七': -7.817392401429855,
...}
'S': {':': -15.828865681131282,
'一': -4.92368982120877,
...}
...}
关于训练模型的生成,作者在这里有解释,来源主要有两个:标准的切分语料 + ICTCLAS切分的txt小说。还有一个大家可能会疑惑的问题,为什么Jieba中的概率矩阵中出现了负数?不急,我们先来看看Viterbi算法的实现——jieba.finalseg.viterbi
函数:
PrevStatus = {
'B': 'ES',
'M': 'MB',
'S': 'SE',
'E': 'BM'
}
def viterbi(obs, states, start_p, trans_p, emit_p):
V = [{}] # tabular
path = {}
for y in states: # init
V[0][y] = start_p[y] + emit_p[y].get(obs[0], MIN_FLOAT)
path[y] = [y]
for t in xrange(1, len(obs)):
V.append({})
newpath = {}
for y in states:
em_p = emit_p[y].get(obs[t], MIN_FLOAT)
(prob, state) = max(
[(V[t - 1][y0] + trans_p[y0].get(y, MIN_FLOAT) + em_p, y0) for y0 in PrevStatus[y]])
V[t][y] = prob
newpath[y] = path[state] + [y]
path = newpath
(prob, state) = max((V[len(obs) - 1][y], y) for y in 'ES')
return (prob, path[state])
为了适配中文分词任务,Jieba对Viterbi算法做了如下的修改:
- 状态转移时应满足
PrevStatus
条件,即状态(B)的前一状态只能是(E)或者(S),... - 最后一个状态只能是(E)或者(S),表示词的结尾。
与此同时,Jieba在实现公式eqref{eq:hmm}时,对其求对数,将相乘转化成了相加:
这就回答了上面的问题——为什么概率矩阵中出现了负数,是因为对其求了对数。
Jieba的HMM分词:
from jieba.finalseg import cut
sentence = "小明硕士毕业于中国科学院计算所,后在日本京都大学深造"
print('/'.join(cut(sentence)))
分词结果为“小明/硕士/毕业于/中国/科学院/计算/所/,/后/在/日/本京/都/大学/深造”,我们发现:关于“日本京都”出现分词错误的情况。这是因为最大条件概率(P(I|O))对应的状态序列不一定是分词正确的标注序列。此外,HMM做了两个基本假设:
- 齐次Markov性假设,即任意时刻t的状态仅与前一时刻状态相关,与其他时刻的状态、时刻t均无关;
- 观测独立性假设,任意时刻t的观测仅依赖于该时刻HMM的状态,与其他的观测及状态均无关。
HMM受限于这两个假设(字符(c_t)仅与前一字符(c_{t-1})相关),而不能学习到更多的特征,泛化能力有限。
4. 参考资料
[1] Xue, Nianwen. "Chinese word segmentation as character tagging." Computational Linguistics and Chinese Language Processing 8.1 (2003): 29-48.
[2] 李航. "统计学习方法." 清华大学出版社, 北京 (2012).
[3] Itenyh, Itenyh版-用HMM做中文分词二:模型准备.
[4] Django梦之队, 对Python中文分词模块结巴分词算法过程的理解和分析.(源链接挂了,为转载链接)