arXiv reaDer
クラス認識選択的損失を使用した部分注釈によるマルチラベル分類
Multi-label Classification with Partial Annotations using Class-aware Selective Loss
大規模なマルチラベル分類データセットは、一般的に、そしておそらく必然的に、部分的に注釈が付けられます。つまり、サンプルごとにラベルの小さなサブセットのみに注釈が付けられます。欠落しているラベルを処理するためのさまざまな方法は、モデルにさまざまなプロパティを引き起こし、その精度に影響を与えます。この作業では、部分的なラベル付けの問題を分析し、2つの重要なアイデアに基づいて解決策を提案します。まず、注釈のないラベルは、データセット全体のクラス分布と特定のデータサンプルの特定のラベル尤度の2つの確率量に従って選択的に処理する必要があります。専用の一時モデルを使用してクラス分布を推定することを提案し、データセットの部分的な注釈を使用して計算された単純な推定よりも効率が向上していることを示します。次に、ターゲットモデルのトレーニング中に、専用の非対称損失を使用して、元々注釈が付けられていないラベルよりも注釈が付けられたラベルの寄与を強調します。私たちの斬新なアプローチにより、OpenImagesデータセットで最先端の結果を達成します(たとえば、V6で87.3 mAPに到達します)。さらに、LVISとシミュレートされたCOCOで実施された実験は、私たちのアプローチの有効性を示しています。コードはhttps://github.com/Alibaba-MIIL/PartialLabelingCSLで入手できます。
Large-scale multi-label classification datasets are commonly, and perhaps inevitably, partially annotated. That is, only a small subset of labels are annotated per sample. Different methods for handling the missing labels induce different properties on the model and impact its accuracy. In this work, we analyze the partial labeling problem, then propose a solution based on two key ideas. First, un-annotated labels should be treated selectively according to two probability quantities: the class distribution in the overall dataset and the specific label likelihood for a given data sample. We propose to estimate the class distribution using a dedicated temporary model, and we show its improved efficiency over a naive estimation computed using the dataset's partial annotations. Second, during the training of the target model, we emphasize the contribution of annotated labels over originally un-annotated labels by using a dedicated asymmetric loss. With our novel approach, we achieve state-of-the-art results on OpenImages dataset (e.g. reaching 87.3 mAP on V6). In addition, experiments conducted on LVIS and simulated-COCO demonstrate the effectiveness of our approach. Code is available at https://github.com/Alibaba-MIIL/PartialLabelingCSL.
updated: Thu Oct 21 2021 08:10:55 GMT+0000 (UTC)
published: Thu Oct 21 2021 08:10:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト