Affiliations 

  • 1 School of Computer Science and Technology, Dongguan University of Technology, Dongguan, 523820, China
  • 2 School of Computer Science and Technology, Dongguan University of Technology, Dongguan, 523820, China. likuan@dgut.edu.cn
  • 3 School of Computer Science, University of Nottingham Malaysia, Selangor, 43500, Malaysia
Sci Rep, 2024 Nov 18;14(1):28444.
PMID: 39558072 DOI: 10.1038/s41598-024-79934-7

Abstract

Out-of-distribution (OOD) detection is a crucial problem in practice, especially, for the safe deployment of machine learning models in industrial settings. Previous work has used free energy as a score function and proposed a fine-tuning method that utilized OOD data in the training phase of the classification model, which achieves a higher performance on the OOD detection task compared with traditional methods. One key drawback, however, is that the loss function parameters are highly dependent on involved datasets, which means it cannot be dynamically adapted and implemented in others settings; in other words, the general ability of the energy score is considerably limited. In this work, our point of departure is to enlarge distinguishability between in-distribution features and OOD data. Consequently, we present a simple yet effective sparsity-regularized (SR) tuning framework for this purpose. Our framework has two types of workflows depending on if external OOD data is available, the complexity of the original training loss is sharply reduced by adopting this modification, meanwhile, the adapted ability and detection performance are enhanced. Also, we contribute a mini dataset as a light and efficient alternative of the previous large-scale one. In the experiments, we verify the effectiveness of our framework in a wide range of typical datasets along with common network architectures.

* Title and MeSH Headings from MEDLINE®/PubMed®, a database of the U.S. National Library of Medicine.