用空间上下文约束解决工业点云分割中因形状相似导致的误分类问题。
Resolving Primitive-Sharing Ambiguity in Long-Tailed TLS-Based Industrial MEP Point Cloud Segmentation via Spatial Context Constraints
- 通过边界和密度约束在损失层增强模糊区域判断,不修改主干网络。
- 尾类识别率提升21.7%,减速器、阀门等关键部件性能显著改善。
- 适合数字孪生与扫描转BIM应用,尤其关注安全组件精准识别。
基于地面激光扫描(TLS)的机械、电气与管道(MEP)点云分割中,减速器、阀门等安全关键组件长期存在误分类问题,阻碍工程知识可靠提取。根源在于极端类别不平衡(215:1)与几何模糊性双重困境:多数尾类与主类共享圆柱形基元,现有基于频率重加权的方法无法解决。本文提出空间上下文约束,利用邻域预测一致性来区分局部相似结构。方法在类别平衡损失基础上引入两种架构无关机制:边界-CB,基于熵的边界增强约束,融入MEP装配拓扑先验;密度-CB,基于密度的补偿约束,反映扫描依赖变化与TLS传感器物理特性。两者均作用于损失层,可无缝集成至现有流程。在工业3D数据集(61270万标注点,来自水处理设施)上,本方法达55.74% mIoU,超越三个代表性全监督主干基线(39.83%-52.48% mIoU),尾类性能相对提升21.7%(29.59% vs. 24.32%),头部类别精度保持在88.14%。存在基元共享模糊性的组件显著受益:减速器从0%提升至21.12% IoU,阀门相对提升24.3%。结果表明,空间上下文约束有效降低目标工业场景下的基元混淆错误,支持更可靠的安全部件识别,适用于数字孪生与扫描转建筑信息模型(Scan-to-BIM)应用。代码开源:https://github.com/PointCloudYC/LongTail3D.git。
原文摘要 · Abstract (English)
In terrestrial laser scanning (TLS)-based mechanical, electrical, and plumbing (MEP) point cloud segmentation, safety-critical components such as reducers and valves are persistently misclassifed, blocking reliable engineering knowledge extraction. This stems from a dual crisis--extreme class imbalance (215:1) compounded by geometric ambiguity, since most tail classes share cylindrical primitives with dominant head classes--that existing frequencybased re-weighting methods cannot resolve. We propose spatial context constraints that exploit neighborhood prediction consistency to disambiguate locally similar structures. Our approach extends Class-Balanced (CB) Loss with two architecture-agnostic mechanisms: Boundary-CB, an entropy-based constraint that emphasizes ambiguous boundaries and encodes an MEP assemblytopology prior, and Density-CB, a density-based constraint that compensates for scan-dependent variations and encodes TLS sensor-physics knowledge. Both operate at the loss level and integrate into existing pipelines without backbone modifcations. On the Industrial3D dataset (612.7M labelled points from water treatment facilities), our method achieves 55.74% mIoU, exceeding the strongest of three representative fully supervised backbone baselines (39.83-52.48% mIoU), with a 21.7% relative improvement on tail-class performance (29.59% vs. 24.32%) while preserving head-class accuracy (88.14%). Components with primitive-sharing ambiguity show strong gains: reducer improves from 0% to 21.12% IoU, and valve improves by 24.3% relative. These results show that spatial context constraints reduce primitive-sharing errors in the target industrial MEP setting and support more reliable identifcation of safety-critical components for Digital Twin and Scan-to-BIM applications. Code: https://github.com/PointCloudYC/LongTail3D.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。