论文标题
partcom:3D开放式识别的部分组成学习
PartCom: Part Composition Learning for 3D Open-Set Recognition
论文作者
论文摘要
3D识别是在许多新兴领域(例如自动驾驶和机器人)中进行3D深度学习的基础。存在3D方法主要集中于识别固定类别的固定类别,并在测试过程中忽略了可能的未知类别。这些未知类别可能会在安全至关重要的应用中造成严重事故,即自动驾驶。在这项工作中,我们首次尝试解决3D开放式识别(OSR),以便分类器可以识别已知类别并注意未知类别。我们分析了3D域中的开放式风险,并指出使现有方法在3D OSR任务上的表现不佳的过度自信和不足的问题。为了解决上述问题,我们提出了一种新型的基于零件原型的OSR方法,名为PartCom。我们使用零件原型代表3D形状作为零件组成,因为零件组成可以代表形状的整体结构,并可以帮助区分不同的已知类别和未知类别。然后,我们对部分原型制定了两个约束,以确保其有效性。为了进一步降低开放式风险,我们通过混合不同类别的零件复合特征,设计了一个PUFS模块,以将未知特征作为未知样品的代表综合。我们基于CAD形状数据集和扫描形状数据集对三种3D OSR任务进行实验。广泛的实验表明,我们的方法在对已知类别和未知类别的分类方面具有强大的作用,并且在所有3D OSR任务上都可以比SOTA基准获得更好的结果。该项目将发布。
3D recognition is the foundation of 3D deep learning in many emerging fields, such as autonomous driving and robotics.Existing 3D methods mainly focus on the recognition of a fixed set of known classes and neglect possible unknown classes during testing. These unknown classes may cause serious accidents in safety-critical applications, i.e. autonomous driving. In this work, we make a first attempt to address 3D open-set recognition (OSR) so that a classifier can recognize known classes as well as be aware of unknown classes. We analyze open-set risks in the 3D domain and point out the overconfidence and under-representation problems that make existing methods perform poorly on the 3D OSR task. To resolve above problems, we propose a novel part prototype-based OSR method named PartCom. We use part prototypes to represent a 3D shape as a part composition, since a part composition can represent the overall structure of a shape and can help distinguish different known classes and unknown ones. Then we formulate two constraints on part prototypes to ensure their effectiveness. To reduce open-set risks further, we devise a PUFS module to synthesize unknown features as representatives of unknown samples by mixing up part composite features of different classes. We conduct experiments on three kinds of 3D OSR tasks based on both CAD shape dataset and scan shape dataset. Extensive experiments show that our method is powerful in classifying known classes and unknown ones and can attain much better results than SOTA baselines on all 3D OSR tasks. The project will be released.