Argonne AI for Science Colloquium Marks Challenges and Progress

By John Russell

November 9, 2021

It’s an understatement to say the effort to adapt AI technology for use in scientific computing has gained steam. Last spring, the Department of Energy released a formal report – AI for Science – suggesting an AI program not unlike the exascale program reaching fruition now. There’s also the broader U.S. National Artificial Intelligence Initiative pushing for AI use throughout society. Last week, as part of a year-long celebration of its 75th founding anniversary, Argonne National Laboratory held a Director’s Special Colloquium on AI for Science: From Atoms to the Cosmos.

Rick Stevens, ANL

Led by Rick Stevens (Argonne’s associate laboratory director for computing, environment and life sciences) and featuring a keynote and panel, the virtual meeting provided an interesting glimpse into both progress and challenges for AI use in science. It was the fourth colloquium in a series. The others tackled: Decarbonization Within Reach; The Quantum Revolution; and Energy Storage for a Changing World.

Jonathan Rowe, University of Birmingham

Keynote speaker Jonathan Rowe, a prominent computer scientist and mathematician with posts at the University of Birmingham and the Alan Turing Institute, set the stage: “Often, when I give talks on AI, I concentrate on all the great things that AI methods have already done in different scientific areas. But today, I’m going to do something slightly different, and we’ll talk about a number of challenges that that still remain in trying to make AI even better and more effective in helping with scientific research. We’ve made some progress on some, we’ve got some ideas about [some] and many, I think we’re still scratching our heads on.”

With that he plunged ahead, calling out ten AI challenges. (A recording of the full ANL colloquium is posted.) Perhaps number one wasn’t surprising – Data Management.

Considered boring by too many, said Rowe, “it’s essential that we start getting to grips with issues to do with data management. There’s all sorts of basic questions about how we manage data [of] that scale. For example, a very basic question is: what should you keep, and what should you throw away when you’re producing data? As an example, CERN routinely throw away nearly all the data that they generate from each experimental run because they simply can’t store everything they produce. They can barely store the essentials. They only store what’s necessary to support the conclusion they’re writing about as a result of the experiment.”

“So what does that mean? That means if you come up with an alternative theory about what might have happened in a particular experiment, you can’t go back and check the data. You have to ask to rerun the experiment and collect what you need for it. So that’s becoming a really big issue. We can generate so much data, but you have to somehow decide what are you going to keep and what are you going to throw away? That raises the question, ‘what [does] open access to data actually mean?’ The big programs like Square Kilometer Array, LSST (Large Synoptic Survey Telescope), they’re publicly funded, and they all promise their data is going to be open access. So LSST will produce something like 15 terabytes of data for one night’s observation. Could someone in the world just ask for that? It’s just not practical,” said Rowe.

“The second challenge is the whole question of how we incorporate scientific knowledge into our AI. The whole purpose of doing science is understanding and generating more knowledge. AI is really good at predicting stuff based on data and that’s useful in a lot of situations. But scientists are not satisfied with just being able to predict stuff. They want to be able to understand the system that’s producing the prediction. That’s really tricky. How can we use our existing scientific knowledge to make sure the AI methods produce better results? How can we make sure the results are even scientifically valid?

“One idea, which I guess quite a few people do, is to incorporate the current physical constraints or scientific constraints into your loss function when you’re when you’re doing your neural network. The example I’ve actually got here is from the lab for molecular biology, where they’re using some Bayesian system, and then incorporate information about molecular structure as a constraint in the Bayesian optimization system. That kind of gives you results that are, you know, physically more realistic than if you don’t do that, but still doesn’t necessarily produce stuff that really obeys the laws of physics,” he said.

“Another idea is to incorporate your physical understanding via a model or a simulation. What I’m showing here is the output of a system we developed with British Antarctic Survey for predicting at the Arctic sea ice. We trained it on hundreds of years’ worth of data, because that data was artificially generated through a physics based model. Then we fine-tuned it with actual observations. But that meant we’ve got a system that really does tend to conform to what’s known physically about the Arctic,” said Rowe.

Here’s the list of challenges that Rowe discussed: data management; scientific knowledge; uncertainty and noise; finding rare things; hidden structures; finding all of the 3D structures in a volume; counting and tracking; AI for digital twins; benchmarking; and closing the loop.

Rowe explored each challenge area and briefly discussed a few solution approaches being explored. One topic that touched on all of the challenges was benchmarking.

“We’ve got lots of different AI methods now. And we’ve got lots of different scientific data sets. What you’d like to be able to do is to work out for each different kind of data set, and for each different kind of scientific question, what are the best AI methods that are available. Or if you’re a computer scientist, and you’ve come up with what you think is a really neat AI method, you’d like to know how it compares to some of the other ones. And this is really hard right now,” said Rowe. “Similarly, the datasets are produced in different labs around the world and just kind of sit there. Somehow, you need to get these together, and you need to get them together in a way that makes it very easy to do comparisons and benchmarking.”

Rowe noted there is a fair amount of work is being done around benchmarking and cited work by the SciML group. “I want to call out one because this is done by the guys from the scientific machine learning group at the Rutherford [Appleton] labs, who we collaborate with. They’ve started putting forward something called SciML bench available on GitHub, which is really good start putting together a framework to do this. And they’ve got data now from environmental sciences, particle physics, astronomy, and so forth, where you can begin to do this benchmarking, if you’re interested in that. Please go and check it out and see how we might be able to add to it and help,” he said.

(An excerpt from the SciML website showing its tools is included at the end of the article)

The panel was also fascinating. Besides Stevens and Rowe, panelists included: Patrick Riley who leads the AI group at Relay Therapeutics, applying learning methods to the discovery process; Douglas Finkbeiner, professor of Astronomy and Physics at Harvard University; Subramanian Sankaranarayanan, group leader of the theory and modeling group in the Nanoscience and Technology division at ANL; and Rebecca Willett, a professor of statistics and computer science at the University of Chicago.

It’s best to watch the video directly to catch the interplay. One interesting current use of AI was cited by Douglas Finkbeiner of Harvard University. “For example, we want to see the dark mass of the universe. [Using] gravitational lensing, we can see the distortions in background objects behind the mass that requires measurements of the shapes and brightnesses of billions of galaxies,” said Finkbeiner. “Two ways we’ve used AI for that are to keep the telescope system in focus and to de-blend galaxies.”

“Jonathan [Rowe] mentioned the LSST survey at the Bureau Rubin telescope that will produce several petabytes of data over the years. I’ve probably ordered 1000 detections [for] each of tens of billions of objects. So it’s quite a bit of data. [J]ust keeping a telescope like that very complex eight-meter optical system in focus is a bit of a challenge. There are 50 control parameters in the optical system that need to be fixed correctly. It turns out, there’s a nice way to do that with convolutional neural nets. Then once you’ve actually got the images, you can pull out information about the individual galaxies. That’s not so hard if that galaxy is just off by itself isolated, but often these galaxies are kind of overlapping each other, and that is much more of a challenge. We’ve been applying convolutional neural nets to deep-learning the galaxies.”

Link to video: https://www.youtube.com/watch?v=sUYCCfdJkjM

Link to ANL website hosting AI in Science colloquia material: https://www.anl.gov/event/ai-for-science-from-atoms-to-the-cosmos

EXCERPT FROM SCIML WEBSITE

SciML: Open Source Software for Scientific Machine Learning

SciML is a NumFOCUS sponsored open source software organization created to unify the packages for scientific machine learning. This includes the development of modular scientific simulation support software, such as differential equation solvers, along with the methodologies for inverse problems and automated model discovery. By providing a diverse set of tools with a common interface, we provide a modular, easily-extendable, and highly performant ecosystem for handling a wide variety of scientific simulations.

Core Components

High Performance and Feature-Filled Differential Equation Solving. The library DifferentialEquations.jl is a library for solving ordinary differential equations (ODEs), stochastic differential equations (SDEs), delay differential equations (DDEs), differential-algebraic equations (DAEs), and hybrid differential equations which include multi-scale models and mixtures with agent-based simulations. The templated implementation allows arbitrary array and number types to be compatible, giving compatibility with arbitrary precision floating point numbers, GPU-based computations, unit-checked arithmetic, and other features. DifferentialEquations.jl is designed for both high performance on large-scale and small-scale problems, and routinely benchmarks at the top of the pack.

Physics-Informed Model Discovery and Learning. SciML contains a litany of modules for automating the process of model discovery and fitting. Tools like DiffEqParamEstim.jl and DiffEqBayes.jl provide classical maximum likelihood and Bayesian estimation for differential equation based models, while DiffEqFlux.jl enables the training of embedded neural networks inside of differential equations (neural differential equations or universal differential equations) for discovering unknown dynamical equations, DataDrivenDiffEq.jl estimates Koopman operators (DMD) and utilizes methods like SInDy to turn timeseries data into LaTeX for driving differential equations, and ReservoirComputing.jl for Echo State Networks that learn to predict the dynamics of chaotic systems.

A Polyglot Userbase. While the majority of the tooling for SciML is built using the Julia programming language, SciML is committed to ensure that these methodologies can be used throughout the greater scientific community. Tools like diffeqpy and diffeqr bridge the DifferentialEquations.jl solvers to Python and R respectively, and we hope to see many more developments along these lines in the near future.

Compiler-Assisted Model Analysis and Sparsity Acceleration. Scientific models generally have structures like locality which leads to sparsity in the program structures that can be exploited for major performance acceleration. The SciML builds a set of interconnected tools for generating numerical solver code directly on the models that are being simulated. SparsityDetection.jl can automatically detect the sparsity patterns of Jacobians and Hessians from arbitrary source code, while ModelingToolkit.jl can rewrite differential equation models to re-arrange equations for better stability and automatically parallelize code. These tools then connect with affiliated packages like SparseDiffTools.jl to accelerate solving with DifferentialEquations.jl and training with DiffEqFlux.jl.

ML-Assisted Tooling for Model Acceleration. SciML supports the development of the latest ML-accelerated toolsets for scientific machine learning. Methods like Physics-Informed Neural Networks (PINNs) and Deep BSDE methods for solving 1000 dimensional partial differential equations are productionized in the NeuralPDE.jl library. Surrogate-based acceleration methods are provided by Surrogates.jl.

Differentiable Scientific Data Structures and Simulators. The SciML ecosystem contains pre-built scientific simulation tools along with data structures for accelerating the development of models. Tools like LabelledArrays.jl and MultiScaleArrays.jl make it easy to build large-scale scientific models, while other tools like NBodySimulator.jl provide full-scale simulation simulators.

Tools for Accelerated Algorithm Development and Research. SciML is an organization dedicated to helping state-of-the-art research in both numerical simulation methods and methodologies in scientific machine learning. Many tools throughout the organization automate the process of benchmarking and testing new methodologies to ensure they are safe and battle tested, both to accelerate the translation of the methods to publications and to users. We invite the larger research community to make use of our tooling like DiffEqDevTools.jl and our large suite of wrapped algorithms for quickly test and deploying new algorithms.

Subscribe to HPCwire's Weekly Update!

Be the most informed person in the room! Stay ahead of the tech trends with industry updates delivered to you every week!

AI Saves the Planet this Earth Day

April 22, 2024

Earth Day was originally conceived as a day of reflection. Our planet’s life-sustaining properties are unlike any other celestial body that we’ve observed, and this day of contemplation is meant to provide all of us Read more…

Intel Announces Hala Point – World’s Largest Neuromorphic System for Sustainable AI

April 22, 2024

As we find ourselves on the brink of a technological revolution, the need for efficient and sustainable computing solutions has never been more critical.  A computer system that can mimic the way humans process and s Read more…

Empowering High-Performance Computing for Artificial Intelligence

April 19, 2024

Artificial intelligence (AI) presents some of the most challenging demands in information technology, especially concerning computing power and data movement. As a result of these challenges, high-performance computing Read more…

Kathy Yelick on Post-Exascale Challenges

April 18, 2024

With the exascale era underway, the HPC community is already turning its attention to zettascale computing, the next of the 1,000-fold performance leaps that have occurred about once a decade. With this in mind, the ISC Read more…

2024 Winter Classic: Texas Two Step

April 18, 2024

Texas Tech University. Their middle name is ‘tech’, so it’s no surprise that they’ve been fielding not one, but two teams in the last three Winter Classic cluster competitions. Their teams, dubbed Matador and Red Read more…

2024 Winter Classic: The Return of Team Fayetteville

April 18, 2024

Hailing from Fayetteville, NC, Fayetteville State University stayed under the radar in their first Winter Classic competition in 2022. Solid students for sure, but not a lot of HPC experience. All good. They didn’t Read more…

AI Saves the Planet this Earth Day

April 22, 2024

Earth Day was originally conceived as a day of reflection. Our planet’s life-sustaining properties are unlike any other celestial body that we’ve observed, Read more…

Kathy Yelick on Post-Exascale Challenges

April 18, 2024

With the exascale era underway, the HPC community is already turning its attention to zettascale computing, the next of the 1,000-fold performance leaps that ha Read more…

Software Specialist Horizon Quantum to Build First-of-a-Kind Hardware Testbed

April 18, 2024

Horizon Quantum Computing, a Singapore-based quantum software start-up, announced today it would build its own testbed of quantum computers, starting with use o Read more…

MLCommons Launches New AI Safety Benchmark Initiative

April 16, 2024

MLCommons, organizer of the popular MLPerf benchmarking exercises (training and inference), is starting a new effort to benchmark AI Safety, one of the most pre Read more…

Exciting Updates From Stanford HAI’s Seventh Annual AI Index Report

April 15, 2024

As the AI revolution marches on, it is vital to continually reassess how this technology is reshaping our world. To that end, researchers at Stanford’s Instit Read more…

Intel’s Vision Advantage: Chips Are Available Off-the-Shelf

April 11, 2024

The chip market is facing a crisis: chip development is now concentrated in the hands of the few. A confluence of events this week reminded us how few chips Read more…

The VC View: Quantonation’s Deep Dive into Funding Quantum Start-ups

April 11, 2024

Yesterday Quantonation — which promotes itself as a one-of-a-kind venture capital (VC) company specializing in quantum science and deep physics  — announce Read more…

Nvidia’s GTC Is the New Intel IDF

April 9, 2024

After many years, Nvidia's GPU Technology Conference (GTC) was back in person and has become the conference for those who care about semiconductors and AI. I Read more…

Nvidia H100: Are 550,000 GPUs Enough for This Year?

August 17, 2023

The GPU Squeeze continues to place a premium on Nvidia H100 GPUs. In a recent Financial Times article, Nvidia reports that it expects to ship 550,000 of its lat Read more…

Synopsys Eats Ansys: Does HPC Get Indigestion?

February 8, 2024

Recently, it was announced that Synopsys is buying HPC tool developer Ansys. Started in Pittsburgh, Pa., in 1970 as Swanson Analysis Systems, Inc. (SASI) by John Swanson (and eventually renamed), Ansys serves the CAE (Computer Aided Engineering)/multiphysics engineering simulation market. Read more…

Intel’s Server and PC Chip Development Will Blur After 2025

January 15, 2024

Intel's dealing with much more than chip rivals breathing down its neck; it is simultaneously integrating a bevy of new technologies such as chiplets, artificia Read more…

Choosing the Right GPU for LLM Inference and Training

December 11, 2023

Accelerating the training and inference processes of deep learning models is crucial for unleashing their true potential and NVIDIA GPUs have emerged as a game- Read more…

Baidu Exits Quantum, Closely Following Alibaba’s Earlier Move

January 5, 2024

Reuters reported this week that Baidu, China’s giant e-commerce and services provider, is exiting the quantum computing development arena. Reuters reported � Read more…

Comparing NVIDIA A100 and NVIDIA L40S: Which GPU is Ideal for AI and Graphics-Intensive Workloads?

October 30, 2023

With long lead times for the NVIDIA H100 and A100 GPUs, many organizations are looking at the new NVIDIA L40S GPU, which it’s a new GPU optimized for AI and g Read more…

Shutterstock 1179408610

Google Addresses the Mysteries of Its Hypercomputer 

December 28, 2023

When Google launched its Hypercomputer earlier this month (December 2023), the first reaction was, "Say what?" It turns out that the Hypercomputer is Google's t Read more…

AMD MI3000A

How AMD May Get Across the CUDA Moat

October 5, 2023

When discussing GenAI, the term "GPU" almost always enters the conversation and the topic often moves toward performance and access. Interestingly, the word "GPU" is assumed to mean "Nvidia" products. (As an aside, the popular Nvidia hardware used in GenAI are not technically... Read more…

Leading Solution Providers

Contributors

Shutterstock 1606064203

Meta’s Zuckerberg Puts Its AI Future in the Hands of 600,000 GPUs

January 25, 2024

In under two minutes, Meta's CEO, Mark Zuckerberg, laid out the company's AI plans, which included a plan to build an artificial intelligence system with the eq Read more…

China Is All In on a RISC-V Future

January 8, 2024

The state of RISC-V in China was discussed in a recent report released by the Jamestown Foundation, a Washington, D.C.-based think tank. The report, entitled "E Read more…

Shutterstock 1285747942

AMD’s Horsepower-packed MI300X GPU Beats Nvidia’s Upcoming H200

December 7, 2023

AMD and Nvidia are locked in an AI performance battle – much like the gaming GPU performance clash the companies have waged for decades. AMD has claimed it Read more…

Nvidia’s New Blackwell GPU Can Train AI Models with Trillions of Parameters

March 18, 2024

Nvidia's latest and fastest GPU, codenamed Blackwell, is here and will underpin the company's AI plans this year. The chip offers performance improvements from Read more…

Eyes on the Quantum Prize – D-Wave Says its Time is Now

January 30, 2024

Early quantum computing pioneer D-Wave again asserted – that at least for D-Wave – the commercial quantum era has begun. Speaking at its first in-person Ana Read more…

GenAI Having Major Impact on Data Culture, Survey Says

February 21, 2024

While 2023 was the year of GenAI, the adoption rates for GenAI did not match expectations. Most organizations are continuing to invest in GenAI but are yet to Read more…

The GenAI Datacenter Squeeze Is Here

February 1, 2024

The immediate effect of the GenAI GPU Squeeze was to reduce availability, either direct purchase or cloud access, increase cost, and push demand through the roof. A secondary issue has been developing over the last several years. Even though your organization secured several racks... Read more…

Intel’s Xeon General Manager Talks about Server Chips 

January 2, 2024

Intel is talking data-center growth and is done digging graves for its dead enterprise products, including GPUs, storage, and networking products, which fell to Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire