Pretrained graph neural network for embedding semantic, spatial, and topological data in building information models
Jin Han, Xinzheng Lu, Jia‐Rui Lin
Abstract
Large foundation models have demonstrated significant advantages in civil engineering, but they primarily focus on textual and visual data, overlooking the rich semantic, spatial, and topological features in building information modeling (BIM) models. Therefore, this study develops the first large-scale graph neural network, BIGNet, to learn and reuse multidimensional design features embedded in BIM models. First, a scalable graph representation is introduced to encode the “semantic-spatial-topological” features of BIM components, and a dataset with nearly 1 million nodes and 3.5 million edges is created. Subsequently, BIGNet is proposed by introducing a new message-passing mechanism to GraphMAE2 and further pretrained with a node masking strategy. Finally, BIGNet is evaluated in various transfer learning tasks for BIM-based design checking. Results show that: (1) homogeneous graph representation outperforms heterogeneous graph in learning design features, (2) considering local spatial relationships in a 30 cm radius enhances performance, and (3) BIGNet with graph attention network-based feature extraction achieves the best transfer learning results. This innovation leads to a 72.7% improvement in average F1-score over non-pretrained models, demonstrating its effectiveness in learning and transferring BIM design features and facilitating their automated application in future design and lifecycle management.