vix.ing · top · new · best · stats · spec

Selective Classification for Deep Neural Networks

2017/05/23 by Yonatan Geifman, Geifman, Yonatan, Ran El‐Yaniv +2 · 1 voice · 65 citations
Computer Science · #Anomaly Detection Techniques and Applications #Artificial Intelligence (cs.AI) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning and Data Classification #Neural Networks and Applications #cs.AI #cs.LG

paper · pdf · doi:10.48550/arxiv.1705.08500

openalex publication_date 2017/05/23 · arxiv published 2017/05/23 · arxiv updated 2017/06/01 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Selective classification techniques (also known as reject option) have not yet been considered in the context of deep neural networks (DNNs). These techniques can potentially significantly improve DNNs prediction performance by trading-off coverage. In this paper we propose a method to construct a selective classifier given a trained neural network. Our method allows a user to set a desired risk level. At test time, the classifier rejects instances as needed, to grant the desired risk (with high probability). Empirical results over CIFAR and ImageNet convincingly demonstrate the viability of our method, which opens up possibilities to operate DNNs in mission-critical applications. For example, using our method an unprecedented 2% error in top-5 ImageNet classification can be guaranteed with probability 99.9%, and almost 60% test coverage.

Citations

Cited by

Discussions

Related