arXivDaily arXiv每日学术速递 周一至周五更新
今晚才能更新了~~大家休息下再来吧
arXiv 2403.03412cs.LGcs.CV
收藏

通过数据纯化与动态激活函数设计推进分布外检测

Advancing Out-of-Distribution Detection through Data Purification and Dynamic Activation Function Design

Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);University of Chinese Academy of Sciences(中国科学院大学);Zhejiang University(浙江大学)

Yingrui Ji, Yao Zhu, Zhigang Li, Jiansheng Chen, Yunlong Kong, Jingbo Chen

AI总结:

针对分布外样本检测难题,研究提出降噪数据集OOD-R与动态激活函数ActFun,分别优化数据质量与模型响应,显著提升OOD检测性能并凸显数据集完整性的重要性。

摘要:

在机器学习与深度学习的动态领域中,模型的鲁棒性与可靠性至关重要,尤其是在关键的现实应用场景中。该领域的一项核心挑战是应对分布外(Out-of-Distribution, OOD)样本,这类样本会显著提升模型误分类与不确定性的风险。本研究通过增强神经网络对OOD样本的检测与管理能力来应对这一挑战。我们提出了OOD-R(Out-of-Distribution-Rectified,分布外校正数据集),这是一套经过精心整理的开源数据集集合,具备增强的降噪特性。现有OOD数据集中的分布内(In-Distribution, ID)噪声会导致检测算法的评估结果不准确。基于这一认知,OOD-R引入了噪声过滤技术对数据集进行优化,确保OOD检测算法的评估结果更准确、更可靠。该方法不仅提升了数据的整体质量,还有助于更好地区分OOD与ID样本,可使模型准确率最高提升2.5%,假阳性率最低降低3.2%。此外,我们提出了ActFun,这是一种创新性方法,能够微调模型对不同输入的响应,从而提升特征提取的稳定性并减少特异性问题。ActFun通过策略性地降低隐藏单元的影响,解决了OOD检测中常见的模型过度自信问题,增强了模型更准确估计OOD不确定性的能力。在OOD-R数据集上应用ActFun后,模型性能得到了显著提升,其中GradNorm方法的AUROC提升了18.42%,Energy方法的FPR95降低了16.93%。总体而言,本研究不仅推动了OOD检测方法的发展,还强调了数据集完整性对于准确算法评估的重要性。

英文摘要:

In the dynamic realms of machine learning and deep learning, the robustness and reliability of models are paramount, especially in critical real-world applications. A fundamental challenge in this sphere is managing Out-of-Distribution (OOD) samples, significantly increasing the risks of model misclassification and uncertainty. Our work addresses this challenge by enhancing the detection and management of OOD samples in neural networks. We introduce OOD-R (Out-of-Distribution-Rectified), a meticulously curated collection of open-source datasets with enhanced noise reduction properties. In-Distribution (ID) noise in existing OOD datasets can lead to inaccurate evaluation of detection algorithms. Recognizing this, OOD-R incorporates noise filtering technologies to refine the datasets, ensuring a more accurate and reliable evaluation of OOD detection algorithms. This approach not only improves the overall quality of data but also aids in better distinguishing between OOD and ID samples, resulting in up to a 2.5\% improvement in model accuracy and a minimum 3.2\% reduction in false positives. Furthermore, we present ActFun, an innovative method that fine-tunes the model's response to diverse inputs, thereby improving the stability of feature extraction and minimizing specificity issues. ActFun addresses the common problem of model overconfidence in OOD detection by strategically reducing the influence of hidden units, which enhances the model's capability to estimate OOD uncertainty more accurately. Implementing ActFun in the OOD-R dataset has led to significant performance enhancements, including an 18.42\% increase in AUROC of the GradNorm method and a 16.93\% decrease in FPR95 of the Energy method. Overall, our research not only advances the methodologies in OOD detection but also emphasizes the importance of dataset integrity for accurate algorithm evaluation.

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

↑