arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.01201cs.LGcs.CV
收藏

Average of Pruning:提高分布外检测的性能和稳定性

Average of Pruning: Improving Performance and Stability of Out-of-Distribution Detection

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院)

Zhen Cheng, Fei Zhu, Xu-Yao Zhang, Cheng-Lin Liu

摘要:

在开放世界中,检测分布外(OOD)输入一直是神经网络的关键问题。然而,OOD检测在训练过程中沿着优化轨迹的不稳定行为尚未被清楚地探索。在本文中,我们首先发现OOD检测的性能在训练过程中受到过拟合和不稳定性的影响:1)当训练误差接近零时,性能可能会下降,2)在训练的最后阶段,性能会急剧变化。基于我们的研究结果,我们提出了平均修剪(AoP),包括模型平均和修剪,以减轻不稳定的行为。具体而言,模型平均可以通过平滑景观来帮助实现稳定的性能,并且修剪被证明可以通过消除冗余特征来消除过拟合。在各种数据集和架构上进行了全面的实验,以验证我们的方法的有效性。

英文摘要:

Detecting Out-of-distribution (OOD) inputs have been a critical issue for neural networks in the open world. However, the unstable behavior of OOD detection along the optimization trajectory during training has not been explored clearly. In this paper, we first find the performance of OOD detection suffers from overfitting and instability during training: 1) the performance could decrease when the training error is near zero, and 2) the performance would vary sharply in the final stage of training. Based on our findings, we propose Average of Pruning (AoP), consisting of model averaging and pruning, to mitigate the unstable behaviors. Specifically, model averaging can help achieve a stable performance by smoothing the landscape, and pruning is certified to eliminate the overfitting by eliminating redundant features. Comprehensive experiments on various datasets and architectures are conducted to verify the effectiveness of our method.

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

↑