Skip to main content
QUICK REVIEW

[論文レビュー] Look-into-Object: Self-supervised Structure Modeling for Object Recognition

Mohan Zhou, Yalong Bai|arXiv (Cornell University)|Mar 31, 2020
Advanced Neural Network Applications参考文献 39被引用数 12
ひとこと要約

本稿では、オブジェクト全体の構造を2つのプラグインモジュールであるオブジェクト拡大学習(OEL)と空間的文脈学習(SCL)を用いてモデル化することで、オブジェクト認識を向上させる自己教師付きフレームワークであるLook-into-Object(LIO)を提案する。OELはオブジェクトの全範囲を局所化し、SCLはオブジェクト部品間の相対的な空間的関係を予測する。本手法は、クラス内領域相関と相対座標予測を用いた自己教師付き学習により、追加のアノテーションや推論コストなしにImageNet、CUB、Cars、Aircraftベンチマークで顕著な精度向上を達成する。

ABSTRACT

Most object recognition approaches predominantly focus on learning discriminative visual patterns while overlooking the holistic object structure. Though important, structure modeling usually requires significant manual annotations and therefore is labor-intensive. In this paper, we propose to "look into object" (explicitly yet intrinsically model the object structure) through incorporating self-supervisions into the traditional framework. We show the recognition backbone can be substantially enhanced for more robust representation learning, without any cost of extra annotation and inference speed. Specifically, we first propose an object-extent learning module for localizing the object according to the visual patterns shared among the instances in the same category. We then design a spatial context learning module for modeling the internal structures of the object, through predicting the relative positions within the extent. These two modules can be easily plugged into any backbone networks during training and detached at inference time. Extensive experiments show that our look-into-object approach (LIO) achieves large performance gain on a number of benchmarks, including generic object recognition (ImageNet) and fine-grained object recognition tasks (CUB, Cars, Aircraft). We also show that this learning paradigm is highly generalizable to other tasks such as object detection and segmentation (MS COCO). Project page: https://github.com/JDAI-CV/LIO.

研究の動機と目的

  • 現在のオブジェクト認識モデルが判別的視覚パターンに注目する一方で、オブジェクト全体の構造を無視するという限界を是正すること。
  • 手動アノテーションを必要とせず、推論時間も増加させない自己教師付き学習フレームワークを構築すること。
  • 汎用的でプラグイン可能なモジュールを用いて、深層バックボーンへの構造的インダクティブバイアスの注入により表現学習を向上させること。
  • オブジェクトの範囲と内部空間的文脈をモデル化することで、特徴マップの局所化と分類のロバストネスが向上することを示すこと。
  • MS COCOにおける検出およびセグメンテーションを含む多様なタスクにわたる一般化を検証すること。

提案手法

  • オブジェクト拡大学習(OEL)は、同じカテゴリに属するポジティブ画像との間で領域レベルの相関を測定することで、クエリ画像のオブジェクト範囲の疑似マスクを構築する。この手法は画像レベルのラベルのみを用いる。
  • OELモジュールは、バックボーンが疑似マスクをエンドツーエンドで回帰できるように訓練することで、ボクセルボックスやセグメンテーションアノテーションなしにオブジェクト全体の範囲を局所化する能力を学習可能にする。
  • 空間的文脈学習(SCL)は、オブジェクト範囲内に存在する領域ペア間の相対的な極座標を予測することで、自己教師付きの方法で内部構造的構成をモデル化する。
  • SCLは二重ストリームの対照的学習設定を用い、オブジェクト部品の構造的表現を学習するようネットワークを促進する。
  • OELおよびSCLの両モジュールは軽量であり、推論時に分離可能であるため、元のバックボーンのモデルサイズと推論速度を維持する。
  • 本フレームワークは任意の標準的な分類バックボーン(例:ResNet、ResNeXt)と統合可能であり、分類用の交差エントロピー損失に加え、OELおよびSCL用の補助損失を組み合わせた損失関数で訓練される。

実験結果

リサーチクエスチョン

  • RQ1オブジェクト範囲と内部空間的構造の自己教師付き学習が、追加のアノテーションなしにオブジェクト認識性能を向上させることができるか?
  • RQ2オブジェクト全体の範囲をモデル化することで、特徴マップにおける判別的領域の局所化にどのような影響を与えるか?
  • RQ3オブジェクト部品間の相対的空間的関係を予測することで、特徴表現の質がどの程度向上するか?
  • RQ4提案されたモジュールが標準的なバックボーンに効果的にプラグイン可能で、多様な認識タスクに一般化可能か?
  • RQ5自己教師付き構造モデリングにより、オブジェクト検出やインスタンスセグメンテーションなどの下流タスクで性能向上が達成されるか?

主な発見

  • CUBデータセットではLIOがトップ-1精度87.31%を達成し、ResNet-50(85.50%)を1.81%上回るという顕著な向上を示した。これは細分化認識において顕著な向上を示している。
  • CarsデータセットではLIOが93.89%の精度を達成し、ResNet-50を1.16ポイント上回り、視覚的変動に対して強いロバストネスを示した。
  • アブレーションスタディにより、OELおよびSCLが独立的かつ補完的に寄与していることが確認された。特にSCLはオブジェクト範囲内における判別的領域を体系的にマイニングする有効な方法を提供した。
  • OELにバッチあたり3枚のポジティブ画像を使用するだけで、計算コストと構造学習品質のバランスがほぼ最適化された性能が得られた。
  • 疑似マスクを正例のセマンティックマスクに置き換えても、精度はわずか0.1%低下するのみ(CUBでは87.3%対87.4%)であり、自己教師付きマスク生成の高品質さを示している。
  • 追加パラメータは0.26百万(ResNet-50の1.01%)にとどまり、両モジュールは推論時に削除可能であるため、モデル効率が維持された。

より良い研究を、今すぐ始めましょう

論文の読解から最終レビューまで、研究時間を劇的に削減しましょう。

クレジットカード登録不要

このレビューはAIが作成し、人間の編集者が確認しました。