跳到正文
OpenReview · Representation learning· Yunzhong Lou; Xueyang Li; Haotian Chen; Xiangdong Zhou·· 2023-10-21精选AI 评分74

BRep-BERT:基于子图节点对比学习的边界表示 BERT 预训练

BRep-BERT: Pre-training Boundary Representation BERT with Sub-graph Node Contrastive Learning

AI 导读

基于摘要分析。该研究面向计算机辅助设计(CAD)中的边界表示(B-Rep)实体特征学习,针对大规模标注数据及任务专用标签稀缺的问题,提出无监督预训练方法 BRep-BERT。其贡献是将几何与结构信息引入离散实体表示、掩码建模和子图对比学习,而非直接研究 EEG 或 BCI。 方法上,GNN Tokenizer 为实体生成包含几何与结构语义的离散标签,再依据结构关系构造实体序列,通过 Masked Entity Modeling(MEM)开展预训练。针对大型几何模型中的注意力稀疏问题,作者在注意力模块中引入图结构信息和可学习相对位置编码,并结合几何子图与多层级对比学习强化区域特征。摘要未披露具体损失形式、子图采样方式或各组件的独立贡献。 作者报告,在多个 B-Rep 数据集的全量数据训练与少样本学习任务中,该方法达到当时最优表现,少样本场景的优势尤其明显;但材料未给出数据集名称、数据划分、基线、指标或数值,实验协议、消融及统计信息尚未验证。摘要称代码将发布,不能据此认定代码与预训练权重已可获取。 平台推测(待验证):其潜在可迁移点是结构感知的掩码预训练与局部子图对比,而非 CAD 几何标签本身。假设 EEG 通道关系能够形成稳定且有意义的图,可探索将实体改为通道—时间片单元;Tokenizer、相对位置编码和子图采样均需适配时序及电极布局。EEG 低信噪比、跨被试差异、通道变化及小数据条件可能使离散标签与局部结构不稳定。可先在固定划分、相同骨干与训练预算下,比较仅掩码预训练与加入结构编码及子图对比的方案,检验跨被试迁移收益;这只是可证伪假设,已有 EEG 相关工作尚未检索确认。

阅读价值

值得关注其将几何拓扑离散化、Masked Entity Modeling 与子图对比学习结合的表示学习机制,但摘要不足以核验性能优势,且 CAD 结果不能直接作为 EEG/BCI 有效性的证据。

来源:OpenReview · Representation learning · openreview.net