Optimizing CAD Models with Latent Space Manipulation

Jannes Elstner,Raoul G. C. Schönhof,Steffen Tauber,Marco F Huber
2023-03-09
Abstract:When it comes to the optimization of CAD models in the automation domain, neural networks currently play only a minor role. Optimizing abstract features such as automation capability is challenging, since they can be very difficult to simulate, are too complex for rule-based systems, and also have little to no data available for machine-learning methods. On the other hand, image manipulation methods that can manipulate abstract features in images such as StyleCLIP have seen much success. They rely on the latent space of pretrained generative adversarial networks, and could therefore also make use of the vast amount of unlabeled CAD data. In this paper, we show that such an approach is also suitable for optimizing abstract automation-related features of CAD parts. We achieved this by extending StyleCLIP to work with CAD models in the form of voxel models, which includes using a 3D StyleGAN and a custom classifier. Finally, we demonstrate the ability of our system for the optimiziation of automation-related features by optimizing the grabability of various CAD models. This is an open access article under the CC BY-NC-ND license (<a class="link-external link-http" href="http://creativecommons.org/licenses/by-nc-nd/4.0/" rel="external noopener nofollow">this http URL</a>) Peer review under the responsibility of the scientific committee of the 33rd CIRP Design Conference.
Computer Vision and Pattern Recognition,Artificial Intelligence,Computational Engineering, Finance, and Science,Machine Learning
What problem does this paper attempt to address?
### Problems Addressed by the Paper In the field of automation, the ability to optimize CAD (Computer-Aided Design) models is an important research direction. However, the application of neural networks in this area is still limited. Particularly for some abstract features (such as automation capabilities), these features are difficult to evaluate through simulation, too complex to handle with rule-based systems, and lack sufficient data for machine learning methods. This paper aims to optimize the abstract features of CAD models by drawing on the successful experiences in the field of image processing, especially by utilizing the latent space of Generative Adversarial Networks (GAN). Specifically, the authors extend the StyleCLIP method to handle CAD models to optimize abstract features related to automation. For example, they demonstrate how to improve the automation assembly capability by optimizing the graspability of screws and bolts. ### Main Contributions 1. **Extending StyleCLIP**: The authors extend the StyleCLIP method from the field of image processing to the optimization of CAD models by using 3D StyleGAN and custom classifiers. 2. **Optimizing Abstract Features**: By manipulating the latent space, the authors successfully optimized the graspability of CAD models, an abstract feature closely related to automated assembly. 3. **Experimental Validation**: The authors conducted experiments using a small-scale dataset, demonstrating the potential and limitations of their method in practical applications. ### Key Technologies - **3D StyleGAN**: Used for generating and encoding CAD models. - **Custom Classifier**: Used to evaluate optimization targets. - **Latent Space Manipulation**: Includes latent optimization and latent mapper methods. ### Conclusion Although many optimization results have high perceptual quality and rationality, there are still some unrealistic or distorted optimization results. This indicates that the current method still faces challenges in handling CAD models, especially in maintaining the authenticity and functionality of the models. Future work can explore the use of generative models with natural language representations to better handle complex CAD datasets and preserve specific functional structures.