论文标题

可靠的OOD图像分类的多层表示学习

Multi-layer Representation Learning for Robust OOD Image Classification

论文作者

Ballas, Aristotelis, Diou, Christos

论文摘要

卷积神经网络已成为图像分类的规范。然而,在过去的几年中,它们在整个数据集中保持高精度的困难已变得显而易见。为了在实际情况和应用程序中利用此类模型,它们必须能够对看不见的数据提供值得信赖的预测。在本文中,我们认为从CNN的中间层中提取功能可以帮助模型的最终预测。具体而言,我们将HyperColumns方法适应RESNET-18,并在评估NICO数据集时发现模型的准确性显着提高。

Convolutional Neural Networks have become the norm in image classification. Nevertheless, their difficulty to maintain high accuracy across datasets has become apparent in the past few years. In order to utilize such models in real-world scenarios and applications, they must be able to provide trustworthy predictions on unseen data. In this paper, we argue that extracting features from a CNN's intermediate layers can assist in the model's final prediction. Specifically, we adapt the Hypercolumns method to a ResNet-18 and find a significant increase in the model's accuracy, when evaluating on the NICO dataset.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源