Partitioned Active Learning for Heterogeneous Systems

Abstract

Active learning is a subfield of machine learning that focuses on improving the data collection efficiency in expensivetoevaluate systems. Active learningapplied surrogate modeling facilitates costefficient analysis of demanding engineering systems, while the existence of heterogeneity in underlying systems may adversely affect the performance. In this article, we propose the partitioned active learning that quantifies informativeness of new design points by circumventing heterogeneity in systems. The proposed method partitions the design space based on heterogeneous features and searches for the next design point with two systematic steps. The global searching scheme accelerates exploration by identifying the most uncertain subregion, and the local searching utilizes circumscribed information induced by the local Gaussian process (GP). We also propose Cholesky updatedriven numerical remedies for our active learning to address the computational complexity challenge. The proposed method consistently outperforms existing active learning methods in three realworld cases with better prediction and computation time.