arXiv reaDer
分類重みの転送によるオブジェクト検出のスケーリング
Scaling Object Detection by Transferring Classification Weights
  大規模なオブジェクト検出データセットは、クラスと注釈の数の観点から常にサイズを増やしています。ただし、検出データセットで注釈が付けられているオブジェクトレベルのカテゴリの数は、画像レベルの分類ラベルよりも桁違いに小さくなっています。最先端のオブジェクト検出モデルは、監視された方法でトレーニングされ、これにより、検出できるオブジェクトクラスの数が制限されます。本論文では、分類ネットワークの重みから検出ネットワークの重みに知識を効果的かつ効率的に転送するための新しい重み転送ネットワーク(WTN)を提案し、ボックス監視なしで新しいクラスを検出できるようにします。最初に、バニラWTNのトレーニング中にアンダーフィットを抑制するために、入力および機能の正規化スキームを導入します。次に、再構成損失を使用してターゲットネットワークのすべてのクラスで分類ネットワークの情報を保存し、新規クラスへの一般化を保証するオートエンコーダーWTN(AE-WTN)を提案します。バニラWTNと比較して、AE-WTNは、500個の新規クラスと57個の新規クラスを含む2つのOpen Images評価セットで6%、200個の新規クラスを含むVisual Genome評価セットで25%の絶対パフォーマンスを獲得します。コードはhttps://github.com/xternalz/AE-WTNで入手できます。
Large scale object detection datasets are constantly increasing their size in terms of the number of classes and annotations count. Yet, the number of object-level categories annotated in detection datasets is an order of magnitude smaller than image-level classification labels. State-of-the art object detection models are trained in a supervised fashion and this limits the number of object classes they can detect. In this paper, we propose a novel weight transfer network (WTN) to effectively and efficiently transfer knowledge from classification network's weights to detection network's weights to allow detection of novel classes without box supervision. We first introduce input and feature normalization schemes to curb the under-fitting during training of a vanilla WTN. We then propose autoencoder-WTN (AE-WTN) which uses reconstruction loss to preserve classification network's information over all classes in the target latent space to ensure generalization to novel classes. Compared to vanilla WTN, AE-WTN obtains absolute performance gains of 6% on two Open Images evaluation sets with 500 seen and 57 novel classes respectively, and 25% on a Visual Genome evaluation set with 200 novel classes. The code is available at https://github.com/xternalz/AE-WTN.
updated: Sun Sep 15 2019 13:59:29 GMT+0000 (UTC)
published: Sun Sep 15 2019 13:59:29 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アソシエイト