Abstract
In computer-aided design (CAD), software tools support design engineers during the modeling of assemblies, i.e., products that consist of multiple components. Selecting the right components is a cumbersome task for design engineers as they have to pick from a large number of possibilities. Therefore, we propose to analyze a data set of past assemblies composed of components from the same component catalog, represented as connected, undirected graphs of components, in order to suggest the next needed component. In terms of graph machine learning, we formulate this as graph classification problem where each class corresponds to a component ID from a catalog and the models are trained to predict the next required component. In addition to pretraining of component embeddings, we recursively decompose the graphs to obtain data instances in a self-supervised fashion without imposing any node insertion order. Our results indicate that models based on graph convolution networks and graph attention networks achieve high predictive performance, reducing the cognitive load of choosing among 2,000 and 3,000 components by recommending the ten most likely components with 82–92% accuracy, depending on the chosen catalog.
Research leading to this paper was funded by the Bavarian State Ministry for Economic Affairs, Regional Development and Energy (StMWi), Project KOGNIA, in cooperation with Cadenas GmbH.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
Notes
- 1.
Mating conditions define relative positions of components to each other.
- 2.
Deep Graph Library https://www.dgl.ai/.
- 3.
References
Bakarov, A.: A survey of word embeddings evaluation methods (2018). https://doi.org/10.48550/ARXIV.1801.09536
Bronstein, M.M., Bruna, J., Cohen, T., Veličković, P.: Geometric deep learning: grids, groups, graphs, geodesics, and gauges (2021). https://doi.org/10.48550/ARXIV.2104.13478
Cunningham, J.D., Simpson, T.W., Tucker, C.S.: An investigation of surrogate models for efficient performance-based decoding of 3D point clouds. J. Mech. Des. 141(12) (2019). https://doi.org/10.1115/1.4044597
Goodfellow, I., Bengio, Y., Courville, A.: Deep Learning. MIT Press, Cambridge (2016)
Grover, A., Leskovec, J.: Node2vec: scalable feature learning for networks. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. KDD ’16, pp. 855–864. Association for Computing Machinery, New York (2016). https://doi.org/10.1145/2939672.2939754
Guo, X., Zhao, L.: A systematic survey on deep generative models for graph generation (2020). https://doi.org/10.48550/ARXIV.2007.06686
Guo, X., Li, W., Iorio, F.: Convolutional neural networks for steady flow approximation. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. KDD ’16, pp. 481–490. Association for Computing Machinery, New York (2016). https://doi.org/10.1145/2939672.2939738
Hamilton, W., Ying, Z., Leskovec, J.: Inductive representation learning on large graphs. In: Advances in Neural Information Processing Systems, vol. 30. Curran Associates, Inc. (2017)
Hamilton, W.L.: Graph representation learning. Synth. Lect. Artif. Intell. Mach. Learn. 14(3), 1–159 (2020)
Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks (2017). https://doi.org/10.48550/ARXIV.1609.02907
Li, W., Matejka, J., Grossman, T., Konstan, J.A., Fitzmaurice, G.: Design and evaluation of a command recommendation system for software applications. ACM Trans. Comput.-Hum. Interact. (TOCHI) 18(2), 1–35 (2011). https://doi.org/10.1145/1970378.1970380
Li, Y., Vinyals, O., Dyer, C., Pascanu, R., Battaglia, P.: Learning deep generative models of Graphs (2018). https://doi.org/10.48550/ARXIV.1803.03324
Lupinetti, K., Pernot, J.P., Monti, M., Giannini, F.: Content-based CAD assembly model retrieval: survey and future challenges. Comput. Aided Des. 113, 62–81 (2019). https://doi.org/10.1016/j.cad.2019.03.005
Mikolov, T., Chen, K., Corrado, G., Dean, J.: Efficient estimation of word representations in vector space (2013). https://doi.org/10.48550/ARXIV.1301.3781
Sarcar, M., Rao, K.M., Narayan, K.L.: Computer Aided Design and Manufacturing. PHI Learning Pvt Ltd., New Delhi (2008)
Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., Bengio, Y.: Graph attention networks (2018). https://doi.org/10.48550/ARXIV.1710.10903
Wu, R., Xiao, C., Zheng, C.: DeepCAD: a deep generative network for computer-aided design models. In: Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), pp. 6772–6782 (2021)
Wu, S., Sun, F., Zhang, W., Xie, X., Cui, B.: Graph neural networks in recommender systems: a survey (2020). https://doi.org/10.48550/ARXIV.2011.02260
Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., Yu, P.S.: A comprehensive survey on graph neural networks. IEEE Trans. Neural Netw. Learn. Syst. 32(1), 4–24 (2021). https://doi.org/10.1109/tnnls.2020.2978386
Yoo, S., Lee, S., Kim, S., Hwang, K.H., Park, J.H., Kang, N.: Integrating deep learning into CAD/CAE system: generative design and evaluation of 3D conceptual wheel. Struct. Multidiscip. Optim. 64(4), 2725–2747 (2021)
Yun, S., Jeong, M., Kim, R., Kang, J., Kim, H.J.: Graph transformer networks. In: Advances in Neural Information Processing Systems, vol. 32. Curran Associates, Inc. (2019)
Zaheer, M., Kottur, S., Ravanbakhsh, S., Poczos, B., Salakhutdinov, R.R., Smola, A.J.: Deep sets. In: Advances in Neural Information Processing Systems, vol. 30. Curran Associates, Inc. (2017)
Zhang, H., Li, M., Wang, M., Zhang, Z.: Understand graph attention network, February 2022. https://www.dgl.ai/blog/2019/02/17/gat.html. Accessed 06 Apr 2022
Zhang, Z., Jaiswal, P., Rai, R.: Featurenet: machining feature recognition based on 3d convolution neural network. Comput. Aided Des. 101, 12–22 (2018). https://doi.org/10.1016/j.cad.2018.03.006
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Gajek, C., Schiendorfer, A., Reif, W. (2023). A Recommendation System for CAD Assembly Modeling Based on Graph Neural Networks. In: Amini, MR., Canu, S., Fischer, A., Guns, T., Kralj Novak, P., Tsoumakas, G. (eds) Machine Learning and Knowledge Discovery in Databases. ECML PKDD 2022. Lecture Notes in Computer Science(), vol 13713. Springer, Cham. https://doi.org/10.1007/978-3-031-26387-3_28
Download citation
DOI: https://doi.org/10.1007/978-3-031-26387-3_28
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-26386-6
Online ISBN: 978-3-031-26387-3
eBook Packages: Computer ScienceComputer Science (R0)
