arXiv:2412.03666cs.LG2024-12被引 2

提出悲观双层优化,提升小样本下超参数搜索的泛化能力

Hyperparameter Tuning Through Pessimistic Bilevel Optimization

  • 采用悲观视角处理内层解的不确定性,避免乐观假设带来的风险
  • 在有限数据下,新方法使模型预测性能提升,测试扰动时更稳定
  • 适合小样本、高不确定性的超参数调优场景,如医疗或金融建模

机器学习中的自动化超参数搜索常被建模为双层优化问题:上层决定超参数,下层完成模型训练。现有方法多假设给定超参数下最优模型唯一,或在非唯一时采取乐观假设。但在复杂模型与有限数据下,该假设可能失效,导致模型不确定性。本文提出悲观双层超参数优化,显式考虑内层解集的潜在不确定性,以获得更具泛化能力的外层超参数。为此,我们设计一种基于松弛的近似求解方法,能获得更鲁棒的预测模型。在二分类器的自动超参数搜索实验中,当训练数据有限或测试数据受扰时,悲观解的表现优于传统乐观解,验证了悲观策略的必要性。

原文摘要 · Abstract (English)

Automated hyperparameter search in machine learning, especially for deep learning models, is typically formulated as a bilevel optimization problem, with hyperparameter values determined by the upper level and the model learning achieved by the lower-level problem. Most of the existing bilevel optimization solutions either assume the uniqueness of the optimal training model given hyperparameters or adopt an optimistic view when the non-uniqueness issue emerges. Potential model uncertainty may arise when training complex models with limited data, especially when the uniqueness assumption is violated. Thus, the suitability of the optimistic view underlying current bilevel hyperparameter optimization solutions is questionable. In this paper, we propose pessimistic bilevel hyperparameter optimization to assure appropriate outer-level hyperparameters to better generalize the inner-level learned models, by explicitly incorporating potential uncertainty of the inner-level solution set. To solve the resulting computationally challenging pessimistic bilevel optimization problem, we develop a novel relaxation-based approximation method. It derives pessimistic solutions with more robust prediction models. In our empirical studies of automated hyperparameter search for binary linear classifiers, pessimistic solutions have demonstrated better prediction performances than optimistic counterparts when we have limited training data or perturbed testing data, showing the necessity of considering pessimistic solutions besides existing optimistic ones.

超参数优化双层优化小样本学习鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。