Transfer Learning on Multi-Dimensional Data: A Novel Approach to Neural Network-Based Surrogate Modeling

๐Ÿ“… 2024-10-16
๐Ÿ›๏ธ Journal of Machine Learning for Modeling and Computing
๐Ÿ“ˆ Citations: 1
โœจ Influential: 0
๐Ÿ“„ PDF
๐Ÿค– AI Summary
Convolutional neural networks (CNNs) for surrogate modeling of high-dimensional partial differential equations (PDEs) suffer from prohibitive computational costs due to reliance on large-scale, high-fidelity numerical simulations. Method: We propose a cross-dimensional transfer learning framework featuring a novel hybrid-dimensional (d- and (dโˆ’1)-dimensional) joint training paradigm. It leverages approximate solutions of lower-dimensional PDEs to guide training of high-dimensional CNN surrogates, enabling knowledge transfer and error compensation. The architecture employs a fully convolutional encoderโ€“decoder, multi-scale transfer mechanisms, and PDE-informed dimensionality-reduced data generation, augmented with uncertainty quantification. Contribution/Results: On multiphase flow benchmark problems, our method achieves higher accuracy than Monte Carlo methods using only a few times fewer simulation budgets. Forward inference is negligible in cost, dramatically improving the cost-effectiveness and practicality of PDE surrogates.

Technology Category

Machine Learning: Transfer, Domain Adaptation, Multi-Task LearningComputer Vision: Learning & Optimization for CVSearch and Optimization: Learning to Search

Application Category

Graph Algorithms and Modeling for the Web: Graph neural networks and deep learning approaches for Web-related graphsSearch and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for rankingSystems and Infrastructure for Web, Mobile and WoT: Web applications in cross-disciplinary domains and verticals such as mixed reality, smart cities, and digital health
๐Ÿ“ Abstract
The development of efficient surrogates for partial differential equations (PDEs) is a critical step towards scalable modeling of complex, multiscale systems-of-systems. Convolutional neural networks (CNNs) have gained popularity as the basis for such surrogate models due to their success in capturing high-dimensional input-output mappings and the negligible cost of a forward pass. However, the high cost of generating training data -- typically via classical numerical solvers -- raises the question of whether these models are worth pursuing over more straightforward alternatives with well-established theoretical foundations, such as Monte Carlo methods. To reduce the cost of data generation, we propose training a CNN surrogate model on a mixture of numerical solutions to both the $d$-dimensional problem and its ($d-1$)-dimensional approximation, taking advantage of the efficiency savings guaranteed by the curse of dimensionality. We demonstrate our approach on a multiphase flow test problem, using transfer learning to train a dense fully-convolutional encoder-decoder CNN on the two classes of data. Numerical results from a sample uncertainty quantification task demonstrate that our surrogate model outperforms Monte Carlo with several times the data generation budget.
Problem

Research questions and friction points this paper is trying to address.

Complex Multilayer Systems
Convolutional Neural Networks (CNNs)
Monte Carlo Methods
Innovation

Methods, ideas, or system contributions that make the work stand out.

Convolutional Neural Networks
High-dimensional and Low-dimensional Data Integration
Efficiency Advantage over Monte Carlo Methods
๐Ÿ”Ž Similar Papers
No similar papers found.
๐Ÿ’ผ Related Jobs
No related jobs found.
Stanford University
A
A. M. Propp
Institute for Computational and Mathematical Engineering, Stanford University, Stanford, CA 94305
D
D. Tartakovsky
Department of Energy Science and Engineering, Stanford University, Stanford, CA 94305