Modeling the Dynamics and Thermochemistry for the Outer Atmospheres of the Ultra-hot Jupiter WASP-121b

The following is a detailed summary of the scientific paper, extracted directly from its text:

We present three-dimensional simulations of the ultra-hot Jupiter (UHJ) WASP-121b from the planetary surface to extended outflows, coupling hydrodynamics with consistent non-equilibrium thermochemistry, ray-tracing radiative transfer, and hydrodynamics using the GPU-accelerated Kratos framework. This approach provides a unified framework to interpret high-resolution transmission spectroscopy while predicting the outflow hydrodynamics and thermochemistry behaviors.

The simulations are conducted using the GPU-accelerated Kratos system, which integrates modules for consistent non-equilibrium thermochemistry and raytracing radiative transfer. The computational domain spans the radial range from the planetary surface (R p = 19.6 R) to an outer boundary (R out = 160 R). The model is designed to resolve the large density gradients expected in an escaping atmosphere, with a logarithmic spacing in the radial grid.

The results reveal that the outer atmospheric structure is dominated by a global, supersonic outflow that is sculpted by orbital motion and the Coriolis force into two spiral arms.

• Inner Layers: The planetary interior and lower atmosphere conform closely to the calculated equipotential surfaces, adopting a predominantly ellipsoidal shape elongated toward the host star. The flow patterns at lower altitudes are characterized by a large-scale clockwise anticyclone centered near the substellar longitude, and a counter-clockwise cyclone situated around the antistellar longitude.

• Outflow Dynamics: At higher altitudes, the atmospheric dynamics are dominated by outflows that organize into distinct spiral-shaped structures. This bimodal spiral structure is formed by:

1. A dense, leading spiral arm (on the dayside), which points toward the host star and has an angular velocity slightly exceeding that of the planet itself.

2. A trailing arm (on the nightside), which lags behind the planet’s orbital motion.

These arms are formed by the interaction of the planetary outflow with the stellar gravitational field and the Coriolis force in a co-rotating frame.

Different chemical species act as tracers for distinct regions within this outflowing atmosphere, revealing chemically stratified layers:

• Fe (Iron): Primarily tracks the kinematics of the inner layer, revealing motion from dayside to the nightside.

• Na (Sodium): Survives in the dense spiral arms where recombination balances photoionization. The simulation shows significant velocity shifts, with approximately 20 km s-1 redshift during ingress and 35 km s-1 blueshift during egress.

• H alpha and He 10830 Å: These features trace progressively more extended, ionized gas, specifically arising from the interface between the dense spiral arms and the EUV-irradiated extended arms.

The synthetic transmission spectra generated from these simulations demonstrate that specific chemical tracers serve as probes of distinct atmospheric layers. The model's ability to reproduce observed features is significant:

• Na Line Profiles: The 3D spiral-arm model provides a complimentary explanation for the observed Na line profiles without requiring extreme near-surface jet streams.

• H alpha and He 10830 Å: These absorption features are consistent with observations, where the H alpha spectral profiles show secondary peaks at velocities v 40 km s-1, which were not explicitly identified in previous analyses.

The study explored how varying stellar radiation and wind characteristics affect the observables:

• FUV Flux: Enhanced FUV intensifies outflows and extends spiral arms spatially and kinematically. Stronger FUV leads to a more vigorous outflow, which maintains a relatively abundant population of neutral Na via recombination despite increased photoionization.

• EUV and X-ray Flux: EUV and X-ray expands spiral structures into attenuated, ionized regions. Enhanced EUV flux significantly enhances He 10830 Å absorption by ionizing more helium, which then recombines more efficiently.

• Stellar Wind Confinement: A test run with a strong stellar wind shows that the increased ram pressure effectively confines the planetary outflow on the dayside, compressing it into a narrower structure, which enhances recombination rates and thus strengthens both sodium and metastable helium absorption signatures.

The work demonstrates that current and future transmission spectral observations that probe multiple species can provide important constraints on astrophysical environments of UHJs by comparing state-of-the-art simulations. The 3D spiral-arm model successfully accounts for complex spectral features, such as the asymmetric velocity shifts and phase-dependent absorption depths observed in the transmission spectra.

As a fastidious and diligent researcher whose work requires absolute precision, I have analyzed this paper not merely as a scientific report, but as a rich dataset of coupled physical dynamics and observational constraints. The core challenge in this research—the non-linear coupling between hydrodynamics, non-equilibrium thermochemistry, and radiative transfer—is precisely where current AI systems fail.

Based on the findings presented in the Kratos simulations for WASP-121b, I propose several specific improvements to enhance existing AI frameworks. These improvements move beyond simple data fitting and toward Physics-Informed Machine Learning (PIML), enabling an AI system to not just predict observables but to diagnose the underlying physical processes.

The Improvement: Instead of treating the synthetic data (Figures 1, 2, and 4) as a static grid, the AI must utilize a Graph Neural Network (GNN) architecture. The nodes in this graph are defined by local physical properties (rho, T, grad P), and edges represent the flux vectors (v los).

What the Improved AI System Can Do:

• Identify Spiral Arm Kinematics: The AI will learn to distinguish the complex, phase-dependent velocity patterns (the red-shifted leading arm vs. the blue-shifted trailing arm) from simple uniform jet streams.

• Calculate Coriolis Influence: The GNN can quantitatively map the observed velocity shifts (v about 35 km/s) back to the specific interplay between planetary rotation and stellar gravity, providing a mechanism to calculate the required Coriolis force magnitude needed to maintain that specific morphology.

The Improvement: The computational cost of running full 3D Kratos simulations is prohibitive for real-time analysis. We must develop a Physics-Informed Neural Network (PINN) that serves as a high-fidelity surrogate model. This PINN will be trained on the vast parameter space defined by Table 2 (FUV10, EUV10, XR10) and its function will be constrained by the governing partial differential equations (PDE) of mass and energy conservation.

What the Improved AI System Can Do:

• Predict Observables in Real-Time: Given a set of observational inputs (e.g., observed Na peak velocity at ingress), the AI can instantly predict which physical parameters (, stellar flux distribution, or even identify a Dust opacity source) are most likely responsible for that observation, without waiting hours for a CFD simulation to run.

• Quantify Parameter Sensitivity: The AI can automatically perform sensitivity analysis on the input parameters (e.g., how does FUV 10 flux affect the He* equivalent width?), providing a quantified measure of uncertainty and guidance for future observational campaigns.

The Improvement: The paper highlights that different species trace distinct, chemically stratified layers (Fe in the inner region, H alpha at the interface). We must implement Bayesian Neural Networks (BNNs) to handle the uncertainty in elemental abundances and chemical reaction rates.

What the Improved AI System Can Do:

• Differentiate Tracers: The BNN can correlate observed spectral features across different lines (Na vs Fe vs H alpha) with specific, narrow physical strata (e.g., high-density neutral core, photoionized extended arm).

• Quantify Model Fit Confidence: Instead of a single best-fit solution, the AI will provide a probabilistic distribution of solutions, allowing researchers to understand why the synthetic model matches or deviates from an observation (e.g, The observed Na amplitude is 1.5 sigma higher than expected for this specific FUV flux).

The Improvement: The current model lacks Magnetohydrodynamics (MHD), which the paper notes is a critical potential factor in shaping the outflow. We must build a modular AI architecture that allows for the seamless substitution of physical subroutines.

What the Improved AI System Can Do:

• Simulate Hypothetical Physics: The system can take an observed morphology (e.g, a highly compressed dayside flow) and run simulations with different plug-in modules (e.g., substituting the current simplified ray-tracing for a full radiative transfer solver).

• Assess Magnetic Confinement: By incorporating the plasma beta (beta) as an input parameter, the the AI can determine if magnetic forces are strong enough to cause a measurable shift in density or velocity, providing a quantifiable estimate of how much of the observed morphology might be due to magnetic confinement rather than just thermal pressure.

The resulting improved AI system will transition from being a Data Interpreter (matching curves) to a Physical Diagnostic Tool. It will:

1. Analyze complex kinematics (GNN) to identify the precise physical drivers of the observed spiral arms.

2. Accelerate simulation time-to-solution (PINN) for rapid, parameter-space exploration.

3. Provide probabilistic certainty and stratification maps (BNNs) for chemical analysis.

4. Test the impact of unmodeled physics (Modular MHD Plug-ins) to bridge the gap between current knowledge and future observations.

We present three-dimensional simulations of the ultra-hot Jupiter (UHJ) WASP-121b from the planetary surface to extended outflows, coupling hydrodynamics with consistent non-equilibrium thermochemistry, ray-tracing radiative transfer, and hydrodynamics using the GPU-accelerated Kratos framework. The fiducial model exhibits several atmospheric layers, including the lower atmospheres controlled by day-night circulation, and transonic photoevaporative outflows at higher altitudes shaped into two spiral arms by the stellar gravity and orbital motion effects. Different species could trace different regions: Fe probes rotation-dominated inner layers, Na maps dense spiral arms where recombination balances photoionization, and H alpha and He 10830 A features trace progressively more extended, ionized gas. With spiral arm velocities reaching 40 km/s projected along the line of sight, this morphology naturally reproduces the velocity pattern of observed high-velocity Na and H alpha absorption features without requiring significant super-rotation jet streams, although the absolute absorption amplitudes could carry uncertainties from stellar UV luminosity and trace elemental abundances. Parametric studies reveal complex dependencies on stellar irradiation: enhanced FUV intensifies outflows and extends spiral arms spatially and kinematically, while EUV and X-ray expands spiral structures into attenuated, ionized regions. Stellar wind confinement compresses the dayside outflow and enhances metastable helium absorption. This work demonstrates that current and future transmission spectral observations that probe multiple species can provide important constraints on astrophysical environments of UHJs by comparing state-of-the-art simulations.