[发明专利]一种基于AdaBoost的链路预测算法

说明书

技术领域

本发明涉及互联网技术，具体涉及一种链路预测的实现方法。

背景技术

链路预测是链路挖掘中将链接作为挖掘对象的应用。主要预测已经存在但尚未被发现的链接以及尚未链接的节点间未来产生链接的可能性。随着一些链路预测算法开始在商业领域得到应用，与之相关的研究已经成为一个热门领域，其中基于拓扑图的链路预测算法研究工作在近年来受到了广泛重视。例如Facebook采用基于RWR(Random Walk with Restart)的方法预测用户的朋友关系，据此提高好友推荐的成功率。

基于网络拓扑图的链路预测算法主要包括基于节点邻居的相似性，基于最大似然估计以及基于概率模型等三种类型。代表性算法包括基于局部信息相似性的共同邻居(CommonNeighbors)算法，基于路径相似性的Katz算法和基于随机游走相似性的RWR算法。其中，基于节点邻居相似性的链路预测算法研究较早，在实际工作中取得了广泛应用。另一类取得实际推广应用的方法是基于随机游走的链路预测算法。这类算法的基本思想都是对图中节点所有可能的组合进行排序，选择其中最可能出现在新图中的节点对(即图中的边)。然而近一两年来，无论是在对已有算法的改进，还是在提出新算法方面，都没有出现有突破性的成果，基于拓扑的链路预测算法的召回率依然较低。

发明内容

本发明的目的是提供一种基于AdaBoost的链路预测算法。使用本发明提供的实施例，可以对当前网络拓扑图中将来可能发生链接的节点对进行预测。

为了克服当前主流的基于网络拓扑结构的链路预测算法普遍存在召回率较低的问题。通过我们的研究发现，现有的主流链路预测方法的预测结果并不完全相交，利用算法结果的叠加提高召回率。但是，直接累加求和并不可行，因为会降低总的算法精度。据此考虑采用Boosting方法对其进行改进。首先将链路预测问题看作二分类问题，对下一时刻网络中每一条可能存在的边(节点对)，其分类结果为两类：存在或不存在。接下来借用Boosting方法通过错误反馈提升弱学习算法得到强学习算法的思想，根据一定的原则选择若干链路预测算法作为弱分类器，基于AdaBoost算法提出并实现了一个新的链路预测方法。

该方法的步骤包括：

读取预测训练样本以及预测测试样本；

为预测训练样本附上其真实所在类的标签值；

为每个样本的权重赋初始值；

选取若干链路预测算法作为弱分类器；

使用各个分类器为训练样本做分类；

计算每个分类器的投票权重；

使用每个分类器为预测测试集合中的样本做分类；

按上述各分类器的分类结果为预测测试集合中的样本投票，做出最终预测；

输出对预测测试集合中样本的预测结果；

最后，实施本发明具有以下有益效果：

本发明实施例的有益效果是，将Boosting思想应用于链接预测之中，相对于现有的各种常用算法而言，具有更高的灵敏度和更低的误报率，能够在显著提高算法召回率的同时，保持计算结果的正确性。

附图说明

附图是本发明改进现有链路预测算法提出的一种基于AdaBoost的链路预测算法的算法流程。

具体实施方式

下面结合附图对本发明的具体实施方式进行描述，以便本领域的技术人员更好地理解本发明。

在本实施例中，如图所示，提供了一个优化的算法流程：

步骤101、读取预测训练样本以及预测测试样本；

对于预测训练样本以及训练测试样本，读取其信息并生成网络拓扑图。

步骤102、为预测训练样本附上其真实所在类的标签值；

对于一组长度为m的预测训练集合C。Ω表示x_i被分类的类型值的集合。对于x_i，如果它确实出现在下一时间段的图中，则y_i＝1，反之，y_i＝-1。

步骤103、为每个样本的权重赋初始值；

每个样本的权重初始值相等，是整个样本长度的倒数，即为1/m。

步骤104、选取若干链路预测算法作为弱分类器；

按照预测结果互补的原则选取基于节点邻居的相似性，基于最大似然估计以及基于概率模型等三种类型链路预测的方法作为弱分类器。

步骤105、使用各个分类器为训练样本做分类；