lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

时间:2024-03-29 16:54:17

Finite StateTransducers 简称 FST,中文名:有穷状态转换器。在自然语言处理等领域有很大应用,其功能类似于字典的功能(STL 中的map,C# 中的Dictionary),但其查找是O(1)的,仅仅等于所查找的key长度。目前Lucene4.0在查找Term时就用到了该算法来确定此Term在字典中的位置。

      FST 可以表示成FST<Key, Value>的形式,我们可以用O(length(key))的复杂度,找到key所对应的值。除此之外,FST 还支持用Value来查找key以及查找Value最优的key等功能。


在Lucene4中,很重要的一种数据结构就是FST(Finite State Transducer),又称Mealy Machine。

在SynonymFilter的实现中,FST可以用HashMap代替。但是相比HashMap,FST有以下优点:

1、紧凑的结构,通过对词典中单词前缀和后缀的重复利用,压缩了存储空间。

2、O(len(str))的查询时间复杂度。

如果不考虑FST的输出,FST本质上是一个最小的,有向无环DFA。

经过差不多一个星期的学习,终于对Lucene中FST的构造过程有了初步的了解。为了简化问题,我们把FST的输出都置为相同的因子。

案例:

对aaaa,bbaa,ccbaa,ddcbaa这4个单词构建FST。(注意:单词插入前必须先进行排序,否则就无法生成最小FST)


先给出最终的结果:

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

(注:这副图来自于http://examples.mikemccandless.com/fst.py)


那么其构造的过程是怎么样的呢?


我们从Lucene42的org.apache.lucene.util.fst.Builder类的add()方法作为切入口。

对于Lucene42的源代码,有这样几点需要理解。

1、UnCompiledNode类和CompiledNode类。

UnCompiledNode可以理解为没有编入FST的结点,CompliedNode为已经编入FST的结点。在Builder类中,初始化了一个UnCompiledNode数组。这个数组中存储的是什么呢?论文《Direct Buildingof Mimimal Automation for A Given List》中描述构造一个伪最小FST的方法:

假设有{w1,w2....,wn} n个单词。

a、先构造一个除单词w1外,最小的FST。(此时FST中有w1一个单词)

b、构造一个除w2外,最小的FST。(此时FST中有w1,w2两个单词)

c、构造一个除w3外,最小的FST。(此时FST中有w1,w2,w3三个单词)

d、构造一个除w4外,最小的FST。(此时FST中有w1,w2,w3,w4四个单词)

e、构造一个除ε(空串)外,最小的FST。此时FST最小。

这其实就是一种迭代的思想。UnCompiledNode存储的就是上文提到的单词w1,w2……。


2、freezeTail方法的作用是什么?

freezeTail方法的作用就是把UnCompiledNode进行Compile成CompiledNode。也就是构造一个除wi外,最小的FST。通过方法名称,也可以理解为把后缀(tail)固定(freeze)下来。

看论文不如走代码,我们一步步看代码是怎么走吧。


第一步:往FST中插入第一个单词aaaa。

插入的代码如下:

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

此时FST的结构如下:

(状态的编号代表Builder中frontier数组的下标;红色的箭头代表是结束边)

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

上图的FST即是除字符串aaaa外最小的FST,(其实最小FST为空)。


第二步:往FST中插入第二个单词bbaa。

在插入之前,需要对aaaa进行freezeTail,即把除aaaa和bbaa前缀之外的部分compile到最小FST中。

Compile的过程是从后往前的,也就是先处理State4,然后处理State3……

FST把单词的后缀存储到HashNode这种数据结构中去,如果新插入的字符串在compile过程中,与前面的单词有公共后缀,则直接使用其后缀,而不创建新的结点。

其过程依次如下:(具体的代码参考freezeTail函数)

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

freezeTail的过程中,把frontier的位置空出来了。然后将新的字符串插入到frontier中。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想


第三步:往FST中插入ccbaa。

   在插入ccbaa之前,仍然需要将bbaa进行freezeTail。

由于aaaa和bbaa有{a},{aa}这两个共同的后缀,所以C2 、C4节点可以重复利用。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

这个过程就是在利用后缀对状态进行压缩。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想


第四步:往FST中插入ddcbaa.

在插入之前对ccbaa进行freezeTail操作。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

有了前面的基础,这里就只画出来了它的变化。经过freezeTail后,frontier数组又空出来了。接下来就把ddcbaa插入到FST中去。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想



最后一步,Builder.finish()。

在这一步中,会调用freezeTail(0)。这样的话,整个FST就变成了一个最小的,无环FST了。

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

将两副图进行对比:

lucene源代码学习之FST(Finite State Transducer)在SynonymFilter中的实现思想

是不是一样的?


图中状态里面的数字,如 -1, 2,4 是程序处理过程中后缀存储在NodeHash中的位置。可以这样理解,FST用NodeHash存储了字符串所有可能的后缀状态。


这里的分析过程,忽略了FST对output的处理。Output对FST的结构是没有影响的。


FST作为一种数据结构,在自然语言处理中应用非常广泛,比如语音识别,机器翻译等。在Lucene中也有广泛的应用,比如同义词处理、FuzzyQuery、拼写检查、自动补全等。其理论基础就是有限状态机。本文略过了论文中数学证明等知识点,(其实是我自己也没弄清楚 )从Lucene源代码实现的过程对FST的构造思想进行追踪。


参考资料:

1、 Build your own finite state transducer

(http://blog.mikemccandless.com/2013/06/build-your-own-finite-state-transducer.html)

2、Lucene&solr4 实践(4)(http://blog.sina.com.cn/s/blog_4d58e3c001016gg4.html)

3、《自动机理论、语言和计算导论(第二版)》

4、《编译原理》

5、《Direct Building of Mimimal Automation for A Given List》

6、《Direct Construction of Mimimal Structure Acyclic SubsequentialTransducers》


转载地址:点击打开链接

参考资料:点击打开链接