基于 Nebula Graph 的 BetweennessCentrality 算法

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

>>>>1. 算法介绍

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

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

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

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

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

>>>>2. 应用场景

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

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

>>>>3. 介数中心性公式

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

（公式 1 ）

其中

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

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

（公式 2 ）

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

（公式 3 ）

>>>>4. 求解思路

求节点 v 的介数中心性，即计算，

需要知道节点 v 是否在 s 到 t 的路径上。

（公式 4 ）

当 v 不位于s 到 t 的最短路径上， 即不存在。

综上两种情况得出公式 5 ：

（公式 5 ）

 （2）根据上面公式可得：节点 s 到节点 t 的经过 w 的最短路径条数为

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

（公式 6 ）

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

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

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

基于 NebulaGraph 的 BetweennessCentrality 实现了针对有权图和无权图的计算，实现代码见 ​​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：

计算节点 2 的 BC：

计算节点 3 的 BC：

计算节点 4 的 BC：

计算节点 5 的 BC：

所以每个节点的 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​​​

本文转载自公共号Nebula Graph Community