Geometry-Informed Neural Networks
We introduce GINN -- a framework for training shape-generative neural fields without data by leveraging design constraints and avoiding mode-collapse using a diversity loss.
摘要
评审与讨论
The paper proposes geometry-informed neural networks. The idea is to train a generative model not from data, but using objective function and constraints. Basically, the generative model is trained by a specification similar to an optimization problem.
优点
The paper proposes an interesting idea that is worthy of further exploration. The research proposes the idea of a framework, but it does not demonstrate a working instantiation of the idea, so the research is only in its beginning stages. I think the concept is potentially very promising, but there are also predictable and significant obstacles that must be overcome. Design by optimization is a longstanding topic in many fields, but there are also many problems. Most notably, there aren't many or even any good examples of where optimization alone can give rise to interesting designs. Most of the time, this requires a good initialization that is already a design or the combination of user input (design) and optimization. For example, in architectural design you typically need an initial surface and you can design a pattern on the surface (e.g. a paneling). In furniture design, you also typically need to restrict the design space to a meaningful subset, e.g. by providing procedural rules or templates. At the moment, the method is an extension of an idea that does not have a working instantiation.
缺点
The major weakness of the paper is the lack of meaningful designs. Without interesting design examples, the paper is not meaningful. None of the designs shown in the paper are interesting and they cannot be recognized as mechanical, biological, architectural, ... objects. These are abstract designs like abstract art, and it is not meaningful to create art by optimization and then infer how this would transfer to engineering. I would suggest a three-step approach to tackling this problem:
- The project should identify an example where a single interesting design can be created by optimization. This example should not be abstract but specific to an engineering problem and be recognizable as an intentional and meaningful design. This example can be generated by optimization alone. It doesn't matter if the design is from architecture, mechanical engineering, biology, geology, ... but it should be meaningful. If you want to go for something more discrete, I would recommend furniture or CAD designs. This may not work well with your framework, so I could imagine that free-form architecture could be a better application area, e.g. "Geodesic patterns", developable surfaces, self-supporting surfaces, or quad meshes.
- The project should expand from this single design to generate a set of diverse designs by combining optimization with a diversity constraint.
- The project should example from a set of designs generated by optimization to combining optimization with generative modeling.
The current submission does not have a demonstration of either 1, 2, or 3. Competing work (not cited) has at least somewhat of a demonstration of points 1 and 2 (e.g., "Fit and Diverse: Set Evolution for Inspiring 3D Shape Galleries"), but it would be desirable to have even better examples.
It is possible to accept the paper for conceptual novelty, but I am not in favor of such a philosophy. A paper should introduce a conceptual novelty and at the same time introduce a working realization, not only the conceptual novelty. The conceptual novelty by itself should not be enough for publication. I conjecture that the reason why there aren't many other papers on this topic is because people could not find meaningful and working instantiations of this concept. There is an obvious approach: first generate a set of objects by optimization (e.g. furniture) and then train a generative model on this set. I am not aware of such a successful approach and a paper in this space should possibly demonstrate that it can beat such a baseline..
问题
N / A
伦理问题详情
No concerns
We strongly disagree with the reviewer and want to clarify the following:
- It is well known that topology optimization can create near-optimal designs in many engineering applications [1]. The reviewer’s claim is factually incorrect: “Most notably, there aren't many or even any good examples of where optimization alone can give rise to interesting designs.”
- Moreso, these methods start from a solid material block and do not require a good initialization as suggested by the reviewer. “Most of the time, this requires a good initialization that is already a design or the combination of user input (design) and optimization.”
- “The current submission does not have a demonstration of either 1, 2, or 3.” Steps 1 and 2 are indeed what we demonstrate, as explicitly illustrated in Fig. 3. The reviewer’s proposed approach “first generate a set of objects by optimization (e.g. furniture) and then train a generative model on this set” has been tried multiply before and is deemed impractical [2].
- The suggested reference is far-fetched as it requires data and human-feedback in the loop. Even so, it is concluded that "it cannot [..] evolve it into a richly diverse set consisting of complex and advanced shapes." Our method is data-free and explicitly produces diverse shapes.
We hope the provided information helps the reviewer reconsider their assessment and we are happy to discuss further questions.
[1]: Jihong Zhu et al., A review of topology optimization for additive manufacturing: Status and challenges, Chinese Journal of Aeronautics
[2]: Rebekka V. Woldseth et al., On the use of Artificial Neural Networks in Topology Optimisation, Springer: Structural and Multidisciplinary Optimization
I acknowledge that the authors disagree. However, the disagreement is subjective. In my expert opinion, the authors do not meet the quality criteria in computational design. The quality of the designs in Figures 1, 3, and 6 is below what I consider reasonable and below other examples in computational design published at top conferences. I think the paper should include (much) better and a much larger variety of 3D results before being considered for publication at a top conference.
First, we disagree that our arguments are subjective - we provide concrete statements from the literature that objectively contradict the reviewer's claims.
Second, we validate the proposed framework by optimizing shapes for minimal surface area, smoothness, connectedness, higher-order topological properties, and even a physics task with solution multiplicity. Regardless of the reviewer's preference for the shapes, the framework functions as intended, solving the specified design problems.
Third, our experiments are designed to clearly demonstrate key properties of the proposed machine learning framework: constraint satisfaction, objective minimization, scalability to 3D problems, generative ability recovering multiple diverse solutions with the capacity to interpolate and structure the latent space. We would invite the reviewer to judge this submission on the stated contributions instead of singling out the "abstractness" of the results.
This paper presents the framework of GINN, which formulates the objectives and constraints of geometry problem as the loss function to train the neural networks. More importantly, it proposes a diversity constraint, which avoids the mode-collapse of other generative models and promotes the model to generate diverse solutions for the geometry problem. In the experiment section, it validates the performance of GINN on four problems and additionally conducts an engineering design case study. It is able to find the accurate solution for the famous geometry problems, and produce diverse solutions for the engineering design problem.
优点
- Although there have been a number of existing works using some similar losses to train their network, as mentioned in the related work, this paper presents the GINN framework with a very comprehensive summary and discussion on the common constraints used in the validation experiments. This would help the following works in formulating their own applications.
- Comparing to the physics-informed neural networks or the topology optimization, this paper proposes the diversity constraints for geometry problems. Comparing to other generative models such as boltzmann generators, the proposed diversity constraints help to avoid mode-collapse.
缺点
- My main concern is about the technical novelty and the evaluation. Although this paper has presented many validation experiments, most of the results only show the performance of the GINN and some ablation settings, without comparing to other existing methods. In Section 4 it claims that there’s no established baselines, problems, and metrics for the data-free shape-generative modeling. Does it mean that there is no related work trying to solve the engineering design problem presented in Section 4.3??
- I understand that there are very much information to be presented in the paper. But it is a bit difficult to capture the key information. For example, the “topology” “smoothness” “surface sampling” are the defined metrics for each constraint? It’s better to have “constrained optimization”, “metrics”, “models” in bold font, and list all the metrics under the heading. The same problem exists in Section 4.3, there are too many headings to understand the structure of the writing.
问题
I’m not very familiar with the specific field (engineering design) of this paper. In my understanding, the idea of using objectives and constraints as loss functions is not completely new in the field of 3D generative models. But this paper has extended this idea with many more kinds of constraints and proposed the diversity constraint specifically for the engineering design problem. My only doubt is whether there are indeed no existing works to be compared with. I'd like to increase my rating if the authors can give more explanation on this.
We thank the reviewer for their comments. We would like to clarify the raised concerns and questions.
W1: No established baselines: We affirm the reviewer’s question that there are no established baselines that train a shape-generative model without data. The essence of a generative model is to map a source to a target probability distribution. In the data-free regime, classical topology optimization generates a single shape. But no prior work trains a generative model for shapes without training data. As such, we are not aware of a direct comparison that we could make.
“Does it mean that there is no related work trying to solve the engineering design problem presented in Section 4.3??”
The jet-engine bracket problem can be solved using topology optimization (TO) [1]. However, we do not consider TO an insightful baseline, as most metrics would be nearly trivial (Betti-numbers: 1, interface- and envelope-metrics: 0). On the other hand, naive TO and our results use different objectives (compliance vs surface smoothness), which also makes the comparison w.r.t. these metrics not meaningful. Lastly, the primary focus of our work is the generative aspect to produce diverse solutions. As classical TO can only produce a single solution, the diversity comparison would provide little value. We hope these arguments clarify why we omit a comparison. If the reviewer disagrees, we could provide a comparison to TO or the SimJEB dataset [2].
W2: We thank the review for their feedback on the format and structure. In Section 4 we state “we define metrics for each constraint in the main problem as detailed in Appendix C.1”. We reformulated this sentence to more clearly state that the metrics definitions are in Appendix C. As the paragraph titles “topology”, “smoothness”, “surface sampling” etc. in Section 4.1 highlight different experimental details, we would leave this structure unchanged. Please let us know if further clarification or changes are needed.
[1]: Carter et al., The GE aircraft engine bracket challenge: an experiment in crowdsourcing for mechanical design concepts, 2014 University of Texas at Austin
[2]: Whalen et al., SimJEB: Simulated Jet Engine Bracket Dataset, Computer Graphics Forum 2021
I thank the authors' response. In terms of the comparison, I accept the authors' statements that naive TO and the proposed generative model have different objectives and diversity is the primary focus of this work. However, we need some evidence to be convinced that the generated results are both diverse and of high-quality. So I still encourage the authors to provide the results of TO as a reference example.
Actually, I took a look at the papers [1] and [2]. Comparing to the figures in these papers, the presented generated solutions of the generative model seem not very satisfactory. Could the authors provide a discussion on the quality of generated results?
This paper presents a framework for achieving controllable generation of industrial part geometry. Specifically, the authors utilize an implicit neural field to represent the surface of industrial parts. Through optimization objectives in different aspects, the implicit neural field is adjusted to closely match the given targets, such as ensuring the zero-level set of the implicit field aligns with the geometric surface and the first-order derivatives of the implicit field align with the surface normals. Additionally, the authors attempt to introduce a regularization term to promote diverse optimization results. Finally, the authors conduct experiments on a challenging example and demonstrate the effectiveness of their approach.
优点
This paper presents a well-formulated approach for shape-generative modeling driven by geometric constraints and objectives. It starts by considering the problem from a theoretical perspective and successfully translates it into an executable framework. The research problem addressed in this paper is valuable, and the preliminary experimental results provided by the authors demonstrate the effectiveness of their proposed method. Additionally, the paper is well-written and easy to follow.
缺点
The framework solution proposed in the paper is effective from a high-level perspective, but I anticipate numerous challenges when it comes to practical implementation. My primary concern is whether we truly have the capability to account for all objective functions based on individual intuition, especially considering that these functions must be feasible, differentiable, and ideally, non-conflicting. For instance, how should we approach the generation of a nut that matches a specific type of screw, or a particular joint bearing?
Moreover, while the article emphasizes its data-free approach, it appears to me as an optimization-based strategy, with the optimization target being the implicit field represented by the neural network—a relatively conventional method. I do not find this particularly innovative.
Additionally, although Figure 6 does showcase a variety of workpieces, I doubt the compliance of these pieces with standard usage requirements; many seem to be diverse merely for the sake of diversity.
Lastly, the experiments are conducted on only one workpiece, which, despite its complexity, does not suffice to demonstrate the universality of the method.
问题
Although the authors have attempted to provide some formulations in Table 1, I still feel that these modules are far from sufficient to support the design of a workpiece. However, designing new constraints requires extreme caution and a significant amount of skill. Therefore, I am not particularly optimistic about this section. If the authors could propose a more universal paradigm for generating such constraints, many of my concerns would be alleviated.
We thank the reviewer for their time and effort in reviewing our submission and the overall positive outlook. We would like to clear the raised concerns.
W1, W4, Q: Universality: The proposed approach is at least as universally applicable as any classical shape or topology optimization (TO) approach. Classical TO methods are predominantly gradient-based for which the constraints must satisfy the same properties. As such, we can reuse numerous existing works [e.g., 1-4], which cover most practically relevant constraints. Of course, some requirements or implicit know-how are hard to describe formally or efficiently, which is a known limitation of computational design.
We would also like to address the statement that “the experiments are conducted on only one workpiece”. While the jet-engine bracket is our primary experiment, we demonstrate three other geometry problems. In addition, our approach generalizes to other scientific domains where the problems are ubiquitously formalized as differential or integral equations, as we demonstrate with the reaction-diffusion system in Figure 4. These experiments echo the same main findings. This can be done with minimal hyperparameter tuning thanks to the adaptive Lagrangian method which balances the different constraints automatically further supporting universality.
Lastly, the final experiment itself is set up to illustrate generality, as it seeks general, non-extruded, asymmetric 3D parts with many common and transferable constraints.
W2: Optimization with a NN representation: The reviewer is correct that optimizing neural fields with objectives has been demonstrated before, as we also describe in related work using PINNs and TO. The primary novelty of our contribution is a framework for training shape-generative neural fields without data by leveraging design constraints and avoiding mode-collapse using a diversity constraint.
W3: Diversity in Fig. 6: The shapes in Fig. 6 are diverse and close-to-optimal w.r.t. to the defined objective of surface smoothness. As we describe in Section 4.3 and Limitations, this objective is different from physical compliance, which would produce more conventionally looking shapes, but would add computational and implementation complexity that detracts from the primary investigation.
Generation of a nut or joint bearing: Generating a nut that matches a specific bolt is fairly straight-forward in this framework. It would require defining the interface to the bolt analogous to the already demonstrated interface constraints. This can be represented either by a discrete mesh or the analytical expression of the swept thread profile known from the engineering standards. A similar procedure must be carried out for the likely desirable flat top and bottom interfaces, as well as the outer interface, which in the simplest form would be two flat and parallel surfaces to take the wrench. This can optionally be extended with angular symmetries. To connect these interfaces and make the part structurally integral, we can either reuse the presented connectedness constraint or add a mechanical stress constraint. However, since bolts are standard parts applying computational design here is probably an overkill, but we hope this explanation provides further insight and we are glad to discuss further questions.
[1] Vatanabe S. et al., Topology optimization with manufacturing constraints: A unified projection-based approach, Advances in Engineering Software, 2016.
[2] Liu J. et al., A survey of manufacturing oriented topology optimization methods, Advances in Engineering Software, 2016.
[3] Ebeling-Rump, M. et al. Topology optimization subject to additive manufacturing constraints. J.Math.Industry, 2021.
[4] Hammond A. et al., Photonic topology optimization with semiconductor-foundry design-rule constraints, Opt. Express, 2021.
This work introduces a novel framework for data-free geometry learning. The framework relies on implicit geometry representation - which is a modulated conditional neural field - where conditioning is on latent variables to ensure diversity. Given such a representation, authors propose to formulate a constrained optimization problem, in practice written down as a set of differentiable losses. The proposed constraints include: minimizing genus, smoothness, and diversity of generated shapes. Authors conduct a series of toy experiments to validate the model, and test it on a single engineering design problem.
优点
- Paper is well-written and is easy to follow.
- Overall framework is meaningful and has a large number of potential applications (in particular in generative design and shape optimization).
- The promise of representation that does not require significant amount of data is very valuable specifically because the amount of data in generative design is often scarce or requires expensive simulations.
- Diversity measure that is demonstrated to work in practice for tackling mode collapse in implicit field learning is a potentially a critical contribution.
缺点
- (minor) Most of the proposed constraints have been developed in previous work.
- (minor) For the method to actually be useful for the real-world applications (e.g. in shape optimization in engineering fields), a mapping to an explicit representation could be a hard requirement to be compatible with existing simulation.
- Although the representation/optimization framework that does not require data has promise, it would be interesting to see if these methods can be used in combination with available data.
- The experimental evaluation is very limited. In particular for shape optimization (which is the only truly realistic test iiuc), there is only single example provided, which makes it hard to understand how much of the results is due to parameter tuning.
问题
- How robust is the method to the choice of hyperparameters?
- Would it be possible to demonstrate results on more than one example for engineering design?
We thank the reviewer for their time and effort in reviewing our submission and the overall positive outlook. We would like to clear the raised concerns.
W1: We do not claim the individual constraints as contributions. An advantage of GINNs is the ability to include different existing domain-specific constraints. For a more detailed discussion on this, we refer to our response (universality) to the reviewer 7pFZ.
W2: The conversion to explicit representations is a known limitation whenever working with implicit representations, such as in state-of-the-art topology optimization or shape-generative models. Correspondingly, there are established methods for extracting meshes, such as marching cubes in the simplest case. It is also worth mentioning that it is also possible to work and simulate directly on the implicit representations, which is an emerging trend in both academia [1] and industry [2,3].
W3: We completely agree with the reviewer that “it would be interesting to see if these methods can be used in combination with available data.” Despite our original motivation being training models on extremely sparse shape datasets, we left this for future work to focus on the key novel aspects, such as the diversity constraint and the training dynamics. We have preliminary results indicating that adding data is possible, but there are many design choices and comparisons that have to be made which we omitted to maintain the focus of this work.
Q1, W4: Hyperparameter tuning: We use the adaptive ALM algorithm (with default RMSProp parameters) to avoid hyperparameter tuning of the different loss weights. We initialize the optimizer such that the loss terms are on the order 1. We use a simple WIRE MLP with 3 hidden layers with 128 neurons each. WIRE is quite robust to hyperparameter choice, as shown in Fig. 3 of the WIRE paper [4].
Q2: In addition to the three simpler geometry problems and the one physics problem, we selected the jet-engine bracket problem as it is a publicly available and very general 3D design challenge without any exploitable symmetries. We could offer to solve another illustrative design problem using GiNNs, e.g. wheel design similar to a recent data-driven design paper [5].
[1]: Sawhney et al., Monte Carlo geometry processing: A grid-free approach to PDE-based methods on volumetric domains, ACM Transactions on Graphics 2020
[4] Saragadam et al., WIRE: Wavelet Implicit Neural Representations, CVPR 2023
[5] Jang et al., Generative design by reinforcement learning: enhancing the diversity of topology optimization designs, Computer-Aided Design 2022
Thanks for the response. I am inclined to keep my original rating, but would encourage authors to show more than 1 jet-engine example to make their work more convincing.
As several reviewers expressed an interest in another engineering experiment, we would like to share an additional result. We investigate the design of a wheel, similar to a task in a data-driven design paper [1].
We want to stress that while we present a 2D experiment due to the limited time for running the experiments and visualizing the results, this experiment generalizes to a 3D wheel design by modifying the interfaces and adding an extrusion constraint.
The design requirements are:
- Domain:
- Design space: the ring between the inner radius and outer radius
- Interface constraints: inner and outer circles,
- Connectedness constraint,
- Diversity constraint,
- 5-fold cyclic symmetry constraint: implemented as the soft-constraint (sample a point, rotate it four times by , evaluate the implicit function at the five points, require the variance of these to be 0.) Alternatively, a hard constraint using a periodic encoding can also be used to achieve exact symmetry.
The cyclic symmetry constraint has not been employed in the previous experiments demonstrating the flexibility of the method. We also reuse the same WIRE architecture with no changes to the ALM hyperparameters.
The results for both 2D (link) and 3D (link) latent space (animated along the third latent dimension) reinforce the previous observations of recovering near-feasible, diverse results organized in the latent space.
[1] Jang et al., Generative design by reinforcement learning: enhancing the diversity of topology optimization designs, Computer-Aided Design 2022
As requested by the reviewers wHB8 and QgvS, we computed the metrics to compare GINN and the SimJEB dataset [1]. The shapes in the dataset are produced by human experts, many of which relied on topology optimization. For single shape metrics, we use shape 148 available in the SimJEB sample files and the GINN generated shape described in Section 4.3. The multi-shape metrics for SimJEB were computed on the first 196 clean shapes. For GINNs, these were computed by equidistantly sampling 14*14 shapes from the 2d latent space model.
- Interface, design region, connectedness: GINN and SimJEB shapes both satisfy these constraints exactly up to discretization errors.
- Curvature: GINNs outperform SimJEB, as this was used as the objective for training GINNs.
- Compliance: SimJEB outperforms GINNs, as the objective of the design contest was to optimize for mechanical performance. Specifically, we use compliance (stored deformation energy) as a metric. While the GINN framework can also optimize for compliance, we avoided it due to the large computational overhead.
- Diversity: GINNs produce more diverse shapes than the dataset created by human experts due to the explicit constraint in GINN training. This metric describes the expected distance between two shapes randomly sampled from the respective collections.
Overall, both methods produce equally feasible shapes and perform best on the objectives that they were created with. Our method also produces a collection of shapes that is more diverse than the dataset that required an estimated collective effort of 14 human years. We hope these metrics and the additional wheel design experiment provide further proof of our contribution and clarify the reviewers’ remaining concerns.
| Metric | SimJEB | GINN |
|---|---|---|
| Single shape | ||
| ↓ Connectedness (0-th Betti-number) | 1 | 1 |
| ↓ Interface (Chamfer distance) | 0.00 | 0.00 |
| ↓ Design region (Volume outside) | 0.00 | 0.00 |
| ↓ Curvature | 838 | 144 |
| ↓ Compliance | 2.01 | 7.81 |
| Multi shape | ||
| ↑ Diversity | 0.099 | 0.167 |
[1]: Whalen et al., SimJEB: Simulated Jet Engine Bracket Dataset, Computer Graphics Forum 2021
The paper introduces geometry-informed neural networks (GINNs) for optimizing 3D shapes given user-specified design requirements and constraints. The authors specifically incorporate the diversity constraint to get many different shapes and prevent mode collapse. The authors show results on the realistic problem of designing a a jet-engine lifting bracket. The reviewers note this work can have large number of potential applications. Due to scarce amount of data in generative design, being able to design shapes without having training is a desirable property. However, reviewers agree that a single experiment on real-world shapes (a jet-engine lifting bracket) was not sufficient to show generality of the approach. I suggest the paper to be rejected for ICLR.
审稿人讨论附加意见
The reviewers shared two main concerns about the paper. The first one is the universality of the approach, such as numerical challenges and sensitivity to the initial conditions, which the authors have address. Second, the original submission provided experiments on designing a single real-world shape, which reviewers did not find to be sufficient.
The authors have added new evaluations on 2D wheel design and multiple shapes from SimJEB dataset. However, the comparison on five metrics was provided for a single shape only, which is not sufficient to judge the generality of the method. Finally, on the provided evaluations, GINN underperforms in terms of compliance metric, which is an important for mechanical performance.
Reject