数据驱动的自动化机器学习流程生成方法

陈高建; 王菁; 栗倩文; 袁云静; 曹嘉琛

doi:10.16088/j.issn.1001-6600.2021071801

广西师范大学学报（自然科学版） ›› 2022, Vol. 40 ›› Issue (03) : 185 -193. DOI: 10.16088/j.issn.1001-6600.2021071801

数据驱动的自动化机器学习流程生成方法

陈高建, 王菁, 栗倩文, 袁云静, 曹嘉琛

作者信息 +

Author information +

文章历史 +

摘要

自动化机器学习是机器学习前沿的一个重要问题，自动化机器学习工具根据数据集及任务需求组合机器学习算子来构造流程，使领域用户在不具备专业机器学习知识的情况下也能完成相应数据分析工作，但目前的自动化机器学习工具普遍存在耗时长和精度低的问题。本文基于数据集相似性和强化学习原理，提出一种数据驱动的自动化机器学习流程的生成方法，利用相似数据集的历史知识，将神经网络与MCTS相结合，指导机器学习流程的生成。实验结果表明：该方法在耗时方面缩短至分钟级别，流程性能也得到提升。