参数化CAD的自监督学习中的掩码拓扑建模
计算机辅助设计(CAD)无处不在,几乎所有现代物体都是用可编辑的CAD工具设计的。然而,由于原生可编辑和参数化格式的CAD数据集匮乏(尤其是边界表示B-Rep),开发数据高效的方法变得非常重要。本文提出了一种新的自监督预训练任务——掩码拓扑建模(MTM),利用B-Rep特有的面邻接图结构,要求编码器重建该结构。MTM通过掩盖部分边并训练一个小型头部从编码器处理后的面特征预测每条掩盖边的凸性和曲线类型。结合MoCo风格的动量队列对比学习、基于B-Rep的增强、BFS连通面区域掩码重建目标,在ABC数据集和新生成数据集上预训练,展示了多项基准测试中的强劲表现。
这条 研究论文 信号说明,来自 arXiv cs.CV 的信息已经不只是单点新闻,而是值得放进产品、研究和行业判断里的趋势线索。