基于消息队列的LightGBM超参数优化

• 作者: 南东亮 王海云 王维庆
• 来源期刊: 计算机工程与科学
• 学科: 工学
• 关键词: 消息队列 LightGBM 超参数
• 年，卷（期）: 2019,（8）
• 所属期刊栏目: 高性能计算
• 页码范围: 1360-1365
• 页数: 6页
• 分类号: TP181
• 字数: 4211字
• 语种: 中文
• DOI: 10.3969/j.issn.1007-130X.2019.08.004

摘要

为了提高LightGBM超参数优化效率,同时得到全局最优模型,提出了以消息队列方式并行优化LightGBM超参数方法.根据超参数的预选范围,将每一组超参数发送到队列中,各节点从队列获取到消息后以并行方式进行模型训练并验证准确率,最后选出准确率最高的模型计算待预测的数据集.实验结果表明,与传统的网格搜索、贝叶斯法、随机搜索方法以及消息队列串行优化相比,消息队列并行优化超参数方法时间最短,AUC值最大.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	王维庆	新疆大学电气工程学院	270	1957	20.0	32.0
2	王海云	新疆大学电气工程学院	142	589	12.0	16.0
3	南东亮	新疆大学电气工程学院	27	55	5.0	6.0