图数据库｜基于 Nebula Graph 的 BetweennessCentrality 算法

本文首发于 Nebula Graph Community 公众号

在图论中，介数（Betweenness）反应节点在整个网络中的作用和影响力。而本文主要介绍如何基于 Nebula Graph 图数据库实现 Betweenness Centrality 介数中心性的计算。

1. 算法介绍

中心性是用来衡量一个节点在整个网络图中所在中心程度的概念，包括度中心性、接近中心性、中介中心性等。其中度中心性通过节点的度数（即关联的边数）来刻画节点的受欢迎程度，接近中心性是通过计算每个节点到全图其他所有节点的路径和来刻画节点与其他所有节点的关系密切程度。

中介中心性则用于衡量一个顶点出现在其他任意两个顶点对之间最短路径上的次数，从而来刻画节点的重要性。

节点介数中心性的定义是：在所有最短路径中经过该节点的路径数目占最短路径总数的占比。

计算图中节点的介数中心性分为两种情况：有权图上的介数中心性和无权图上的介数中心性。两者的区别在于求最短路径时使用的方法不同，对于无权图采用 BFS（宽度优先遍历）求最短路径，对于有权图采用 Dijkstra 算法求最短路径。

下面所介绍的算法都是针对无向图的。

2. 应用场景

介数反应节点在整个网络中的作用和影响力，主要用于衡量一个顶点在图或网络中承担“桥梁”角色的程度，图中节点 C 就是一个重要的桥梁节点。

中心性可用于金融风控领域中反欺诈场景里中介实体的识别。也可用于医药领域中特定疾病控制基因的识别，用以改进药品的靶点。

3. 介数中心性公式

节点介数中心性的计算公式如下：

（公式 1）

其中

：经过节点 v 的 s 到 t 的最短路径条数；

：节点s到节点t的所有最短路径条数；

s 和 t 是属于节点集合的任意一个节点对。

为方便计算，将每对顶点的介数计算定义为：

（公式 2）

所以上面的公式 1 可以用公式 2 代替，即

（公式 3）

4. 求解思路

求节点 v 的介数中心性，即计算，需要知道节点 v 是否在 s 到 t 的路径上。

（1）求节点 v 是否在 s 到 t 的最短路径上，采用下面公式判断表示两点之间的最短路径长度)：

当 v 位于 s 到 t 的最短路径上时，有

（公式 4）

又因为和是互相独立的，根据数学组合知识得知 s 到 t 的最短路径总数是 s 到 v 的最短路径数与 v 到 t 的最短路径数的乘积。

所以有下面公式：

（公式 5）

（2）根据上面公式可得：

节点 s 到节点 t 的经过 w 的最短路径条数为，在图中节点 v 是 w 的前置节点，所以 st 之间经过节点 v 和 w 的最短路径条数计算公式为：

（公式 6）

下面分为两种情况：分别是和

（一）

（公式 7）

（二）时

（公式 8）

（3）所以将上面两种情况加起来，得到经过 v 的 s 到所有顶点的最短路径数占 s 到所有顶点的最短路径数的比值。

（公式 9）

其中即 v 是 s 到 w 路径中 w 的前驱节点。

（4）根据上面的求的公式，下面给出论文中求解无权图时的算法流程，如下所示。

对于无权图实现根据上面流程实现。

有权图的介数中心性计算需要将求解最短路径的方法改成采用 Dijkstra 方法，即改动第一个 while 循环内的代码。

基于 Nebula Graph 的 Betweenness Centrality 实现了针对有权图和无权图的计算，实现代码见 https://github.com/vesoft-inc/nebula-algorithm/blob/master/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala。

5. 计算示例

首先读取 Nebula Graph 中的图数据，可以指定其边数据进行数据读取。

其次针对 Nebula Graph 的边数据构造拓扑图，执行中心性计算。

读取的 Nebula Graph 图数据以该无权图为例：

计算节点 1 的 BC：

经过1节点的最短路径节点对	节点对之间的最短路径总数	占通过 1 节点的最短路径数
2-4	3 （2-3-4，2-5-4，2-1-4）	1
节点 1 的 BC：	1/3

计算节点 2 的 BC：

经过 2 节点的最短路径节点对	节点对之间的最短路径总数	占通过 1 节点的最短路径数
1-3	2 （1-2-3，1-4-3）	1
3-5	2（3-2-5，3-4-5）	1
节点 2 的 BC：	1

计算节点 3 的 BC：

经过 3 节点的最短路径节点对	节点对之间的最短路径总数	占通过 1 节点的最短路径数
2-4	3 （2-3-4，2-5-4，2-1-4）	1
节点 3 的 BC：	1/3

计算节点 4 的 BC：

经过 4 节点的最短路径节点对	节点对之间的最短路径总数	占通过 1 节点的最短路径数
1-3	2 （1-4-3，1-2-3）	1
3-5	2（3-4-5.3-2-5）	1
节点 4 的 BC：	1

计算节点 5 的 BC：

经过 5 节点的最短路径节点对	节点对之间的最短路径总数	占通过 1 节点的最短路径数的百分比
2-4	3 （2-3-4，2-5-4，2-1-4）	1
节点 5 的 BC：	1/3

所以每个节点的 BC 值是：

1: 1/3

2: 1

3: 1/3

4: 1

5: 1/3

6. 算法结果示例

数据：读取 Nebula Graph test 中的边数据，以 srcId、dstId 和 rank 分别作为拓扑图中的边的三元组（起点、重点、权重）

(root@nebula) [test]> match (v:node) -[e:relation] -> ()  return e

+------------------------------------+

| e                                  |

+------------------------------------+

| [:relation "3"->"4" @1 {col: "f"}] |

+------------------------------------+

| [:relation "2"->"3" @2 {col: "d"}] |

+------------------------------------+

| [:relation "2"->"5" @4 {col: "e"}] |

+------------------------------------+

| [:relation "4"->"5" @2 {col: "g"}] |

+------------------------------------+

| [:relation "1"->"5" @1 {col: "a"}] |

+------------------------------------+

| [:relation "1"->"2" @3 {col: "b"}] |

+------------------------------------+

| [:relation "1"->"4" @5 {col: "c"}] |

+------------------------------------+

读取 Nebula Graph 边数据，设置无权重并执行 BC 算法，输出结果如下：

vid: 4 BC: 1.0

vid: 1 BC: 0.3333333333333333

vid: 3 BC: 0.3333333333333333

vid: 5 BC: 0.3333333333333333

vid: 2 BC: 1.0

读取 Nebula Graph 边数据，设置有权重并执行 BC 算法，输出结果如下：

vid: 4 BC: 2.0

vid: 1 BC: 0.5

vid: 3 BC: 1.0

vid: 5 BC: 2.0

vid: 2 BC: 0.0

7. 参考资料

论文《A Faster Algorithm for Betweenness Centrality》
Python 的 NetworkX 实现介数中心性的源码：https://github.com/networkx/networkx/blob/master/networkx/algorithms/centrality

本文中如有任何错误或疏漏，欢迎去 GitHub：https://github.com/vesoft-inc/nebula issue 区向我们提 issue 或者前往官方论坛：https://discuss.nebula-graph.com.cn/ 的 建议反馈 分类下提建议；交流图数据库技术？加入 Nebula 交流群请先填写下你的 Nebula 名片，Nebula 小助手会拉你进群~~