Predicting Critical Heat Flux with Uncertainty Quantification and Domain Generalization Using Conditional Variational Autoencoders and Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
Deep generative models (DGMs) can generate synthetic data samples that closely resemble the original dataset, addressing data scarcity. In this work, we developed a conditional variational autoencoder (CVAE) to augment critical heat flux (CHF) data used for the 2006 Groeneveld lookup table. To compare with traditional methods, a fine-tuned deep neural network (DNN) regression model was evaluated on the same dataset. Both models achieved small mean absolute relative errors, with the CVAE showing more favorable results. Uncertainty quantification (UQ) was performed using repeated CVAE sampling and DNN ensembling. The DNN ensemble improved performance over the baseline, while the CVAE maintained consistent results with less variability and higher confidence. Both models achieved small errors inside and outside the training domain, with slightly larger errors outside. Overall, the CVAE performed better than the DNN in predicting CHF and exhibited better uncertainty behavior.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 13 of its 35 references that are in the map. First posted on arXiv 9 Sep 2024.
Builds on
13 of 35 references.
References · 13
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2016TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systemscited by 664
- 2018Tune: A Research Platform for Distributed Model Selection and Trainingcited by 395
- 2021An Introduction to Deep Generative Modelingcited by 104
- 2016An Uncertain Future: Forecasting from Static Images using Variational Autoencoderscited by 87
- 2016Amortised MAP Inference for Image Super-resolutioncited by 68
- 2020Semi Conditional Variational Auto-Encoder for Flow Reconstruction and Uncertainty Quantification from Limited Observationscited by 13
- 2021Dizygotic Conditional Variational AutoEncoder for Multi-Modal and Partial Modality Absent Few-Shot Learningcited by 4
Built on by
Citing papers per year: 2024: 2; 2025: 4.
6 papers.
- 2024Bayesian optimized deep ensemble for uncertainty quantification of deep neural networks: a system safety case study on sodium fast reactor thermal stratification modelingcited by 5
- 2025Physics-Based Hybrid Machine Learning for Critical Heat Flux Prediction with Uncertainty Quantificationcited by 2
- 2025Uncertainty Quantification for Data-Driven Machine Learning Models in Nuclear Engineering Applications: Where We Are and What Do We Need?cited by 2
- 2024An Investigation on Machine Learning Predictive Accuracy Improvement and Uncertainty Reduction using VAE-based Data Augmentationcited by 1
- 2025A Three-Stage Bayesian Transfer Learning Framework to Improve Predictions in Data-Scarce Domainscited by 0
- 2025Towards Overcoming Data Scarcity in Nuclear Energy: A Study on Critical Heat Flux with Physics-consistent Conditional Diffusion Modelcited by 0
- 2025Towards Overcoming Data Scarcity in Nuclear Energy: A Study on Critical Heat Flux with Physics-consistent Conditional Diffusion Model20 Nov 2025
- 2025A Three-Stage Bayesian Transfer Learning Framework to Improve Predictions in Data-Scarce Domains30 Oct 2025
- 2025Uncertainty Quantification for Data-Driven Machine Learning Models in Nuclear Engineering Applications: Where We Are and What Do We Need?16 Mar 2025
- 2025Physics-Based Hybrid Machine Learning for Critical Heat Flux Prediction with Uncertainty Quantification26 Feb 2025
- 2024Bayesian optimized deep ensemble for uncertainty quantification of deep neural networks: a system safety case study on sodium fast reactor thermal stratification modeling11 Dec 2024
- 2024An Investigation on Machine Learning Predictive Accuracy Improvement and Uncertainty Reduction using VAE-based Data Augmentation24 Oct 2024