ZS

Zehua Song

Hello. I am Zehua Song 👋. My research focuses on using artificial intelligence to solve complex physical system problems.

I received my Master's degree in Artificial Intelligence from China University of Petroleum (Beijing). I was advised by Professor Jin Yang and Associate Professor Yu Song.

My research develops predictive modeling and decision optimization methods for actual engineering problems.

During my graduate studies, I published over 10 papers in international journals and conferences. Six of these were as first author or primary contributor. I also collaborated with companies including CNOOC, CNPC, and Kingsoft Cloud on research projects.

I am the recipient of the Doctoral Research Excellence Assistantship from Texas A&M University, the highest honor for incoming doctoral students, awarded to only 4 recipients across the Texas A&M University System per year. I was also named one of the 100 students nationwide in the People's Daily's list of exemplary National Scholarship recipients.

If you are interested in my work, please feel free to reach out.

Research Directions

My Research Directions

HBPINN: Physics-Informed Bayesian Modeling

The HBPINN framework integrates partial differential equations with Bayesian inference. It builds predictive models for complex physical systems with uncertainty quantification capabilities.

PINNs
Uncertainty Quantification
Probabilistic Assessment

DANN + Online Learning: Simulation-to-Reality Model Adaptation

This is a two-stage model adaptation framework. DANN handles initial domain transfer. Online learning enables continuous improvement. The result is a dynamically updatable model deployable in the field.

DANN
Online Learning
Evergreen Model

DN-SAC: Intelligent Decision-Making and Optimization for Complex Systems

The DN-SAC agent learns optimal decision policies in Markov Decision Process (MDP) environments. It enables intelligent optimization for complex systems.

Deep Reinforcement Learning
MDP
Sequential Decision Making

Honors and Awards

2026Doctoral Research Excellence AssistantshipTop honor · Only 4 recipients per yearTexas A&M University
2026Research AssistantshipDeclined$45,162 per yearUniversity of Massachusetts Amherst
2026Research AssistantshipDeclined$38,000 per yearColorado School of Mines
2025List of 100 Exemplary Graduate Students (National Scholarship)Top 100 nationwidePeople's Daily
2025Beijing Outstanding GraduateBeijing Municipal Education Commission
2025Professional Practice Excellence AwardTop 1%China University of Petroleum (Beijing)
2024National Scholarship for PostgraduatesTop 1.5%Ministry of Education, PRC
2023Outstanding Graduate Student AwardChina University of Petroleum (Beijing)
Featured Projects

Check out my latest work

Latent Space Controllable Generation and Inverse Design Tool for Porous Structures

This tool trains a latent space model on voxelized regular tetrahedron porous structure samples to disentangle hole count and pore size features. Given a target hole count and average radius, it searches for candidate structures via latent space sampling and connectivity analysis, enabling inverse generation from macro to micro. It evaluates generation controllability and diversity using structural connectivity and feature matching as objectives.

Latent Space Disentanglement
Conditional Inverse Generation
Multi-solution Candidate Search
Connectivity Constraints

Drill Bit Dynamics Parameter Inversion and Cumulative Work Analysis Tool

This tool uses mud logging data and high-frequency vibration measurements while drilling. It inverts downhole dynamics parameters to quantify energy dissipation from stick-slip, bit bounce, and lateral whirl. It establishes an energy-wear mapping relationship to assess bit cutting structure degradation and remaining useful life. It also recommends optimal drilling parameters for each section to minimize wear and maximize drilling efficiency.

Dynamics Parameter Inversion
Energy Dissipation Quantification
Wear Assessment
Drilling Parameter Optimization

PDC Bit Cutting Structure Parametric Modeling and Optimization Tool

This tool builds a 3D bit model based on blade geometry equations. It uses explicit dynamics simulations to evaluate cutter-rock interactions and local wear characteristics. An implicit wear model extrapolates long-term structural degradation trends. The tool optimizes blade profiles, cutter layouts, and back-rake angles to improve wear distribution and cutting efficiency.

Parametric Geometry Modeling
Explicit Dynamics
Implicit Wear Model
Structural Optimization
Research

Publications

Under Review

PeSNO: A Physics-embedded Stream Neural Operator as a Warm-Start Preconditioner for Mantle Convection Solvers

PeSNO: A Physics-embedded Stream Neural Operator as a Warm-Start Preconditioner for Mantle Convection Solvers

This paper proposes a Stream Neural Operator for mantle convection simulation. It predicts stream functions to construct divergence-free velocity fields. The method serves as a warm-start preconditioner for numerical solvers. It reduces linear solver iterations by more than half.

Authors: Zehua Song, Yu Song, Jin Yang

Neural Operator
Warm-Start Preconditioning
Mantle Convection Simulation
Hierarchical Bayesian Physics-Informed Network Model for Open-Hole Wellbore Instability: From Parameter Uncertainty to Reliability Assessment

Hierarchical Bayesian Physics-Informed Network Model for Open-Hole Wellbore Instability: From Parameter Uncertainty to Reliability Assessment

This paper presents a physics-informed network with hierarchical Bayesian inference for open-hole wellbore instability analysis. The model quantifies parameter uncertainty and provides dynamic failure time profiles.

Authors: Zehua Song, Yu Song, Jin Yang, Kejin Chen, Jianqiao Zhang, Xing Wang

Physics-Informed Neural Networks
Bayesian Inference
Reliability Assessment
Intelligent Prediction and Optimization of Open-Hole Wellbore Multiphysics Stability: A Synergistic PINN-DRL Approach

Intelligent Prediction and Optimization of Open-Hole Wellbore Multiphysics Stability: A Synergistic PINN-DRL Approach

This paper proposes a synergistic framework combining PINN and deep reinforcement learning for wellbore multiphysics stability. It enables high-speed state prediction and drilling fluid parameter optimization.

Authors: Yu Song, Zehua Song, Jin Yang, Kejin Chen, Kun Jiang, Jizhou Tang

Physics-Informed Neural Networks
Deep Reinforcement Learning
Wellbore Multiphysics Stability

Published

Enhancing Energy Efficiency and Sustainability in Offshore Drilling through Real-Time Multi-Objective Optimization: Considering Lag Effects and Formation Variability

Enhancing Energy Efficiency and Sustainability in Offshore Drilling through Real-Time Multi-Objective Optimization: Considering Lag Effects and Formation Variability

This paper presents a multi-objective optimization framework for offshore drilling. It explicitly accounts for formation variability and measurement lag effects. Field data validation demonstrates improved drilling efficiency.

Authors: Yu Song, Zehua Song, Jin Yang, Kejin Chen, Jizhou Tang

Multi-Objective Optimization
Offshore Drilling
Energy Efficiency and Sustainability
A Multi-Objective Reinforcement Learning Framework for Real-Time Drilling Optimization Based on Symbolic Regression and Drilling Perception

A Multi-Objective Reinforcement Learning Framework for Real-Time Drilling Optimization Based on Symbolic Regression and Drilling Perception

This paper combines symbolic regression with deep reinforcement learning for real-time drilling parameter optimization. The method achieves adaptive optimization across different operating conditions.

Authors: Zehua Song, Yu Song, Jin Yang, Baosheng Liu, Bingzhen Gao, Jizhou Tang

Symbolic Regression
Deep Reinforcement Learning
Real-Time Drilling Optimization
A Real-Time Inversion Framework for Carbon Equivalent Emissions in Oil and Gas Extraction Based on Vision Transformer

A Real-Time Inversion Framework for Carbon Equivalent Emissions in Oil and Gas Extraction Based on Vision Transformer

This paper presents a Vision Transformer-based real-time inversion framework for carbon emissions quantification in oil and gas extraction. The framework enables spatial distribution analysis of carbon equivalent emissions.

Authors: Zehua Song, Xiaoyang Yu, Yu Song, Jin Yang, Dongsheng Xu, Kejin Chen, Fangfei Huang, Bin Chen, Yanwei Song

Vision Transformer
Real-Time Inversion
Carbon Emissions Quantification
Multi-Objective Optimization Framework for Deepwater Riser Jetting Installation Parameters Using Deep Reinforcement Learning

Multi-Objective Optimization Framework for Deepwater Riser Jetting Installation Parameters Using Deep Reinforcement Learning

This paper presents a deep reinforcement learning optimization framework for deepwater riser jetting installation. It integrates jetting time prediction models with mechanical mechanism analysis.

Authors: Yu Song, Zehua Song, Jin Yang, Lei Li

Deep Reinforcement Learning
Multi-Objective Optimization
Deepwater Riser Jetting
Early Warning of Deep-Water Drilling Influx Based on Machine Learning

Early Warning of Deep-Water Drilling Influx Based on Machine Learning

This paper develops a machine learning method for early warning of deep-water drilling influx. The approach incorporates bypass line signal enhancement and lag elimination strategies.

Authors: Qishuai Yin, Zehua Song, Kejin Chen, Xu Zhou, Mayank Tyagi, Li Li

Machine Learning
Early Warning
Deep-Water Drilling
A Parzen-Estimator-Enhanced Bayesian Optimization Framework for Deepwater Conductor Jetting

A Parzen-Estimator-Enhanced Bayesian Optimization Framework for Deepwater Conductor Jetting

This paper proposes a segmented Bayesian optimization framework for deepwater conductor jetting process optimization. It combines global search with local refinement strategies.

Authors: Yu Song, Jin Yang, Zehua Song

Bayesian Optimization
Global Search and Local Refinement
Deepwater Conductor Jetting

Co-authored

Deep Learning Based Early Warning Methodology for Gas Kick of Deepwater Drilling Using Pilot-Scale Rig Data

Authors: Qishuai Yin, Qikang Zhu, Zehua Song, et al.

Deep Learning
Early Warning
Gas Kick Detection

Research on Pore Pressure Prediction Technology of HTHP Wells in South China Sea Based on Machine Learning

Authors: Dongsheng Xu, Jin Yang, ..., Zehua Song, et al.

Machine Learning
Pore Pressure Prediction

Digital Twin: Hybrid Virtual Simulation and Physical Monitoring Strategy for Subsea Wellhead Fatigue Assessment

Authors: Kejin Chen, Jin Yang, ..., Zehua Song, et al.

Digital Twin
Fatigue Assessment

Deep Ocean Hydrographic Element Acquisition Device Based on Edge Computing and Real-Time Data Processing

Authors: Bingzhen Gao, Jin Yang, ..., Zehua Song, et al.

Edge Computing
Real-Time Data Processing

In Preparation

Adaptive Drilling Decision-Making Framework for Extended-Reach Wells under Multiple Objectives

Adaptive Drilling Decision-Making Framework for Extended-Reach Wells under Multiple Objectives

Authors: Yu Song, Zehua Song, ...

Multi-Objective Decision Making
Extended-Reach Wells

Skills

Python
MATLAB
C#
PyTorch
TensorFlow
Ansys
OrcaFlex
COMSOL

Internship Experience

C

CNOOC Ltd.

Hainan & Tianjin | Summer 2023 & 2025
Visiting Student Researcher
K

Kingsoft Cloud

Beijing | 07/2021 - 09/2021
Data Science Intern
C

CNPC Xinjiang Oilfield

Karamay | 07/2019 - 08/2019
Field Engineering Intern