arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.01575cs.LGcs.HCstat.ML

StackGenVis:基于性能指标的堆叠集成学习数据、算法与模型对齐

StackGenVis: Alignment of Data, Algorithms, and Models for Stacking Ensemble Learning Using Performance Metrics

Linnaeus University(林奈大学)

Angelos Chatzimparmpas, Rafael M. Martins, Kostiantyn Kucher, Andreas Kerren

AI总结:

本研究针对堆叠集成学习从零构建模型堆叠繁琐的问题,提出结合可视化的知识生成模型与可视化分析系统StackGenVis,支持多环节优化,通过用例和专家访谈验证了其有效性。

摘要:

在机器学习(ML)领域,bagging、boosting和stacking等集成方法是已被广泛认可的常用方法,通常能实现顶尖的预测性能。Stacking(也称为“堆叠泛化”)是一种集成方法,它将至少一层的异构基模型组合起来,再使用另一个元模型来汇总这些模型的预测结果。尽管stacking可能是提升机器学习预测性能的高效方法,但从零开始构建模型堆叠可能是一个繁琐的试错过程。这一挑战源于可用解决方案的空间极为庞大:可用于训练的数据实例和特征组合多种多样,有多种算法可供选择,且这些算法采用不同参数的实例化结果(即模型)在不同指标下表现各异。在本研究中,我们提出了一个支持集成学习的知识生成模型(结合可视化技术),以及一套面向堆叠泛化的可视化分析系统。我们的系统StackGenVis可协助用户动态调整性能指标、管理数据实例、为给定数据集筛选最重要的特征、挑选一组性能优异且具有多样性的算法,以及衡量预测性能。因此,我们提出的工具可帮助用户在不同模型间做出选择,并通过剔除表现过拟合预期和性能不佳的模型来降低最终堆叠结构的复杂度。我们通过两个用例验证了StackGenVis的适用性和有效性:一个真实医疗保健数据集,以及一组与文本情感/立场检测相关的数据。最后,我们通过对三位机器学习专家的访谈对该工具进行了评估。

英文摘要:

In machine learning (ML), ensemble methods such as bagging, boosting, and stacking are widely-established approaches that regularly achieve top-notch predictive performance. Stacking (also called "stacked generalization") is an ensemble method that combines heterogeneous base models, arranged in at least one layer, and then employs another metamodel to summarize the predictions of those models. Although it may be a highly-effective approach for increasing the predictive performance of ML, generating a stack of models from scratch can be a cumbersome trial-and-error process. This challenge stems from the enormous space of available solutions, with different sets of data instances and features that could be used for training, several algorithms to choose from, and instantiations of these algorithms using diverse parameters (i.e., models) that perform differently according to various metrics. In this work, we present a knowledge generation model, which supports ensemble learning with the use of visualization, and a visual analytics system for stacked generalization. Our system, StackGenVis, assists users in dynamically adapting performance metrics, managing data instances, selecting the most important features for a given data set, choosing a set of top-performant and diverse algorithms, and measuring the predictive performance. In consequence, our proposed tool helps users to decide between distinct models and to reduce the complexity of the resulting stack by removing overpromising and underperforming models. The applicability and effectiveness of StackGenVis are demonstrated with two use cases: a real-world healthcare data set and a collection of data related to sentiment/stance detection in texts. Finally, the tool has been evaluated through interviews with three ML experts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑