A U-net segmentation model for predicting free convection over confined isothermal tubes

Behzad Baghapour
DOI: https://doi.org/10.1016/j.icheatmasstransfer.2024.107271
IF: 6.782
2024-02-02
International Communications in Heat and Mass Transfer
Abstract:A deep-learning segmentation model was proposed to predict steady-state thermal fields. The U-Net architecture was used in a reversed order, with the body segments serving as inputs to the model, and the temperature and velocity fields as outputs. Inherent relationships between the segment shapes and fluid domains were extracted via convolution-sampling stages, incorporating skip connections between high- and low-resolution data. An attention gate was used to enhance focus on the body regions. CFD was used to generate the ground truth data. To inform the model about body temperature, a Rayleigh-scale range was considered for body segments. Geometry representation was simplified by utilizing only the boundary nodes , thereby eliminating the need for complex distance functions. The model accuracy was examined for free convection over elliptical cylinders of various shapes and circular cylinders at different positions within an enclosure. Attention U-Net (AUN) achieved the highest R-squared scores for Nusselt numbers at the enclosure wall, with values of 0.9933 and 0.9822 for elliptical and circular cylinder cases, respectively. It could improve the accuracy by 80.6%, with a 14.5% increase in complexity compared to the conventional auto-encoder. GPU accelerated training by 45 times and testing by 10 times compared to a single CPU core.
thermodynamics,mechanics
What problem does this paper attempt to address?