cs.LGMar 10, 2026

A Unified Hierarchical Multi-Task Multi-Fidelity Framework for Data-Efficient Surrogate Modeling in Manufacturing

Authors: Manan Mehta, Zhiqiao Dong, Yuhang Yang, Chenhui Shao

Organizations: Department of Mechanical Science and Engineering, University of Illinois at Urbana-Champaign, Urbana, IL 61801, United States · Department of Mechanical Engineering, University of Michigan, Ann Arbor, MI 41809, United States

Abstract

Surrogate modeling is an essential data-driven technique for quantifying relationships between input variables and system responses in manufacturing and engineering systems. Two major challenges limit its effectiveness: (1) large data requirements for learning complex nonlinear relationships, and (2) heterogeneous data collected from sources with varying fidelity levels. Multi-task learning (MTL) addresses the first challenge by enabling information sharing across related processes, while multi-fidelity modeling addresses the second by accounting for fidelity-dependent uncertainty. However, existing approaches typically address these challenges separately, and no unified framework simultaneously leverages inter-task similarity and fidelity-dependent data characteristics. This paper develops a novel hierarchical multi-task multi-fidelity (H-MT-MF) framework for Gaussian process-based surrogate modeling. The proposed framework decomposes each task's response into a task-specific global trend and a residual local variability component that is jointly learned across tasks using a hierarchical Bayesian formulation. The framework accommodates an arbitrary number of tasks, design points, and fidelity levels while providing predictive uncertainty quantification. We demonstrate the effectiveness of the proposed method using a 1D synthetic example and a real-world engine surface shape prediction case study. Compared to (1) a state-of-the-art MTL model that does not account for fidelity information and (2) a stochastic kriging model that learns tasks independently, the proposed approach improves prediction accuracy by up to 19% and 23%, respectively. The H-MT-MF framework provides a general and extensible solution for surrogate modeling in manufacturing systems characterized by heterogeneous data sources.

Figures & tables

Explore similar work

CardsList
  1. Multifidelity-Augmented Gaussian Process Inputs for Surrogate Modeling from Scarce Data

    Mar 23, 2026Atticus Rex, Elizabeth Qian, David PetersonSurrogate ModelingMultifidelity Modeling

  2. MAST: A Multi-fidelity Augmented Surrogate model via Spatial Trust-weighting

    Feb 24, 2026Ahmed Mohamed Eisa Nasr, Ali Elham, Haris Moazam SheikhSurrogate ModelingMultifidelity Modeling

  3. Multi-fidelity surrogates for mechanics of composites: from co-kriging to multi-fidelity neural networks

    May 4, 2026Haizhou Wen, Elham Kiyani, Gang Li +3Surrogate ModelingSurrogate-Assisted Optimization