Python-based Surrogate Modeling Objects (PySMO)
What is PySMO?
Python-based Surrogate Modeling Objects (PySMO) is an open-source framework for general-purpose surrogate modeling techniques, integrated with the Pyomo mathematical optimization framework (on which IDAES is also based). PySMO provides a suite of tools for building, training, validating, and deploying surrogate models in optimization problems. It supports various surrogate modeling techniques, including polynomial regression, Kriging (Gaussian process regression), and radial basis functions. PySMO is designed to be flexible and extensible, allowing users to easily incorporate new surrogate modeling methods and customize existing ones to suit their specific needs. PySMO supports flowsheeting and direct integration into an equation-oriented modeling framework, providing the full suite of tools for both sampling and surrogate model generation.
PySMO's surrogate modeling capabilities have been shown to produce comparable results and performance in mathematical optimization problems to ML models trained with open-source tools such as Keras, as well as commercial surrogate modeling tools (e.g., ALAMO) [1 , 2].
PySMO is free and open-source, and is released under the BSD license as part of IDAES. Further information may be found on GitHub. Documentation for PySMO may be found here .
Research Using PySMO
- HA Pedrozo, MA Zamarripa, A Uribe-RodrÃguez, G Panagakos, MS Diaz, LT Biegler. Surrogate model optimization: a comparison case study with pooling problems of CO2 point sources. Comp. Chem. Eng., Vol. 200, 109199, 2025.
- D.M. Fardis, D. Oh, N.V. Sahinidis, A. Garciadiego, A. Lee. Surrogate modeling and optimization of the leaching process in a rare earth elements recovery plant. Comp. Chem. Eng., Vol. 197, 109061, 2025.
- O.O. Amusat, A.A. Atia, A.V. Dudchenko, and T.V. Bartholomew. Modeling Framework for Cost Optimization of Process-Scale Desalination Systems with Mineral Scaling and Precipitation. ACS ES&T Engineering, 2024.
- M. Rebosolan, M. van Soestbergen, J.J.M. Zaal, T. Hauck, A. Dasgupta, B. Chen. Effect of microstructural variability on fatigue simulations of solder joints.Microelectronics Reliability Vol. 162, 115511, 2024.
- O.O. Amusat, A.V. Dudchenko, A.A. Atia, T.V. Bartholomew. Cost-optimal Selection of pH Control for Mineral Scaling Prevention in High Recovery Reverse Osmosis Desalination. Proceedings of the 10th International Conference on the Foundations of Computer-Aided Process Design (FOCAPD), Colorado, USA, 2024.
- A. Rocher, V. Ruhlmann-Kleider, E. Burtin and A. de Mattia. Halo occupation distribution of Emission Line Galaxies: fitting method with Gaussian processes. Journal of Cosmology and Astroparticle Physics, 2023.
- J. Galindo, R. Navarro, F. Moya, A. Conchado. Comprehensive Method for Obtaining Multi-Fidelity Surrogate Models for Design Space Approximation: Application to Multi-Dimensional Simulations of Condensation Due to Mixing Streams. Appl. Sci. 13(11), 6361; 2023.
- J. Wang, E.A. Eugene, A W. Dowling. Scalable Stochastic Programming with Bayesian Hybrid Models.Computer Aided Chemical Engineering, Vol. 49, 1309-1314, 2022.