Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

PCoTTA: Continual Test-Time Adaptation for Multi-Task Point Cloud Understanding

About

In this paper, we present PCoTTA, an innovative, pioneering framework for Continual Test-Time Adaptation (CoTTA) in multi-task point cloud understanding, enhancing the model's transferability towards the continually changing target domain. We introduce a multi-task setting for PCoTTA, which is practical and realistic, handling multiple tasks within one unified model during the continual adaptation. Our PCoTTA involves three key components: automatic prototype mixture (APM), Gaussian Splatted feature shifting (GSFS), and contrastive prototype repulsion (CPR). Firstly, APM is designed to automatically mix the source prototypes with the learnable prototypes with a similarity balancing factor, avoiding catastrophic forgetting. Then, GSFS dynamically shifts the testing sample toward the source domain, mitigating error accumulation in an online manner. In addition, CPR is proposed to pull the nearest learnable prototype close to the testing feature and push it away from other prototypes, making each prototype distinguishable during the adaptation. Experimental comparisons lead to a new benchmark, demonstrating PCoTTA's superiority in boosting the model's transferability towards the continually changing target domain.

Jincen Jiang, Qianyu Zhou, Yuhang Li, Xinkui Zhao, Meili Wang, Lizhuang Ma, Jian Chang, Jian Jun Zhang, Xuequan Lu• 2024

Related benchmarks

TaskDatasetResultRank
DenoisingModelNet40
CD (Round 1)21.4
19
DenoisingScanObjectNN
CD (Round 1)28.3
19
Point Cloud ReconstructionModelNet40 (test)
CD (Round 1)6.3
19
ReconstructionScanObjectNN
CD (Round 1)8.9
19
RegistrationModelNet40
CD (Round 1)15.4
19
RegistrationScanObjectNN
CD (Round 1)20.7
19
Question AnsweringXTREME QA Order 3 - Short Sequence (test)
EM54.37
16
Question AnsweringXTREME QA Order 5 - Long Sequence (test)
EM51
16
Question AnsweringXTREME QA Order 2 - Short Sequence (test)
EM48.78
16
Question AnsweringXTREME QA Average across all orders (test)
EM46.5
16
Showing 10 of 13 rows

Other info

Code

Follow for update