AI-Based Surrogate Model