Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 23 additions & 18 deletions engibench/problems/thermoelastic2d/v0.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,17 +25,23 @@ class ThermoElastic2D(Problem[npt.NDArray]):
## Problem Description
This is 2D topology optimization problem for minimizing weakly coupled thermo-elastic compliance subject to boundary conditions and a volume fraction constraint.

## Motivation
As articulated in their respective sections, both the Beams2D and HeatConduction2D problems found in the EngiBench library are fundamental engineering design problems that have historically served as benchmarks for the development and testing of optimization methods.
While their relevance is supported by needs in real engineering design scenarios (aerospace, automotive, consumer electronics, etc...), their mono-domain nature ignores the reality that coupling between domains exists, and should be accounted for in scenarios where performance in one domain significantly impacts performance in another.
To address this distinction, a multi-physics topology optimization problem is developed that captures the coupling between structural and thermal domains.

## Design space
The design space is represented by a 2D tensor of continuous design variables in the range [0, 1] that represent the material density at each voxel in the design space.
This multi-physics topology optimization problem is governed by linear elasticity and steady-state heat conduction with a one-way coupling from the thermal domain to the elastic domain.
The problem is defined over a square 2D domain, where load elements and support elements are placed along the boundary to define a unique elastic condition.
Similarly, heatsink elements are placed along the boundary to define a unique thermal condition.
The design space is then defined by a 2D array representing density values (parameterized by DesignSpace = [0,1]^{nelx x nely}, where nelx and nely denote the x and y dimensions).

## Objectives
The objectives are defined and indexed as follows:
0. `structural_compliance`: Structural compliance to minimize.
1. `thermal_compliance`: Thermal compliance to minimize.
2. `volume_fraction`: Volume fraction error to minimize.
The objective of this problem is to minimize total compliance C under a volume fraction constraint V by placing a thermally conductive material.
Total compliance is defined as the sum of thermal compliance and structural compliance.

## Boundary Conditions
Creating a problem formulation requires defining a python dict with the following info:
## Conditions
Problem conditions are defined by creating a python dict with the following info:
- `fixed_elements`: Encodes a binary NxN matrix of the structurally fixed elements in the domain.
- `force_elements_x`: Encodes a binary NxN matrix specifying elements that have a structural load in the x-direction.
- `force_elements_y`: Encodes a binary NxN matrix specifying elements that have a structural load in the y-direction.
Expand All @@ -44,8 +50,18 @@ class ThermoElastic2D(Problem[npt.NDArray]):
- `rmin`: Encodes the filter size used in the optimization routine.
- `weight`: Allows one to control which objective is optimized for. 1.0 Is pure structural optimization, while 0.0 is pure thermal optimization.

## Simulator
The simulation code is based on a Python adaptation of the popular 88-line topology optimization code, modified to handle the thermal domain in addition to thermal-elastic coupling.
Optimization is conducted by reformulating the integer optimization problem as a continuous one (leveraging a SIMP approach), where a density filtering approach is used to prevent checkerboard-like artifacts.
The optimization process itself operates by calculating the sensitivities of the design variables with respect to total compliance (done efficiently using the Adjoint method), calculating the sensitivities of the design variables with respect to the constraint value, and then updating the design variables by solving a convex-linear subproblem and taking a small step (using the method of moving asymptotes).
The optimization loop terminates when either an upper bound of the number of iterations has been reached or if the magnitude of the gradient update is below some threshold.

## Dataset
The dataset linked to this problem is on huggingface [Hugging Face Datasets Hub](https://huggingface.co/datasets/IDEALLab/thermoelastic_2d_v0).
This dataset contains a set of 1000 optimized thermoelastic designs in a 64x64 domain, where each design is optimized for a unique set of conditions.
Each datapoint's conditions are randomly generated by arbitrarily placing: a single loaded element along the bottom boundary, two fixed elements (fixed in both the x and y direction) along the left and top boundary, and heatsink elements along the right boundary.
Furthermore, values for the volume fraction constraint are randomly selected in the range $[0.2, 0.5]$.

Relevant datapoint fields include:
- `optimal_design`: An optimized design for the set of boundary conditions
- `strain`: The strain field for the initial uniform design
Expand All @@ -63,17 +79,6 @@ class ThermoElastic2D(Problem[npt.NDArray]):
- `rmin`: The filter size used in the optimization routine
- `weight`: The domain weighting used in the optimization routine

The dataset is generated by enumerating different boundary condition components, and taking the cartesian product of these components. The following components are enumerated:
- `fixed_elements`: 2 fixed elements that can lie anywhere on the left or top boundary
- `force_elements_x`: 1 force element that can lie anywhere on the right boundary
- `force_elements_y`: 1 force element that can lie anywhere on the bottom boundary
- `heatsink_elements`: N heat sink elements that can lie anywhere on the left or top boundary (N is also enumerated to have different sizes)
- `volfrac`: The volume fraction target of the optimized design


## Simulator
The evaluation code models the problem as a weakly coupled thermo-elastic problem. The code is written in pure python, and the evaluation is done in a single process.

## Lead
Gabriel Apaza @gapaza
"""
Expand Down
Loading