Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning

summary

Video file (mp4)

The gist

The paper presents a machine learning-based probabilistic emulator of the Community Radiative Transfer Model (CRTM), applied to the GOES Advanced Baseline Imager (ABI).

This episode discusses

The paper

Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning · Read on arXiv

Lucas Howard, Aneesh C. Subramanian, Gregory Thompson, Benjamin Johnson, Thomas Auligne

University of Colorado, Boulder · University Center for Atmospheric Research, Joint Center for Satellite Data Assimilation

The continuous improvement in weather forecast skill over the past several decades is largely due to the increasing quantity of available satellite observations and their assimilation into operational forecast systems. Assimilating these observations requires observation operators in the form of radiative transfer models. Significant efforts have been dedicated to enhancing the computational efficiency of these models. Computational cost remains a bottleneck, and a large fraction of available data goes unused for assimilation. To address this, we used machine learning to build an efficient neural network based probabilistic emulator of the Community Radiative Transfer Model (CRTM), applied to the GOES Advanced Baseline Imager. The trained NN emulator predicts brightness temperatures output by CRTM and the corresponding error with respect to CRTM. RMSE of the predicted brightness temperature is 0.3 K averaged across all channels. For clear sky conditions, the RMSE is less than 0.1 K for 9 out of 10 infrared channels. The error predictions are generally reliable across a wide range of conditions. Explainable AI methods demonstrate that the trained emulator reproduces the relevant physics, increasing confidence that the model will perform well when presented with new data.

DOI: 10.1175/AIES-D-24-0109.1

Transcript

Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.

Tom: Next we'll be talking about the paper "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning".

Jane: The paper was written by Lucas Howard, Aneesh C. Subramanian, Gregory Thompson, Benjamin Johnson and Thomas Auligne from University of Colorado, Boulder and University Center for Atmospheric Research, Joint Center for Satellite Data Assimilation.

Tom: Stay tuned as we take you through the paper and discuss its implications.

Title: Tom: Welcome back to the show, everyone. Today we're digging into a paper called "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning." Jane, I have to say, that title is a mouthful, but the idea behind it is actually pretty straightforward once you unpack it.

Jane: Absolutely, Tom. So the Community Radiative Transfer Model, or CRTM, is basically the tool that weather forecasters use to simulate what satellites see. When a satellite looks down at Earth, it's measuring radiation, and CRTM helps translate our weather model's predictions into what that radiation should look like. It's the bridge between the model and the observations.

Tom: Right, and that bridge is essential for data assimilation, which is how we feed satellite observations into weather forecasts to improve them. But here's the catch — CRTM is slow. Really slow, compared to how much data we have. The paper says for some sensors, less than one percent of available observations actually get used.

Jane: That's wild. We're throwing away ninety-nine percent of the data because we can't process it fast enough. And that's where machine learning comes in. The idea is to train a neural network to mimic what CRTM does, but do it way faster. The title says "probabilistic" because it doesn't just predict a single brightness temperature — it also predicts how confident it is in that prediction.

Tom: And that's a big deal. If the emulator knows when it's uncertain, forecasters can decide to trust it or fall back on the full CRTM. It's like having a quick calculator that tells you when it's guessing versus when it's certain.

Jane: Exactly. The authors are from the University of Colorado and the Joint Center for Satellite Data Assimilation, so these are the folks actually building the operational tools. They're not just doing this for fun — they want to fix a real bottleneck in weather forecasting.

Tom: And that's what excites me. This isn't just a cool AI trick. It's a practical fix for a problem that's limiting how good our forecasts can be. More data used means better forecasts, and better forecasts save lives and money.

Jane: Well said, Tom. And the fact that they're targeting the GOES satellites, which watch the Western Hemisphere constantly, means this could have immediate impact on severe weather monitoring.

Tom: Alright, so we've got the big picture. Next up, we're going to look at what the paper actually did and how they built this thing.

Summary: Jane: So Tom, we're back with "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning." Let's talk about what the team actually did. They built a neural network that takes the same inputs as CRTM — things like temperature, water vapor, cloud properties, ozone, all across one hundred twenty-seven pressure levels in the atmosphere.

Tom: one hundred twenty-seven levels, that's a lot of data. Plus surface variables like soil temperature and snow cover, and even sensor angles. The paper lists over a thousand input variables total. That's a massive input space.

Jane: Right, and the network has three hidden layers with five hundred twelve nodes each. It's a dense, fully connected network. Nothing fancy architecturally, but that's intentional. They want speed, and a simple network is fast.

Tom: So what does it output? It predicts brightness temperatures for ten channels on the GOES Advanced Baseline Imager, channels seven through sixteen. And on top of that, it predicts an error standard deviation for each channel. That's the probabilistic part.

Jane: And they trained it on data generated by CRTM itself. They used the GFS weather model to create realistic atmospheric states, then ran CRTM to get the "true" brightness temperatures. The neural network learns to match those outputs.

Tom: So it's learning to imitate CRTM, not to predict real observations directly. That's an important distinction. The emulator is only as good as CRTM, but if CRTM is good enough for operational use, then a fast emulator that matches it is also good enough.

Jane: Exactly. And the results are impressive. The average error across all channels is about zero point three Kelvin. For clear sky conditions, nine out of ten infrared channels have errors under zero point one Kelvin. That's really precise.

Tom: And they also checked whether the predicted error is reliable. If the network says the error is small, is it actually small? And for the most part, yes. The calibration is good for the vast majority of cases.

Jane: They also used explainable AI to check that the network learned real physics, not just memorized the training data. We'll get into that more later, but for now, the summary is: they built a fast, accurate, probabilistic emulator of CRTM.

Tom: And that's a solid foundation. But the real question is, how fast is it? And can it actually replace CRTM in an operational setting? That's what we're going to dig into next.

Improvements: Tom: Alright Jane, so we've covered what the paper did. Now let's talk about what this paper suggests as improvements over the current state of things. The big one is speed. They benchmarked the neural network against CRTM on a single CPU, and the ML model was about five times faster.

Jane: Five times faster, and that's without any optimization. The paper notes that CRTM is also slower for cloudy conditions, so the speed advantage would be even bigger in those cases. And in an operational setting, you could use GPUs to make the neural network even faster.

Tom: Right, so the improvement isn't just marginal. It's a step change in computational efficiency. And that's what opens the door to using more of the available satellite data.

Jane: But there's another improvement that I think is even more important. The probabilistic output. Previous ML emulators of CRTM were deterministic — they just gave a point estimate. This one predicts the error too. That means you can set a threshold. If the predicted error is low, you trust the emulator. If it's high, you fall back on CRTM.

Tom: That's a really smart workflow. It's not about replacing CRTM entirely. It's about using the emulator where it's reliable and keeping CRTM as a safety net. The paper calls this out explicitly as a potential operational strategy.

Jane: And the error predictions are generally reliable. The calibration curves show that for most cases, the predicted error matches the actual error well. It only breaks down for very large, rare errors, and even then the relationship is still monotonic — bigger predicted errors still mean bigger actual errors.

Tom: So even in the tail, you can rank-order your confidence. That's useful.

Jane: Another improvement is the explainability. They used SHAP values to verify that the network is learning real physics. For example, the water vapor channels show sensitivity to water vapor at the expected altitudes. The ozone channel is sensitive to ozone. And channel seven which has a reflected solar component, is heavily influenced by solar zenith angle.

Tom: That's the kind of validation that gives you confidence the model will generalize. It's not just fitting noise. It's actually capturing the underlying radiative transfer physics.

Jane: So the improvements are: speed, probabilistic error prediction, and physics-based validation. That's a strong package. But the question is, what does this mean for actual weather forecasting? Let's talk about that.

First Page: Jane: So Tom, let's go back to the very beginning of "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning." The first page sets the stage with a really important problem. Weather forecast quality has improved a lot over the decades, and a big reason is satellite data assimilation.

Tom: Right, and the paper makes the point that we're not using most of that data. For some sensors, less than one percent of observations get assimilated. The computational cost of the observation operator — that's the tool that maps model states to what the satellite sees — is the bottleneck.

Jane: And it's going to get worse. The paper mentions that hyperspectral instruments are coming, and they have way more channels than current sensors. So the computational challenge is going to become more acute.

Tom: That's a sobering thought. We're about to get even more data, and we're already struggling to use what we have.

Jane: But the first page also sets up the solution. Machine learning has been used to create fast observation operators before, but those were deterministic. This paper's contribution is the probabilistic aspect, which is crucial for data assimilation because you need to know the error characteristics of your observation operator.

Tom: And that's what makes this paper stand out. It's not just "here's a faster model." It's "here's a faster model that also tells you when it's uncertain." That's what makes it usable in a real data assimilation system.

Jane: The authors also frame this as a practical path forward. They're not proposing a complete replacement for CRTM. They're proposing a hybrid approach where the ML emulator handles the bulk of the data, and CRTM handles the cases where the emulator is uncertain.

Tom: So the first page really lays out the motivation clearly. We have a problem — too much data, not enough compute. And we have a potential solution — probabilistic ML emulators. The rest of the paper is about proving that solution works.

Jane: And it does work, at least in this test case. The errors are small, the speedup is real, and the physics checks out. So the question is, what's next? How do we get from this research to operational use?

Tom: That's exactly what we should talk about. Let's wrap this up with the big picture.

Conclusion: Tom: Alright, let's bring it home. We've been talking about "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning." Jane, what's the one-sentence summary?

Jane: It's a neural network that mimics CRTM for the GOES ABI instrument, runs about five times faster, predicts its own error, and checks out on the physics. That's a big step toward using way more satellite data in weather forecasts.

Tom: And the implications are huge. Better forecasts mean better severe weather warnings, better flight planning, better agriculture decisions. All of that starts with using the data we already have more effectively.

Jane: And this paper shows a clear path. The probabilistic output is the key innovation. It's what makes the emulator trustworthy enough to use operationally. You can set a threshold, use the fast model where it's confident, and fall back on CRTM where it's not.

Tom: There's still work to do, of course. The paper notes that the error predictions get less reliable for very large errors, and the Jacobians diverge from CRTM at high altitudes where there's little water vapor. But those are refinements, not roadblocks.

Jane: And the explainable AI results give us confidence. The model is learning real physics, not just memorizing patterns. That means it's more likely to generalize to new conditions.

Tom: So we're saying goodbye to this paper, but the ideas in it are going to stick around. The combination of speed, probabilistic uncertainty, and physics validation is a template for future work in this area.

Jane: Absolutely. And as more instruments come online, this approach is going to become even more valuable. We're going to need fast, reliable emulators for all of them.

Tom: Well said, Jane. That's a wrap on "Probabilistic Emulation of the Community Radiative Transfer Model Using Machine Learning." Thanks for listening, and we'll see you next time with a fresh paper to dig into.

Jane: Take care, everyone.

More episodes

← Home