计算机技术与发展2024,Vol.34Issue(4) :168-173.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0025

基于m×2正则化交叉验证的神经网络超参数调优方法

A Method for Hyper-parameter Tuning of Neural Network Based on m×2 Regularized Cross-validation

曹学飞 杨帆 李济洪 王瑞波 牛倩
计算机技术与发展2024,Vol.34Issue(4) :168-173.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0025

基于m×2正则化交叉验证的神经网络超参数调优方法

A Method for Hyper-parameter Tuning of Neural Network Based on m×2 Regularized Cross-validation

曹学飞 1杨帆 1李济洪 2王瑞波 2牛倩1
扫码查看

作者信息

  • 1. 山西大学 自动化与软件学院,山西 太原 030006
  • 2. 山西大学 现代教育技术学院,山西 太原 030006
  • 折叠

摘要

超参数调优是神经网络建模的关键问题.针对传统的超参数调优方法存在的问题,该文提出了一种基于m×2正则化交叉验证的超参数调优方法.目的是给出一种适用于复杂模型、大数据集背景下的计算开销较小且稳健的超参数调优方法.该方法的思想是从完整的数据集上选取少部分数据进行调优,避免模型在数据集较大时非常耗时的超参数调优难题;在m×2交叉验证的基础上设置正则化条件均衡训练集与验证集之间的分布差异,从而减少分布不一致带来的性能波动;使用信噪比作为调优的优化目标,从而可以综合考虑模型性能评价指标的均值和方差;并采用正交设计选择相关性较低的超参数组合以提高调优效率.以命名实体任务为例进行实验,在CoNLL 2003数据集上的实验结果显示,提出的调优方法能够选到和网格搜索性能上没有显著差异的超参数组合,且调优时间可显著降低约66%.

Abstract

Hyper-parameter tuning is a key issue in neural network modeling.From the viewpoint of the problems of traditional hyper-parameter tuning methods,we propose a hyper-parameter tuning method based on m×2 regularized cross-validation.The goal is to present a robust hyper-parameter tuning method with low computational cost suitable for complex models and large datasets.The idea of the proposed method is to select a small number of data from the complete dataset for tuning,so as to avoid the time-consuming problem of hyper-parameter tuning when the dataset is large.Then,on the basis of m×2 cross-validation,regularization is adopted to balance the distribution difference between the training set and the validation set to reduce the performance fluctuation caused by the distribution in-consistency.The signal-to-noise ratio is used as the metric of hyper-parameter tuning,so that the mean and variance of the model per-formance can be comprehensively considered.The orthogonal design is used to select a combination of hyper-parameters with low correlation to improve the tuning efficiency.The experimental results on the CoNLL 2003 dataset show that the proposed method can obtain a combination of hyper-parameters that is not significantly different from the grid search,and the tuning time can be significantly reduced by about 66%.

关键词

m×2交叉验证/正则化/神经网络/超参数调优/信噪比

Key words

m×2 cross-validation/regularization/neural network/hyper-parameter tuning/signal-to-noise

引用本文复制引用

基金项目

国家自然科学基金(61806115)

国家自然科学基金(62076156)

出版年

2024
计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子:0.621
ISSN:1673-629X
参考文献量23
段落导航相关论文