arXiv reaDer
Point-BERT:マスクされたポイントモデリングを使用した3Dポイントクラウドトランスフォーマーの事前トレーニング
Point-BERT: Pre-training 3D Point Cloud Transformers with Masked Point Modeling
トランスフォーマーを学習してBERTの概念を3Dポイントクラウドに一般化するための新しいパラダイムであるPoint-BERTを紹介します。 BERTに触発されて、点群トランスフォーマーを事前トレーニングするためのマスクポイントモデリング(MPM)タスクを考案しました。具体的には、最初にポイントクラウドをいくつかのローカルポイントパッチに分割し、離散変分オートエンコーダー(dVAE)を備えたポイントクラウドトークナイザーは、意味のあるローカル情報を含む離散ポイントトークンを生成するように設計されています。次に、入力ポイントクラウドのパッチをランダムにマスクして、バックボーントランスフォーマーにフィードします。事前トレーニングの目的は、トークナイザーによって取得されたポイントトークンの監視下で、マスクされた場所で元のポイントトークンを回復することです。広範な実験により、提案されたBERTスタイルの事前トレーニング戦略により、標準の点群トランスフォーマーのパフォーマンスが大幅に向上することが実証されています。事前トレーニング戦略を備えた純粋なTransformerアーキテクチャは、ModelNet40で93.8%の精度、ScanObjectNNの最も難しい設定で83.1%の精度を達成し、手作りの設計がはるかに少ない慎重に設計された点群モデルを上回っています。また、Point-BERTによって学習された表現が、新しいタスクやドメインにうまく移行することも示しています。ここで、モデルは、最先端の数ショットの点群分類タスクを大幅に進歩させます。コードと事前トレーニング済みモデルは、https://github.com/lulutang0608/Point-BERTで入手できます。
We present Point-BERT, a new paradigm for learning Transformers to generalize the concept of BERT to 3D point cloud. Inspired by BERT, we devise a Masked Point Modeling (MPM) task to pre-train point cloud Transformers. Specifically, we first divide a point cloud into several local point patches, and a point cloud Tokenizer with a discrete Variational AutoEncoder (dVAE) is designed to generate discrete point tokens containing meaningful local information. Then, we randomly mask out some patches of input point clouds and feed them into the backbone Transformers. The pre-training objective is to recover the original point tokens at the masked locations under the supervision of point tokens obtained by the Tokenizer. Extensive experiments demonstrate that the proposed BERT-style pre-training strategy significantly improves the performance of standard point cloud Transformers. Equipped with our pre-training strategy, we show that a pure Transformer architecture attains 93.8% accuracy on ModelNet40 and 83.1% accuracy on the hardest setting of ScanObjectNN, surpassing carefully designed point cloud models with much fewer hand-made designs. We also demonstrate that the representations learned by Point-BERT transfer well to new tasks and domains, where our models largely advance the state-of-the-art of few-shot point cloud classification task. The code and pre-trained models are available at https://github.com/lulutang0608/Point-BERT
updated: Mon Nov 29 2021 18:59:03 GMT+0000 (UTC)
published: Mon Nov 29 2021 18:59:03 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アソシエイト