PointGAC: Geometric-Aware Codebook for Masked Point Cloud Modeling

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Abiao, Lv, Chenlei, Fang, Yuming, Zuo, Yifan, Zhang, Jian, Mei, Guofeng
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915376200155136
author Li, Abiao
Lv, Chenlei
Fang, Yuming
Zuo, Yifan
Zhang, Jian
Mei, Guofeng
author_facet Li, Abiao
Lv, Chenlei
Fang, Yuming
Zuo, Yifan
Zhang, Jian
Mei, Guofeng
contents Most masked point cloud modeling (MPM) methods follow a regression paradigm to reconstruct the coordinate or feature of masked regions. However, they tend to over-constrain the model to learn the details of the masked region, resulting in failure to capture generalized features. To address this limitation, we propose \textbf{\textit{PointGAC}}, a novel clustering-based MPM method that aims to align the feature distribution of masked regions. Specially, it features an online codebook-guided teacher-student framework. Firstly, it presents a geometry-aware partitioning strategy to extract initial patches. Then, the teacher model updates a codebook via online k-means based on features extracted from the complete patches. This procedure facilitates codebook vectors to become cluster centers. Afterward, we assigns the unmasked features to their corresponding cluster centers, and the student model aligns the assignment for the reconstructed masked features. This strategy focuses on identifying the cluster centers to which the masked features belong, enabling the model to learn more generalized feature representations. Benefiting from a proposed codebook maintenance mechanism, codebook vectors are actively updated, which further increases the efficiency of semantic feature learning. Experiments validate the effectiveness of the proposed method on various downstream tasks. Code is available at https://github.com/LAB123-tech/PointGAC
format Preprint
id arxiv_https___arxiv_org_abs_2507_04801
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PointGAC: Geometric-Aware Codebook for Masked Point Cloud Modeling
Li, Abiao
Lv, Chenlei
Fang, Yuming
Zuo, Yifan
Zhang, Jian
Mei, Guofeng
Computer Vision and Pattern Recognition
Most masked point cloud modeling (MPM) methods follow a regression paradigm to reconstruct the coordinate or feature of masked regions. However, they tend to over-constrain the model to learn the details of the masked region, resulting in failure to capture generalized features. To address this limitation, we propose \textbf{\textit{PointGAC}}, a novel clustering-based MPM method that aims to align the feature distribution of masked regions. Specially, it features an online codebook-guided teacher-student framework. Firstly, it presents a geometry-aware partitioning strategy to extract initial patches. Then, the teacher model updates a codebook via online k-means based on features extracted from the complete patches. This procedure facilitates codebook vectors to become cluster centers. Afterward, we assigns the unmasked features to their corresponding cluster centers, and the student model aligns the assignment for the reconstructed masked features. This strategy focuses on identifying the cluster centers to which the masked features belong, enabling the model to learn more generalized feature representations. Benefiting from a proposed codebook maintenance mechanism, codebook vectors are actively updated, which further increases the efficiency of semantic feature learning. Experiments validate the effectiveness of the proposed method on various downstream tasks. Code is available at https://github.com/LAB123-tech/PointGAC
title PointGAC: Geometric-Aware Codebook for Masked Point Cloud Modeling
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.04801