• Home
  • Search
  • High-Performance Kriging Surrogate Models Using PyTorch: Accelerating Engineering Analysis and Optimization
  • Cite Icon1
  • https://doi.org/10.2514/6.2026-1115Copy DOI Icon

High-Performance Kriging Surrogate Models Using PyTorch: Accelerating Engineering Analysis and Optimization

  • Jan 8, 2026
  • Derrick A Cox +2 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Kriging (Gaussian Process Regression) is a powerful technique for creating surrogate models used in engineering design optimization when dealing with computationally expensive simulations. Despite its effectiveness, traditional Kriging implementations face computational challenges when scaling to large datasets, particularly for repeated inference operations required by optimization and uncertainty quantification applications. We present a high-performance PyTorch implementation of Kriging that significantly accelerates both training and inference while maintaining predictive accuracy. Our implementation leverages PyTorch's optimized tensor operations, automatic differentiation capabilities, and efficient memory management to achieve up to 20x speedup during inference with large datasets compared to traditional NumPy implementations. Benchmark results across datasets ranging from 100-2000+ training points demonstrate comparable interpolation accuracy with substantial runtime improvements, particularly for larger prediction set sizes. The results indicate that numerical stability is largely maintained between the NumPy implementation and PyTorch. This implementation is particularly valuable for optimization, uncertainty quantification, adaptive sampling, and other applications requiring repeated surrogate model evaluations. Our work demonstrates that modern deep learning frameworks can significantly benefit classical statistical learning algorithms beyond neural networks.

Similar Papers
  • Research Article
  • Citations21

Study of a hull form optimization system based on a Gaussian process regression algorithm and an adaptive sampling strategy, Part I: Single-objective optimization

  • Apr 19, 2023
  • Ocean Engineering
  • Penghui Wang +3
  • Research Article
  • Citations18

Wave episode based Gaussian process regression for extreme event statistics in ship dynamics: Between the Scylla of Karhunen–Loève convergence and the Charybdis of transient features

  • Oct 06, 2022
  • Ocean Engineering
  • Stephen Guth +1
  • Research Article
  • Citations5

Machine learning enhanced predictions of ICRF heating: Overcoming numerical limitations via data curation

  • Jun 01, 2025
  • Physics of Plasmas
  • A Sanchez-Villar +8
  • Conference Article

Accelerating Pipeline Corrosion Modeling via Bayesian Active Learning

  • Sep 26, 2022
  • Shun Zhang +4
  • Research Article
  • Citations6

Uncertainty Quantification for Digital Twins in Smart Manufacturing and Robotics: A Review

  • Oct 01, 2024
  • Journal of Physics: Conference Series
  • S Battula +4
  • Research Article
  • Citations7

A Gaussian Process Regression Model for Forecasting Stock Exchange of Thailand

  • Mar 03, 2022
  • Trends in Sciences
  • Kamonrat Suphawan +2
  • Research Article
  • Citations13

Uncertainty quantification of two-phase flow in porous media via the Coupled-TgNN surrogate model

  • Dec 20, 2022
  • Geoenergy Science and Engineering
  • Jian Li +3
  • Conference Article
  • Citations3

Neural surrogate-driven modelling, optimisation, and generation of engineering designs: A concise review

  • Sep 15, 2024
  • Materials research proceedings
  • Siyi Chen
  • Research Article
  • Citations35

Machine Learning Regression Techniques for the Modeling of Complex Systems: An Overview

  • Jan 01, 2021
  • IEEE Electromagnetic Compatibility Magazine
  • Riccardo Trinchero +1
  • Video Transcripts

A surrogate model assisted with a subdomain model for permanent-magnet machine

  • Mar 30, 2021
  • Underline Science Inc.
  • Youtong Fang +2
  • Research Article

Optimization of Thermoplastic Elastomer (TPE) Components for Aerospace Structures Using Computerized Data-Driven Design

  • Jan 01, 2026
  • Computers, materials & continua/Computers, materials & continua (Print)
  • Adwaa Mohammed Abdulmajeed +4
  • Research Article
  • Citations10

Multi-fidelity neural network for uncertainty quantification of chemical reaction models

  • Oct 01, 2023
  • Combustion and Flame
  • Chengcheng Liu +3
  • Research Article
  • Citations6

Adaptive AI-based surrogate modelling via transfer learning for DEM simulation of multi-component segregation

  • Nov 06, 2024
  • Scientific Reports
  • Ahmed Hadi +3
  • Research Article
  • Citations19

Investigating adaptive sampling strategies for optimal building energy performance using artificial neural networks and kriging surrogate models

  • Dec 20, 2023
  • Journal of Building Engineering
  • S Lahmar +2
  • PDF
  • Research Article
  • Citations12

Surrogate Models for Performance Prediction of Axial Compressors Using through-Flow Approach

  • Dec 30, 2019
  • Energies
  • Xiaoxiong Wu +3
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.