一种基于决策树的双跳无线网络中继选择方法

技术领域

本发明涉及无线通信网络技术领域，具体涉及一种基于决策树的双跳无线网络中继选择方法。

背景技术

机器学习作为一种有效的人工智能技术，它可以用来完成各种任务。最近，人们已经尝试到将机器学习应用到无线通信中，例如使用k-NN和支持向量机SVM算法解决天线选择问题，使用SVM和贝叶斯方案解决了无线窃听通道中的天线选择问题。

中继选择是中继辅助网络中的一项重要技术，并在过去十年中得到了广泛的研究。传统的中继选择依赖于最大化算法和准确的信道状态信息(CSI)。当其他问题与中继选择共存时，例如多跳中继或子信道分配问题，还有成本和资源消耗的问题。优化驱动选择的瓶颈和机器学习的发展激发了我们探索全新的中继选择方法。

有学者研究出了一些方案，但是这些方案的计算复杂度较高，会导致资源浪费和较高的成本，因此，需要进一步改进。

发明内容

本发明的目的是提供一种基于决策树的双跳无线网络中继选择方法，运用机器学习的知识研究两跳无线网络中的中继选择问题。

为了达到上述目的，本发明通过以下技术方案实现：

一种基于决策树的双跳无线网络中继选择方法，所述双跳无线网络包括源节点S、目的节点D和K个中继，所述方法包括：

步骤S1，准备训练数据G＝{(x

步骤S2，将训练数据G输入到决策树进行训练，构建决策树模型；

步骤S3，采用所述决策树模型进行双跳无线网络中继选择。

进一步的，

其中，g

进一步的，传统算法的最佳k

其中，g

进一步的，所述双跳无线网络遵循DF协议，通过中继r

进一步的，信道状态信息阈值θ按照如下方式确定；

通过数值仿真计算的方法获得能够使信息增益最大化的最佳阈值θ。

进一步的，信道状态信息阈值θ按照如下方式确定：

通过数值仿真计算的方法获得能够使基尼指数最大化的最佳阈值θ。

进一步的，所述决策树模型采用CART算法。

进一步的，所述决策树模型的分裂标准为信息增益或基尼指数。

本发明与现有技术相比，具有以下有益效果：

1、本发明具有较低的计算复杂度：在预测过程中，决策树不会对预测的每个输入要素进行测试。因此，本发明提供的预测最佳中继的计算复杂度低于基于最大化的算法。

2、本发明具有较小的反馈量：基于决策树的中继预测仅需要量化的信道状态信息(CSI)，其是二进制值，并且每个信道仅占用1比特。

3、本发明具有量化的CSI反馈：估计量化的CSI并将其反馈给中央控制器，这对于难以获得准确CSI的网络来说是一个很大的优势。

附图说明

为了更清楚地说明本发明的技术方案，下面将对描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图是本发明的一个实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图：

图1为本发明一实施例提供的一种基于决策树的双跳无线网络中继选择方法的流程图；

图2为基于决策树的中继选择系统模型图；

图3为最优θ的数值分析图；

图4为四种方案的性能比较图；

图5为正确比率图。

具体实施方式

以下结合附图1～5和具体实施方式对本发明提出的技术方案作进一步详细说明。根据下面说明，本发明的优点和特征将更清楚。需要说明的是，附图采用非常简化的形式且均使用非精准的比例，仅用以方便、明晰地辅助说明本发明实施方式的目的。为了使本发明的目的、特征和优点能够更加明显易懂，请参阅附图。须知，本说明书所附图式所绘示的结构、比例、大小等，均仅用以配合说明书所揭示的内容，以供熟悉此技术的人士了解与阅读，并非用以限定本发明实施的限定条件，故不具技术上的实质意义，任何结构的修饰、比例关系的改变或大小的调整，在不影响本发明所能产生的功效及所能达成的目的下，均应仍落在本发明所揭示的技术内容能涵盖的范围内。

本发明提供的一种基于决策树的双跳无线网络中继选择方法，将中继选择划分为一个多类别分类问题，提出一种基于决策树的解决方案，通过对每个中继的等效信道状态信息进行二进制量化生成输入特征，消除了中继选择对最大化算法的依赖。采用两个分裂标准来优化量化参数，即信息增益和基尼指数，最后的仿真结果验证了量化阈值的最优性，表明基于决策树的中继选择方案可以达到与最优选择相当的平均端到端速率。与现有技术相比，本发明基于决策树的中继选择的优点在于较低的计算复杂度，较小的反馈量和量化的信道状态信息要求。

本发明提供一种基于决策树的双跳无线网络中继选择方法，所述双跳无线网络包括源节点S、目的节点D和K个中继，如图1所示，所述方法包括：

步骤S1，准备训练数据G＝{(x

步骤S2，将训练数据G输入到决策树进行训练，构建决策树模型；

步骤S3，采用所述决策树模型进行双跳无线网络中继选择。

首先对本发明所述的双跳无线网络的系统模型进行介绍。在本发明中，如图2所示，双跳网络包含源节点(S)、目的节点(D)和由r

双跳传输中的常规中继选择是集中式的和优化驱动的，中央控制器收集所有中继的g

将平均E2E速率作为评估不同中继选择方案性能的指标，由式(2)计算得出，

下面对本发明的基于决策树的中继选择的具体方法进行介绍。

本发明考虑用有监督的机器学习解决中继选择问题。具体而言，本发明将中继选择问题建模为K类分类问题，根据训练数据构建分类树模型，并预测当前CSI所属的分类标签。基于决策树的中继选择包括三个阶段：S1准备训练数据，S2构建决策树模型和S3预测中继选择。

S1准备训练数据：基于决策树的中继选择所使用的训练数据定义为G＝{(x

将g

S2构建决策树模型：基于训练数据G中M个样本的输入x

S3预测中继选择：在进行中继选择时，将x＝[x

整个过程总结如下：

本实施例中，信道状态信息阈值θ在提取输入要素中起着重要作用，如何确定θ的值会极大地影响决策树的性能。信息增益和基尼指数是描述数据分裂纯度的两个标准，信息增益衡量分裂后的熵损失，而基尼指数是衡量样本被错误分类的概率。本发明按照上述两个分裂标准推导，目的是通过最佳的计算推导来优化θ。

A、信息增益优化

信息增益表示数据集被F

这里，H(G)是样本数据的经验熵，

H

(8)和(9)中的概率推导如下。首先，令

当i＝k时，

当i≠k时，

将上式代入(8)，得到

H

当i＝k时，

当i≠k时，

把(13)和(14)代入(9)将获得

结合以上得出的所有结果，可以获得信息增益的表达式。现在，考虑两种极端情况：θ＜＜2λ和θ＞＞2λ。

θ＜＜2λ：大多数输入样本的

θ＞＞2λ：大多数输入样本具有

因为信息增益是严格的凹函数，所以存在一个θ最大化信息增益的最佳值。很难明确表达最优θ，因此我们可以通过数值仿真计算的方法得到它。

B、基尼指数优化

为了方便比较，将基尼指数定义为基尼杂质的变化。假设考虑了特征F

Gini(G)表示样本数据中的杂质，与θ无关，所以只需要对

其中G

在与信息增益相同的极端情况分析之后，可以得出结论，存在最优的θ最大化基尼指数并且可以通过数值仿真计算的方法获得。

注意，θ的这种优化方法仅适用于聚集(i.i.d.)场景，其中所有中继CSI具有相同的平均值。在独立的非同分布(i.n.i.d.)场景中，该方案然有效，但需要设计一种新的θ优化方法。

下面对本发明的方法进行性能评估，具体步骤为：

A.仿真模拟结果

通过数值仿真计算的方法检索最优θ，然后采用CART算法构建决策树，由CART算法采用Gini指数作为分裂准则并构建二叉树，通过Matlab生成了10

首先，本发明展示了分裂标准的数值结果，在图3中，可以找到θ的最佳值，图中还描述了通过Monte Carlo模拟计算得出的平均E2E速率，作为一个优化验证的判断标准。曲线的最佳点用小方块标记，当λ＝1时，可以看到存在一个同时取得最佳信息增益和基尼指数的点。同时，该值也使平均E2E率最大化，这证明了最优θ的存在。当λ＝0.5时，三条曲线的最佳点仍然重合。由于信道质量的下降，θ的最佳值变大。值得一提的是，可以采用两个分裂标准中的任何一个，因为它们都在同一个θ点取得最大化。

在图4中，将基于决策树的中继选择的性能与最优选择、随机选择和Q-最优作比较。描述了四种方案在不同中继数量下的平均E2E速率，可以很明显地观察到基于决策树的方案的平均E2E率随着K的增加而增加，这证明了中继选择的优势。还可以观察到，与最优选择相比，基于决策树的方案具有平和的损失率。性能差距随着K的增长而增加，因为最佳选择会选择具有最佳CSI的中继，其CSI随K的增长而增加。相反，本发明所提出的方案使用二进制量化结果作为信道信息，不能像最佳方案一样受益于更大的K。另一个重要的观察结果是，由小方块标记的Q-最佳的平均E2E率与本发明的方案的曲线一致。这意味着如果仅有量化的CSI可用，则本发明的方案可以实现最佳性能。

正确率，即分类的准确性，定义为正确分类的样本数与总数之比。如图5所示，可以看到平均E2E速率和正确率的比较图，可以观察到在取得最大化平均E2E速率的时候，并没有取得最佳正确率。回想一下，当最佳的θ使得分割标准最大化的同时，也会使得平均端到端速率最大化。由此可以得出结论，与常规分类指标(例如正确率)相比，通信指标更适合评估基于决策树的中继选择的性能。

B.成本分析

在表1中比较了基于决策树的选择，最佳选择，Q最优和随机选择的成本。

(1)计算复杂性：优化驱动的中继选择方案依赖于最大化算法和O(K)的成本复杂性。决策树预测的复杂性由树深度决定，导致平均时间复杂度低于O(K)。以平衡树为例，其复杂度为O(log

表1中继选择方案比较

(2)反馈量：在最佳方案中，反馈所有中继的

(3)CSI准确度：优化驱动方案需要准确的CSI，而本发明提出的基于决策树的方案仅需要量化的CSI。这相较于难以获得某些节点的准确的CSI网络来说是一种优势。

综上所述，本发明提出了一种基于决策树的双跳无线网络中继选择方案。通过对每个中继的等效CSI进行二进制量化来生成离散特征。推导出了两个最佳分裂标准，即信息增益和基尼指数，然后用他们来优化量化阈值。仿真结果验证了量化阈值的最优性，并表明基于决策树的中继选择方案可以实现与最优选择相当的平均端到端速率。基于决策树的中继选择的优点在于较低的计算复杂度，较小的反馈量和量化的CSI需求。

需要说明的是，在本文中，诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来，而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下，由语句“包括一个……”限定的要素，并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。

尽管本发明的内容已经通过上述优选实施例作了详细介绍，但应当认识到上述的描述不应被认为是对本发明的限制。在本领域技术人员阅读了上述内容后，对于本发明的多种修改和替代都将是显而易见的。因此，本发明的保护范围应由所附的权利要求来限定。