DeepAI
Log In Sign Up

Conditional Variational Capsule Network for Open Set Recognition

04/19/2021
by   Yunrui Guo, et al.
0

In open set recognition, a classifier has to detect unknown classes that are not known at training time. In order to recognize new classes, the classifier has to project the input samples of known classes in very compact and separated regions of the features space in order to discriminate outlier samples of unknown classes. Recently proposed Capsule Networks have shown to outperform alternatives in many fields, particularly in image recognition, however they have not been fully applied yet to open-set recognition. In capsule networks, scalar neurons are replaced by capsule vectors or matrices, whose entries represent different properties of objects. In our proposal, during training, capsules features of the same known class are encouraged to match a pre-defined gaussian, one for each class. To this end, we use the variational autoencoder framework, with a set of gaussian prior as the approximation for the posterior distribution. In this way, we are able to control the compactness of the features of the same class around the center of the gaussians, thus controlling the ability of the classifier in detecting samples from unknown classes. We conducted several experiments and ablation of our model, obtaining state of the art results on different datasets in the open set recognition and unknown detection tasks.

READ FULL TEXT

page 1

page 2

page 3

page 4

03/19/2020

Conditional Gaussian Distribution Learning for Open Set Recognition

Deep neural networks have achieved state-of-the-art performance in a wid...
08/12/2020

Open Set Recognition with Conditional Probabilistic Generative Models

Deep neural networks have made breakthroughs in a wide range of visual u...
10/28/2022

Towards Few-Shot Open-Set Object Detection

Open-set object detection (OSOD) aims to detect the known categories and...
06/09/2022

OOD Augmentation May Be at Odds with Open-Set Recognition

Despite advances in image classification methods, detecting the samples ...
11/20/2022

PartCom: Part Composition Learning for 3D Open-Set Recognition

3D recognition is the foundation of 3D deep learning in many emerging fi...
03/27/2020

Hybrid Models for Open Set Recognition

Open set recognition requires a classifier to detect samples not belongi...
07/11/2018

Variational Capsules for Image Analysis and Synthesis

A capsule is a group of neurons whose activity vector models different p...

Code Repositories

cvaecaposr

Code for the Paper: "Conditional Variational Capsule Network for Open Set Recognition", Y. Guo, G. Camporese, W. Yang, A. Sperduti, L. Ballan, arXiv:2104.09159, 2021.


view repo