arXiv:2502.16329cs.LG2025-02

用统一方法评估数据与训练集相似性,提升模型泛化可靠性。

Generalization is not a universal guarantee: Estimating similarity to training data with an ensemble out-of-distribution metric

  • 构建SAGE模型,通过低维嵌入空间比较新数据与训练集相似度。
  • 在MNIST/CIFAR-10上,过滤后模型性能显著提升,即使对训练/测试数据也有效。
  • 适用于分类与回归任务,可嵌入模型开发全流程,助力负责任AI。

机器学习模型在新数据上失效是限制AI系统可靠性的核心问题,部分原因在于缺乏简单可靠的手段来比较新数据与原始训练数据。本文提出一种无需依赖模型的标准化方法——用于泛化性估计的监督自编码器(SAGE),通过低维嵌入空间中的kNN距离、输入重建误差和任务表现,定义经验概率测度。以分类任务为例,在MNIST和CIFAR-10上验证:集成输出概率得分可有效区分变形图像与典型测试样本,且对渐进式变换具有鲁棒性。进一步扩展至非图像回归任务(UCI Abalone数据集)。所有情况下,经SAGE评分过滤后,模型性能均提升,包括对自身训练集和测试集数据。该分布外检测方法可贯穿模型构建与评估流程,推动负责任深度学习的发展。

原文摘要 · Abstract (English)

Failure of machine learning models to generalize to new data is a core problem limiting the reliability of AI systems, partly due to the lack of simple and robust methods for comparing new data to the original training dataset. We propose a standardized approach for assessing data similarity in a model-agnostic manner by constructing a supervised autoencoder for generalizability estimation (SAGE). We compare points in a low-dimensional embedded latent space, defining empirical probability measures for k-Nearest Neighbors (kNN) distance, reconstruction of inputs and task-based performance. As proof of concept for classification tasks, we use MNIST and CIFAR-10 to demonstrate how an ensemble output probability score can separate deformed images from a mixture of typical test examples, and how this SAGE score is robust to transformations of increasing severity. As further proof of concept, we extend this approach to a regression task using non-imaging data (UCI Abalone). In all cases, we show that out-of-the-box model performance increases after SAGE score filtering, even when applied to data from the model's own training and test datasets. Our out-of-distribution scoring method can be introduced during several steps of model construction and assessment, leading to future improvements in responsible deep learning implementation.

泛化性数据相似性OOD检测SAGE

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。