arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.01459cs.CV
收藏

一种基于切片Wasserstein距离的类外分布检测方法

A sliced-Wasserstein distance-based approach for out-of-class-distribution detection

University of Virginia(弗吉尼亚大学)

Mohammad Shifat E Rabbi, Abu Hasnat Mohammad Rubaiyat, Yan Zhuang, Gustavo K Rohde

摘要:

对用于许多医学成像、图像处理和计算机视觉应用(诸如面部识别、医学诊断、字符识别和自动驾驶汽车等)的智能系统存在越来越大的兴趣。这些应用程序通常需要解决复杂的分类问题,涉及复杂的图像与未知的数据生成过程。除了依赖于特征工程和深度学习的当前分类方法的最近成功之外,还观察到它们的几个缺点,例如缺乏鲁棒性、可推广性和可解释性。这些方法通常需要大量的训练数据,计算昂贵,并且易受分布外样本的影响,例如,对抗攻击。最近,已经提出了一种准确的,数据高效的,计算高效的,和强大的基于运输的分类方法,它描述了一个生成的基于模型的问题制定和封闭形式的解决方案的一个特定类别的分类问题。然而,所有这些方法都缺乏在训练期间使用的类分布之外检测测试样本的机制。在现实世界中,收集的训练样本无法穷尽或覆盖所有类别,传统的分类方案无法有效地处理不可见的类别,这对于自动驾驶和医学成像诊断等安全关键系统来说尤其重要。在这项工作中,我们提出了一种方法来检测类外分布的基础上的分布的切片Wasserstein距离的Radon累积分布变换(R-CDT)子空间。我们在MNIST和两个医学图像数据集上测试了我们的方法,并报告了比没有类外分布检测程序的最新方法更好的准确性。

英文摘要:

There exist growing interests in intelligent systems for numerous medical imaging, image processing, and computer vision applications, such as face recognition, medical diagnosis, character recognition, and self-driving cars, among others. These applications usually require solving complex classification problems involving complex images with unknown data generative processes. In addition to recent successes of the current classification approaches relying on feature engineering and deep learning, several shortcomings of them, such as the lack of robustness, generalizability, and interpretability, have also been observed. These methods often require extensive training data, are computationally expensive, and are vulnerable to out-of-distribution samples, e.g., adversarial attacks. Recently, an accurate, data-efficient, computationally efficient, and robust transport-based classification approach has been proposed, which describes a generative model-based problem formulation and closed-form solution for a specific category of classification problems. However, all these approaches lack mechanisms to detect test samples outside the class distributions used during training. In real-world settings, where the collected training samples are unable to exhaust or cover all classes, the traditional classification schemes are unable to handle the unseen classes effectively, which is especially an important issue for safety-critical systems, such as self-driving and medical imaging diagnosis. In this work, we propose a method for detecting out-of-class distributions based on the distribution of sliced-Wasserstein distance from the Radon Cumulative Distribution Transform (R-CDT) subspace. We tested our method on the MNIST and two medical image datasets and reported better accuracy than the state-of-the-art methods without an out-of-class distribution detection procedure.

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

↑