arXiv reaDer
Contrast with Reconstruct: Generative Pretraining による対照的な 3D 表現の学習
Contrast with Reconstruct: Contrastive 3D Representation Learning Guided by Generative Pretraining
主流の 3D 表現学習アプローチは、さまざまなダウンストリーム タスクのパフォーマンスの大幅な改善が達成された、対照的または生成的なモデリング プレテキスト タスクに基づいて構築されています。ただし、これら 2 つのパラダイムの方法を調査することにより、(i) 対照的なモデルは表現の過適合の問題に悩まされているデータを大量に消費することがわかりました。 (ii) 生成モデルには、対照的なモデルと比較して劣ったデータスケーリング容量を示すデータ充填の問題があります。これは、2 つのパラダイム間のパターンの違いにより自明ではない、両方のパラダイムのメリットを共有することによって、3D 表現を学習するように私たちを動機付けます。この論文では、これら 2 つのパラダイムを統合する再構成 (ReCon) との対比を提案します。 ReCon は、ジェネレーティブ モデリング ティーチャーとクロスモーダル コントラスト ティーチャーの両方からアンサンブル蒸留を通じて学習するようにトレーニングされています。ストップグラディエントを使用したクロスアテンションを通じて知識を伝達するエンコーダーデコーダースタイルの ReCon ブロックが提案されています。 ReCon は、3D 表現の学習において新しい最先端技術を達成しました。たとえば、ScanObjectNN で 91.26% の精度が達成されました。コードは https://github.com/qizekun/ReCon で公開されます。
Mainstream 3D representation learning approaches are built upon contrastive or generative modeling pretext tasks, where great improvements in performance on various downstream tasks have been achieved. However, by investigating the methods of these two paradigms, we find that (i) contrastive models are data-hungry that suffer from a representation over-fitting issue; (ii) generative models have a data filling issue that shows inferior data scaling capacity compared to contrastive models. This motivates us to learn 3D representations by sharing the merits of both paradigms, which is non-trivial due to the pattern difference between the two paradigms. In this paper, we propose contrast with reconstruct (ReCon) that unifies these two paradigms. ReCon is trained to learn from both generative modeling teachers and cross-modal contrastive teachers through ensemble distillation, where the generative student guides the contrastive student. An encoder-decoder style ReCon-block is proposed that transfers knowledge through cross attention with stop-gradient, which avoids pretraining over-fitting and pattern difference issues. ReCon achieves a new state-of-the-art in 3D representation learning, e.g., 91.26% accuracy on ScanObjectNN. Codes will be released at https://github.com/qizekun/ReCon.
updated: Sun Feb 05 2023 06:58:35 GMT+0000 (UTC)
published: Sun Feb 05 2023 06:58:35 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アソシエイト