English

Exploring 3D Dataset Pruning

Computer Vision and Pattern Recognition 2026-03-03 v1 Machine Learning

Abstract

Dataset pruning has been widely studied for 2D images to remove redundancy and accelerate training, while particular pruning methods for 3D data remain largely unexplored. In this work, we study dataset pruning for 3D data, where its observed common long-tail class distribution nature make optimization under conventional evaluation metrics Overall Accuracy (OA) and Mean Accuracy (mAcc) inherently conflicting, and further make pruning particularly challenging. To address this, we formulate pruning as approximating the full-data expected risk with a weighted subset, which reveals two key errors: coverage error from insufficient representativeness and prior-mismatch bias from inconsistency between subset-induced class weights and target metrics. We propose representation-aware subset selection with per-class retention quotas for long-tail coverage, and prior-invariant teacher supervision using calibrated soft labels and embedding-geometry distillation. The retention quota also serves as a switch to control the OA-mAcc trade-off. Extensive experiments on 3D datasets show that our method can improve both metrics across multiple settings while adapting to different downstream preferences. Our code is available at https://github.com/XiaohanZhao123/3D-Dataset-Pruning.

Keywords

Cite

@article{arxiv.2603.00651,
  title  = {Exploring 3D Dataset Pruning},
  author = {Xiaohan Zhao and Xinyi Shang and Jiacheng Liu and Zhiqiang Shen},
  journal= {arXiv preprint arXiv:2603.00651},
  year   = {2026}
}

Comments

Code: https://github.com/XiaohanZhao123/3D-Dataset-Pruning

R2 v1 2026-07-01T10:57:12.706Z