• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学

• 论文 • 上一篇    下一篇

基于局部收敛权阵进化的BP神经网络MapReduce训练

陈旺虎,俞茂义,马生俊,李金溶,郏文博   

  1. (西北师范大学计算机科学与工程学院,甘肃 兰州 730070)
  • 收稿日期:2016-08-26 修回日期:2016-10-20 出版日期:2016-12-25 发布日期:2016-12-25
  • 基金资助:

    国家自然科学基金(61462076);甘肃省自然科学基金(1104GKCA023);甘肃省科技攻关项目(1208RJZA134);西北师范大学青年教师科研提升计划(NWNULKQN1230)

MapReduce training of BP neural networks
based on local weight matrix evolution

CHEN Wanghu,YU Maoyi,MA Shengjun,LI Jingrong,JIA Wenbo   

  1. (College of Computer Science and Engineering,Northwest Normal University,Lanzhou 730070,China)
  • Received:2016-08-26 Revised:2016-10-20 Online:2016-12-25 Published:2016-12-25

摘要:

为提高大样本集情况下BP神经网络的训练效率,提出了一种基于局部收敛权阵进化的BP神经网络MapReduce训练方法,以各Map任务基于其输入数据分片训练产生的局部收敛权阵作为初始种群,在Reduce任务中通过种群进化,选取适应度最高的权阵作为Map任务下一轮训练的初始权阵,直至该权阵对所有输入数据分片收敛。实验结果表明,与现有方法相比,该方法可有效避免MapReduce训练BP神经网络时容易陷入局部收敛的问题,并大大减少训练时间。

 

关键词: BP神经网络, 遗传算法, MapReduce, 收敛

Abstract:

To improve the efficiency of BP neural network (BPNN) training, we propose an approach with MapReduce based on the local weight matrix evolution. The local weight matrices produced by Map tasks based on its input data splits are passed to Reduce tasks as the initial population of a genetic algorithm. After the evolution of the current population, the weight matrix with highest fitness is selected as the initial weight matrix in the next turn of training. The training does not stop until the weight matrix is convergent on the whole sample data. Experiments show that the proposed approach can maintain the global convergence of a BPNN on its whole training sample data, and improve the efficiency of BP-NN training with MapReduce.

Key words: BP neural network, genetic algorithm, MapReduce, convergence