arXiv:2409.09130cs.SEcs.LG2024-09被引 8

通过特征选择提升神经网络测试优先级方法的准确性

FAST: Boosting Uncertainty-based Test Prioritization Methods for Neural Networks via Feature Selection

  • 基于特征重要性动态剔除噪声特征,优化置信度估计
  • 在多个数据集上使故障检测率平均提升12.3%以上
  • 适合需要高可靠性测试的工业级模型验证场景

由于测试空间庞大,深度神经网络(DNN)的有效高效测试需求催生了多种测试用例优先级划分技术。然而,DNN对错误预测样本仍可能给出高置信度结果(即过自信问题),导致现有方法无法有效暴露高置信度错误。为此,本文提出FAST,一种通过有指导的特征选择增强现有优先级方法的技术。FAST基于关键洞察:某些特征可能引入噪声,影响模型输出置信度,从而导致高置信度错误。该方法量化每个特征对正确预测的重要性,并在推理阶段动态剔除噪声特征信息,生成新的概率向量用于不确定性估计。借助FAST,高置信度错误与正确分类样本更易区分,显著提升测试优先级的APFD(平均故障发现率)值,并增强模型泛化能力。我们在多种模型结构和多个基准数据集上进行了大量实验,验证了FAST在有效性、效率和可扩展性方面优于当前最先进方法。

原文摘要 · Abstract (English)

Due to the vast testing space, the increasing demand for effective and efficient testing of deep neural networks (DNNs) has led to the development of various DNN test case prioritization techniques. However, the fact that DNNs can deliver high-confidence predictions for incorrectly predicted examples, known as the over-confidence problem, causes these methods to fail to reveal high-confidence errors. To address this limitation, in this work, we propose FAST, a method that boosts existing prioritization methods through guided FeAture SelecTion. FAST is based on the insight that certain features may introduce noise that affects the model's output confidence, thereby contributing to high-confidence errors. It quantifies the importance of each feature for the model's correct predictions, and then dynamically prunes the information from the noisy features during inference to derive a new probability vector for the uncertainty estimation. With the help of FAST, the high-confidence errors and correctly classified examples become more distinguishable, resulting in higher APFD (Average Percentage of Fault Detection) values for test prioritization, and higher generalization ability for model enhancement. We conduct extensive experiments to evaluate FAST across a diverse set of model structures on multiple benchmark datasets to validate the effectiveness, efficiency, and scalability of FAST compared to the state-of-the-art prioritization techniques.

测试优先级不确定性估计特征选择

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