As a result of the ever increasing complexity of configuring and fine-tuning machine learning models, the field of automated machine learning (AutoML) has emerged over the past decade. However, software implementations like Auto-WEKA and Auto-sklearn typically focus on classical machine learning (ML) tasks such as classification and regression. Our work can be seen as the first attempt at offering a single AutoML framework for most problem settings that fall under the umbrella of multi-target prediction, which includes popular ML settings such as multi-label classification, multivariate regression, multi-task learning, dyadic prediction, matrix completion, and zero-shot learning. Automated problem selection and model configuration are achieved by extending DeepMTP, a general deep learning framework for MTP problem settings, with popular hyperparameter optimization (HPO) methods. Our extensive benchmarking across different datasets and MTP problem settings identifies cases where specific HPO methods outperform others.
翻译:由于配置和微调机器学习模式日益复杂,在过去十年中出现了自动机学习领域(自动ML),然而,Auto-WEKA和Auto-sklearn等软件的实施通常侧重于古典机学习任务,如分类和回归等。我们的工作可被视为第一次尝试,为属于多目标预测伞下的大多数问题设置提供单一的自动ML框架,其中包括流行的 ML 设置,如多标签分类、多变量回归、多任务学习、dyadic预测、矩阵完成和零镜头学习。通过扩展深度MTP,即MTP问题设置的一般深层次学习框架,采用流行性超参数优化(HPO)方法,实现了自动问题选择和模型配置。我们在不同数据集和MTP问题设置上的广泛基准,确定了具体HPO方法优于其他方法的案例。