arXiv:2501.13018cs.LGcs.IT2025-01NeurIPS被引 1

用可靠性图优化超参数,兼顾准确率与成本。

Multi-Objective Hyperparameter Selection via Hypothesis Testing on Reliability Graphs

论文配图:Multi-Objective Hyperparameter Selection via Hypothesis Testing on Reliability Graphs
图 1 · 摘自论文原文
  • 基于有向无环图建模超参数间可靠性关系
  • 在保证错误发现率的前提下提升搜索效率
  • 适合需要可靠评估的模型调优场景

超参数选择(如大语言模型中的提示模板)需在可靠性与成本之间取得平衡。许多情况下,可通过先验信息和预留数据推断出超参数间预期可靠性关系——例如更长的提示模板可能更详细、更可靠。然而,现有方法要么缺乏正式的可靠性保障,要么无法融入超参数空间中的结构化知识。本文提出基于可靠性图的帕累托检验(RG-PT),一种新的多目标超参数选择框架,能在控制错误发现率(FDR)的前提下,通过有向无环图(DAG)纳入已知的超参数间关系。图中边反映超参数间的可靠性与成本权衡,由布拉德利-特雷西(BT)排序模型从先验信息和预留数据中推断得出。实验表明,相较于学习后测试(LTT)和帕累托测试(PT)等方法,RG-PT在更高效探索超参数空间的同时显著提升性能。

原文摘要 · Abstract (English)

The selection of hyperparameters, such as prompt templates in large language models (LLMs), must often strike a balance between reliability and cost. In many cases, structural relationships between the expected reliability levels of the hyperparameters can be inferred from prior information and held-out data -- e.g., longer prompt templates may be more detailed and thus more reliable. However, existing hyperparameter selection methods either do not provide formal reliability guarantees or are unable to incorporate structured knowledge in the hyperparameter space. This paper introduces reliability graph-based Pareto testing (RG-PT), a novel multi-objective hyperparameter selection framework that maintains formal reliability guarantees in terms of false discovery rate (FDR), while accounting for known relationships among hyperparameters via a directed acyclic graph. Edges in the graph reflect expected reliability and cost trade-offs among hyperparameters, which are inferred via the Bradley-Terry (BT) ranking model from prior information and held-out data. Experimental evaluations demonstrate that RG-PT significantly outperforms existing methods such as learn-then-test (LTT) and Pareto testing (PT) through a more efficient exploration of the hyperparameter space.

超参数优化可靠性评估多目标学习

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