提出高效结构保持的网格下采样方法,提升医学3D形状分类性能。
Benchmark-Ready 3D Anatomical Shape Classification
- 设计非学习型图池化PSPooling,基于几何邻近预计算节点对应关系。
- 在低标注数据下实现更高重建精度与分类准确率,优于现有方法。
- 配套新基准数据集MedShapeNet19,支持可复现的医学3D形状研究。
医学3D形状分类进展受限于网格数据复杂性及缺乏标准化评测基准,亟需稳健的学习方法与可复现评估。本文通过自监督图自动编码,提出迈向临床与评测可用的解剖形状分类的两项关键步骤。我们引入预计算结构池化(PSPooling),一种非学习型网格池化算子,用于高效且结构保持的图粗化。PSPooling基于几何邻近预计算节点对应集,实现并行化、可逆的池化与反池化操作,并保证支撑结构。该设计避免了选择类方法的稀疏性与重构问题,也规避了边收缩方法的串行开销,特别适用于高分辨率医学网格。为验证有效性,我们将PSPooling整合至自监督图自动编码器中,从无标签表面网格学习解剖感知表征。在新构建的MedShapeNet19上进行评估——该数据集源自MedShapeNet,包含19个解剖类别,具有标准训练/验证/测试划分。实验表明,PSPooling在低标注情形下显著提升重建保真度与分类准确率,建立了医学3D形状学习的强大基线。我们期望MedShapeNet19能成为广泛采用的解剖形状分类基准,推动医学3D形状分析研究。代码与数据详见:https://github.com/TomasKrsicka/MedShapeNet19-PSPooling。
原文摘要 · Abstract (English)
Progress in anatomical 3D shape classification is limited by the complexity of mesh data and the lack of standardized benchmarks, highlighting the need for robust learning methods and reproducible evaluation. We introduce two key steps toward clinically and benchmark-ready anatomical shape classification via self-supervised graph autoencoding. We propose Precomputed Structural Pooling (PSPooling), a non-learnable mesh pooling operator designed for efficient and structure-preserving graph coarsening in 3D anatomical shape analysis. PSPooling precomputes node correspondence sets based on geometric proximity, enabling parallelizable and reversible pooling and unpooling operations with guaranteed support structure. This design avoids the sparsity and reconstruction issues of selection-based methods and the sequential overhead of edge contraction approaches, making it particularly suitable for high-resolution medical meshes. To demonstrate its effectiveness, we integrate PSPooling into a self-supervised graph autoencoder that learns anatomy-aware representations from unlabeled surface meshes. We evaluate the downstream benefits on MedShapeNet19, a new curated benchmark dataset we derive from MedShapeNet, consisting of 19 anatomical classes with standardized training, validation, and test splits. Experiments show that PSPooling significantly improves reconstruction fidelity and classification accuracy in low-label regimes, establishing a strong baseline for medical 3D shape learning. We hope that MedShapeNet19 will serve as a widely adopted benchmark for anatomical shape classification and further research in medical 3D shape analysis. Access the complete codebase, model weights, and dataset information here: https://github.com/TomasKrsicka/MedShapeNet19-PSPooling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。