Engineering PapersSearch

SEARCH · Engineering Papers

Results for “weather prediction”

Search indexed NASA NTRS and DOE OSTI research on propulsion, heat transfer, battery materials and energy systems. Follow report and document links to the original sources.

Quote a phrase for an exact phrase match. Source license links do not imply unrestricted reuse.

429 records · Page 4

Wide-ranging predictions of new stable compounds powered by recommendation engines

The computational search for new stable inorganic compounds is faster than ever, thanks to high-throughput density functional theory (DFT). However, stable compound searches remain highly expensive because of the enormous search space and the cost of DFT calculations. To aid these searches, recommendation engines have been developed. We conduct a systematic comparison of the performance of previously developed recommendation engines, specifically ones based on elemental substitution, data mining, and neural network prediction of formation enthalpy. After identifying ways to improve the recommendation engines, we find the neural network to be superior at recommending stable Heusler compounds. Armed with improved recommendation engines, we identify tens of thousands of compounds that are stable at zero temperature and pressure, now available in the Open Quantum Materials Database. We summarize this diverse pool of compounds, including the elusive mixed anion compounds, and two of their many applications: thermoelectricity and solar thermochemical fuel production.

Science & Technology - Other Topics

Improvements to RANS Modeling for Aeroheating Predictions on Blunt Bodies

Accurate predictions of aeroheating are critical for designing thermal protection systems for planetary entry vehicles. For larger vehicles, turbulence in the boundary layer can substantially increase convective heating. This turbulence must be accurately modeled to ensure the thermal protection system is sufficient. The majority of hypersonic turbulence model development and validation focuses on boundary layers developing over flat-plates or sharp cones; these cases are substantially different than the boundary layer that develops over the heatshield of a blunt body traveling at hypersonic speeds. Planetary missions often use blunt body geometries, such as the 70-degree sphere-cone favored by Mars missions or the 45-degree sphere-cone planned for the upcoming DAVINCI mission. Due to smaller vehicle size and the lower velocities in the stagnation region, planetary entry vehicles have relatively low Reynolds numbers. Surface curvature and high enthalpy gradients create additional challenges. These difficulties must be addressed to obtain high accuracy needed for the ambitious planetary missions in the upcoming decade. This work focuses on both assessing and improving Reynolds-averaged Navier-Stokes (RANS) turbulence models for blunt-body geometries typical of planetary entry vehicles, with a focus on one-equation and two-equation formulations.

Mars2020

Electronic structure prediction of medium and high entropy alloys across composition space

We propose machine learning (ML) models to predict the electron density — the fundamental unknown of a material’s ground state — across the composition space of concentrated alloys. From this, other physical properties can be inferred, enabling accelerated exploration. A significant challenge is that the number of descriptors and sampled compositions required for accurate prediction grows rapidly with species. To address this, we employ Bayesian Active Learning (AL), which minimizes training data requirements by leveraging uncertainty quantification capabilities of Bayesian Neural Networks. Compared to the strategic tessellation of the composition space, Bayesian-AL reduces the number of training data points by a factor of 2.5 for ternary (SiGeSn) and 1.7 for quaternary (CrFeCoNi) systems. We also introduce easy-to-optimize, body-attached-frame descriptors, which respect physical symmetries while keeping descriptor-vector size nearly constant as alloy complexity increases. Our ML models demonstrate high accuracy and generalizability in predicting both electron density and energy across composition space.

materials science

Numerical Predictions of the Flow and Heat Transfer Characteristics in the Film Boiling Regime During Tube Quenching

Cryogenic fluid management plays a major role in refueling of spacecrafts while in space for NASA’s future human space exploration missions. Due to the low boiling points of cryogens, storage, transport and handling of these fluids becomes difficult and may result in inefficient operation of the space propulsion systems. For refueling applications in space, the cryogenic fluids have to be transported across different locations and hence, the transfer of cryogenic fluids through pipes become critical. The cryogenic chill-down process is characterized by different regimes of flow boiling, viz., film boiling, transition boiling and nucleate boiling. The prediction of these regimes in a single CFD framework available in the literature is challenging and the present work attempts to address this challenge by initially modeling the film boiling regime accurately and to incorporate an user-defined function for transition and nucleate boiling at a later stage. Hence, the aim of the present work is to numerically model and validate the film boiling regime of the chilldown curve for liquid nitrogen experiments available in the literature. The validations are carried out at different inlet mass fluxes to have a robust simulation methodology. A dispersed mixture model is used to predict the vapor-liquid interface dynamics with the phase change phenomena modeled using the Lee model.

line chilldown

Predicting Operational Performance of xEMU Boot at Lunar South Pole Temperatures using Thermal Desktop ®

The spacesuit boots that will be used on Artemis lunar south pole surface missions will be exposed to extremely cold temperatures (down to ~50 K). To assess the performance of the government’s Exploration Extravehicular Mobility Unit (xEMU) lunar boot in these permanently shadowed regions, testing was performed at the Jet Propulsion Lab (JPL) in the Cryogenic Ice Transfer, Acquisition Development, and Excavation Laboratory (CITADEL) thermal vacuum (TVAC) chamber. This paper documents the data analysis, thermal boot model correlation, and operational predictions conducted using data from the xEMU CITADEL TVAC test. Expected thermal conductances within the boot and between the boot and environment were calculated from test data, which was then used as an initial guess for conductances within a Thermal Desktop (TD) model. Correlation of the TD model using the internal SOLVER feature was performed across 10 different test points which varied external temperature, internal boot ventilation flowrate, and contact pressure. Operational performance at the lunar south pole was then predicted using results from the correlated model. While the predictions provide evidence for acceptable performance of the boots at the 100K environment test point, there is still substantial uncertainty in performance, especially at the 48K test point. This uncertainty is due in part to testing limitations such as contacting the foot to a hard metal plate rather than granular regolith, and model limitations such as the lack of a realistic foot model. These limitations and their impacts are addressed in detail in this paper. The results of this test series and model correlation underscore the importance of additional improved testing and modeling for characterizing the expected thermal resistance between the outside of the boot and the lunar surface.

Spacesuit

Next-Generation Energy Technologies for Connected and Automated On-Road Vehicles (NEXTCAR) - Predictive Data-Driven Vehicle Dynamics and Powertrain Control: from ECU to the Cloud (Final Scientific/Technical Report)

This project developed and demonstrated a predictive, data-driven vehicle control system designed to improve energy efficiency and driving performance. The team created intelligent self-driving car technology that optimizes fuel and electricity use by proactively planning vehicle actions. By combining Level 4 autonomous driving capabilities with vehicle-to-everything (V2X) connectivity, the system enables vehicles to adjust speed and change lanes in response to traffic signals, surrounding vehicles, and road conditions, reducing unnecessary stops and delays. In testing, the system improved vehicle fuel economy by more than 30% and reduced travel time by approximately 10%, compared to a conventional adaptive cruise control baseline. These results demonstrate the technical effectiveness of using predictive, V2X-enabled strategies, such as traffic light timing and surrounding traffic awareness, to inform real-time vehicle powertrain control and driving behavior. Additionally, a supporting cloud platform was developed to provide dispatch and route recommendations as well as to log vehicle data, demonstrating the economic feasibility of this approach at the fleet level. By optimizing dispatching and routing operations, this technology enables electric fleet operators to use their vehicles more efficiently and reduce reliance on diesel backups, lowering both operating costs and energy consumption. Overall, this project’s technology advances the future of clean, energy-efficient transportation, enabling vehicles and fleets to reduce energy waste, cut costs, and lower emissions through intelligent automation and connectivity.

33 ADVANCED PROPULSION SYSTEMS

A physically interpretable precursor framework for sub-seasonal prediction of Northern Hemisphere flash flourishing

Flash flourishing describes rapid vegetation increases that can quickly reshape land–atmosphere exchanges and impacts on ecosystem, yet its large-scale precursors, circulation context, and sub-seasonal predictability remain poorly understood. Here, we identified onset-stage circulation regimes across northern extratropical latitudes (NEL; >30°N) using 200 and 1000 hPa geopotential height, and examined their regional expressions over eastern Asia, western North America, and Europe. Flash flourishing onset in East Asian was associated with a baroclinic circulation regime and was preceded by a North Atlantic sea surface temperature (SST) precursor at a four-pentad lead. In contrast, onset in western North American and European preferentially occurred under barotropic regimes, preconditioned by Great Plains soil moisture at three-pentad lead and North Atlantic SST at a four-pentad lead, respectively. Ridge regression forecasts revealed regime-dependent sub-seasonal predictability, with mean out-of-sample R 2 exceeding 0.3 up to lead times of two pentads in East Asia, three pentads in western North America, and four pentads in Europe. Together, these findings established a mechanistic and regionally specific framework for anticipating rapid vegetation greening at sub-seasonal timescales.

Kong, Xiangxu [Nanjing Univ. of Information Scienc

Prime Time for Model-Predictive Control? Assessing the Technical and Market Readiness of Advanced Controls in Buildings

Despite three decades of extensive research and field testing that have consistently validated the benefits of Model Predictive Control (MPC) in building applications, the technology has seen limited market adoption. This paper evaluates the readiness of MPC for widespread deployment, showcases recent demonstrations and field tests across diverse building types, including residential, small commercial, large commercial, and campus settings. Our results demonstrate that MPC can optimize system operations to achieve load shifting, minimize curtailment of on-site generation, and reduce energy costs by up to 80 %, while maintaining or improving occupant comfort. We also show that MPC can effectively control large assets, such as MW-sized thermal storage systems, and respond to dynamic pricing signals. However, achieving scale remains difficult due to labor-intensive workflows, reliance on a “PhD-in-the-loop” for MPC design and maintenance, susceptibility to fragile data infrastructure, and persistent workforce education and acceptance barriers. To bridge this gap, we outline a transition from bespoke, labor intensive prototypes toward streamlined, segment-targeted deployment strategies that leverage model templates, semantic tools, and generative AI. By automating control configuration and reducing engineering effort, these recommendations provide a pathway for transforming successful research demonstrations into scalable, market ready solutions for MPC-based controls.

Pritoni, Marco

Improving Adhesive Bondline Time of Flight Predictions During Autoclave Cure Utilizing Machine Learning

Composite materials are increasingly being used in aerospace applications due to their superior strength-to-weight ratio compared to commonly used metals. A current limitation to widespread adoption is the certification of adhesively bonded joints. One approach to improving adhesive bonding in composites is accurately measuring the thickness of adhesive bondlines in composite laminates. Precise bondline thickness control is essential for aerospace applications where adhesive layer thickness directly affects joint fracture properties and structural performance. This study focused on implementing machine learning techniques to determine the ultrasonic time of flight (directly correlated to thickness) in adhesive bondlines throughout autoclave cure cycles. A high-temperature (use up to 180°C) ultrasonic scanning system was deployed in an autoclave to provide time of flight data through composite panels. Three experiments were conducted on the curing of 305 mm × 305 mm unidirectional composite panels. In the first experiment, a piecewise function was fit for the temperature correction factor to account for changing autoclave temperatures. Due to deficiencies in the first calibration experiment, a second experiment was run, and the results were used to train a machine learning model. The revised experiment, in combination with the machine learning model, significantly increased the accuracy of the bondline time of flight predictions (~14% error reduced to <1%). Data was processed using the Regression Learner Application in MATLAB®, with a Support Vector Machine selected for the model. The result was a machine learning algorithm capable of reliably quantifying ultrasonic time of flight through adhesive bondlines. The third experiment provided independent test data for the machine learning model, demonstrating that the model produces accurate predictions from data beyond its training set.

Machine Learning

Microstructure prediction for Ti-22Al-25Nb in laser powder bed fusion

This work presents a physics-informed framework for predicting solidification morphology and defect susceptibility in additively manufactured Ti–22Al–25Nb across a broad processing space. The framework integrates solidification microstructure selection (SMS) analysis with a single-track defect-based printability map to establish a unified methodology linking processing parameters to both interfacial morphology and manufacturability. Thermal gradients G and solidification rates R are first computed using the Thermo-Calc Additive Manufacturing (TC-AM) module, a finite-interface-dissipation (FID) phase-field (PF) model coupled with CALPHAD method is then employed to systematically distinguish planar and dendritic regimes as functions of $G$ and $R$. By superimposing the printability map onto the morphology projections, a comprehensive process–structure framework is obtained. Across most processing conditions, the predicted microstructure is predominantly dendritic, while planar growth emerges only under selected laser power $P$ and scan speed $v$ combinations. In addition to morphology classification, the framework quantifies the dendritic area fraction and introduces a width-based morphology descriptor to characterize the spatial extent of planar/dendritic regions within the melt pool. It provides mechanistic insight into the interplay between solidification physics and defect formation, offering practical guidance for parameter selection and microstructural control in Ti–22Al–25Nb additive manufacturing (AM).

36 MATERIALS SCIENCE

Spacecraft Fire Safety Predictions using Verified Saffire Model

A model developed using Fire Dynamics Simulator (FDS) that aimed to determine the effect of a fire in a spacecraft was validated by data collected during the Saffire campaign. The model used inlet and outlet temperatures and CO 2 concentrations of the Saffire payload where fire spread was taking place to determine the amount of heat and combustion products that made it into Northrop Grumman’s Cygnus vehicle. The model was then validated using six remote sensors in various places in the vehicle, as well as a far field device (FFD) in the open zenith section that was representative of average vehicle values. The current work focuses on using the model to predict fire safety scenarios. One simulation aimed to determine the fate of HCl, which sticks to surfaces. The model prediction showed that the HCl was removed from the atmosphere rapidly. This compared well against the FFD data in the Saffire VI campaign event where a bottle of 5% HCl was released into the vehicle. An additional simulation where the Environmental Control and Life Support System (ECLSS) was shut off once the FFD reached 5 ppm of HCl showed that HCl stayed in the atmosphere considerably longer. Continuing the simulation with the ECLSS activated and after temperatures returned to their initial conditions, resulted in a rapid removal of HCL similar to that observed in the original HCl release scenario model. Finally, a simulation that used the heat release rate from a lithium-ion battery test to determine the effect it would have on a spacecraft was performed. This simulation used a heat addition rate that is considerably higher than what was determined from the burning of solid fuels in the Saffire campaign and hence produced a non-trivial temperature increase in more locations within the vehicle.

Fire Safety

Nuclear–Electronic Orbital General Rate Theory: Predicting Hydrogen Kinetic Isotope Effects in the Deep Tunneling Regime

Hydrogen transfer is a critical component of many chemical and biological processes. The ratio of rate constants for hydrogen and deuterium transfer defines the H/D kinetic isotope effect (KIE), which is a powerful tool for elucidating hydrogen transfer mechanisms. Interpretation of experimental H/D KIEs relies on accurate and affordable computational methods. However, due to their light mass, hydrogen and deuterium can undergo tunneling, which is challenging to describe in multidimensional molecular systems. Herein, we introduce the nuclear–electronic orbital general rate theory (NEO-GRT), which enables the efficient prediction of H/D KIEs based on full-dimensional molecular quantum chemistry calculations. The NEO-GRT approach describes the hydrogen transfer rate constant with a general expression that spans the vibrationally adiabatic and nonadiabatic hydrogen tunneling regimes. The input quantities are computed using NEO density functional theory, which treats the transferring hydrogen or deuterium nucleus quantum mechanically on the same level as the electrons. We investigate two intramolecular proton transfer reactions in organic molecules at temperatures down to 50 K to evaluate the performance of NEO-GRT by comparison to transition state theory and ring-polymer instanton theory. The KIEs computed with NEO-GRT agree with those calculated using ring-polymer instanton theory for the full-dimensional molecular systems at the same level of electronic structure theory. This agreement indicates that NEO-GRT captures the deep hydrogen tunneling effects, in contrast to transition state theory, which neglects such effects. Given its relatively low computational cost, NEO-GRT is a promising approach for predicting H/D KIEs in large organic and organometallic systems.

Hydrogen

AutoBEM: A scalable framework for nationwide building energy simulation and retrofit evaluation in the United States

This paper presents AutoBEM, an integrated, automated framework for nationwide building energy modeling and retrofit evaluation in the United States. Unlike prior UBEM platforms that either rely primarily on representative stock sampling or operate at city scale, AutoBEM automates the generation of building-resolved, physics-based EnergyPlus/OpenStudio simulation models at national scale using GIS-derived geometry, prototype-based assumptions, and standardized scalable workflows. Leveraging the Model America dataset and high-performance computing, AutoBEM generates and simulates energy models for 122.9 million buildings, representing 97.8% of the U.S. building stock. These models are being made publicly and freely available as the Model America v1.0 (MAv1) dataset. AutoBEM supports detailed, building-level assessments of energy consumption, CO2 emissions, and post-processed anthropogenic heat emissions (AHE), and evaluates 151 energy conservation measures (ECMs) using localized utility pricing and building characteristics. In addition, AutoBEM incorporates both typical and future climate conditions through integration with Typical Meteorological Year (TMY) and Future TMY (fTMY) weather data derived from IPCC scenarios. In a case study of Phoenix, Arizona, AutoBEM identified several high-efficiency HVAC upgrades and selected envelope measures with short modeled payback periods (1.5 years) for certain building types and standards. Simulations under future climate scenarios (SSP5–RCP8.5) project an 11.3% increase in electricity use and a 32% reduction in natural gas demand by 2100, underscoring the need for climate-adaptive retrofit planning. By enabling reproducible, bottom-up, and location-specific analysis at scale, AutoBEM provides a step toward a national digital twin of the built environment and supports data-driven screening and planning for decarbonization, resilience, and energy equity.

Li, Hang [ORNL] (ORCID:0000000306001920)

Multitask graph neural networks for elastoplastic response prediction in dual-phase polycrystals

Microstructure-sensitive prediction of elastoplastic response remains a recurring bottleneck in multiscale damage and fatigue modeling, where large ensembles of statistically distinct polycrystals are required to quantify variability and extreme-value behavior. In this work, we develop a multitask graph neural network (GNN) surrogate that maps dual-phase ferrite–martensite polycrystal microstructures to Statistical Volume Element (SVE)-level elastoplastic Quantities of Interest (QoIs). Each SVE is represented as a grain-adjacency graph, with node features encoding phase, geometry, and crystallographic orientation, and edge features encoding relative misorientation. A message-passing graph convolution generates node embeddings, which are pooled into a graph representation and passed to a multitask regression head that jointly predicts 10 scalar QoIs and vector-valued stress–strain responses in orthogonal loading directions across multiple martensite volume fractions and SVE sizes. Results show high accuracy for scalar QoIs and strong agreement for full stress–strain trajectories, with population envelopes reproducing both median behavior and finite-SVE variability across compositions and partition scales. A unified model trained on pooled volume-fraction data preserves most within-regime accuracy relative to regime-specific models while also capturing the broader cross-regime variation reflected in the pooled test set. Distributional comparisons further demonstrate that the surrogate preserves heterogeneity under SVE partitioning, enabling statistically consistent block-wise random-field construction for mesoscale analyses. Overall, the proposed grain-graph surrogate provides a practical pathway to accelerate ensemble-based studies of SVE-level constitutive variability in dual-phase polycrystals.

Crystal plasticity

Impact of Crystalline Phases on Low-Activity Waste Glass Durability: Insights from PCT and VHT

During vitrification of nuclear wastes, slow cooling along the container centerline promotes crystalline phase formation, which can alter residual glass composition and reduce chemical durability. This study investigates the effects of crystalline phases on the chemical durability of low-activity waste (LAW) borosilicate glasses using the product consistency test (PCT) and vapor hydration test (VHT) on container centerline cooled (CCC) samples. A preliminary model (R2 = 0.88) was developed to predict CCC PCT responses based on glass composition, PCT data from quenched glasses, and measured crystal fractions. Using the latest LAW glass dataset, the feasibility of predictive modeling is evaluated, limitations in current data and methods are identified, and challenges for improving model accuracy are discussed to guide future data collection and model development.

borosilicate glass

Machine Learning for Predicting Team Functioning in HERA Missions

Team functioning is integral to success in future long term space exploration missions. Proactively detecting declines in team functioning can mitigate conflict and ensure mission success. This project developed a speech-based artificial intelligence (AI) system that unobtrusively predicts degradation in team functioning, including performance and cohesion, in the Human Exploration Research Analog (HERA) Campaigns 4 and 5. The AI system conducted automated analysis of the prosodic (tone of voice) and linguistic (language content) components of speech, modeling interpersonal dynamics at both the turn-taking and day-wide levels. We investigated team functioning via observing structured interactions (i.e., multi-mission space exploration vehicle-extra vehicular activity [MMSEV-EVA], team interaction battery [TIB]) and unstructured interactions before the MMSEV-EVA task. We developed machine learning models to predict team functioning (objective task accuracy, self reported team efficacy and self reported team cohesion) by analyzing OpenSmile acoustic features, linguistic descriptors extracted via the linguistic inquiry and word count (LIWC) dictionary, and semantic embeddings. In the TIB, static models using logistic regression and random forests were not able to predict task accuracy, but predicted team efficacy and cohesion during both the decision making and relational tasks to a moderate level (60-70%). Majority voting on the individual turns to predict day long team efficacy further increased accuracies (70-80%). Finally, long short-term memory (LSTM) models showed the best performance across all variables (80-91%), including task performance. In the MMSEV-EVA, static models achieved an accuracy of 60% with majority voting, which increased to 80% through the incorporation of mission day as a variable, accounting for the learning effect. A key finding across both tasks was the "team-dependent" nature of these interactions; models achieved much higher accuracy when trained on prior days of the same team's data rather than attempting to generalize across entirely different teams, with even 1-2 days of prior data per team achieving 5-15% improvement over team-independent models. In addition, the incorporation of pre-task data from the same team also improves model performance, e.g., incorporating data from the decision-making task of the TIB, which preceded the relational task, improved the prediction of team efficacy and cohesion during the latter. We compared model performance when trained on machine-generated data compared to data that had been further corrected by human annotators. Overall, models trained on human-corrected data exhibited a modest improvement in performance, particularly when acoustic features were used. We found no significant correlation between word error rate (WER) and model accuracy (r(55) = -0.08, p = 0.51), but model’s accuracy was significantly higher for medium/high quality transcription (0.74 (SD = 0.48)) compared to the low-quality group (0.64 (SD = 0.36)) (t(63)=2.82, p = 0.006). Based on these, several design recommendation emerge, that could inform Standards at NASA. Models predicting team functioning should incorporate at least one to two days of historical interaction data, include brief pre-task discussions, and explicitly model temporal learning effects, especially for longer operational tasks. Minimum quality standards for automated speech-processing pipelines are needed, given the performance gains observed with manually corrected acoustic data. Finally, systems should leverage both acoustic features and language embeddings in complementary ways, with modality choices and fusion strategies tailored to mission context, task demands, and data quality requirements.

Shrivatsa Mishra

Thermally unstable roosts influence winter torpor patterns in a threatened bat species

Abstract Many hibernating bats in thermally stable, subterranean roosts have experienced precipitous declines from white-nose syndrome (WNS). However, some WNS-affected species also use thermally unstable roosts during winter that may impact their torpor patterns and WNS susceptibility. From November to March 2017–19, we used temperature-sensitive transmitters to document winter torpor patterns of tricolored bats (Perimyotis subflavus) using thermally unstable roosts in the upper Coastal Plain of South Carolina. Daily mean roost temperature was 12.9 ± 4.9°C SD in bridges and 11.0 ± 4.6°C in accessible cavities with daily fluctuations of 4.8 ± 2°C in bridges and 4.0 ± 1.9°C in accessible cavities and maximum fluctuations of 13.8 and 10.5°C, respectively. Mean torpor bout duration was 2.7 ± 2.8 days and was negatively related to ambient temperature and positively related to precipitation. Bats maintained non-random arousal patterns focused near dusk and were active on 33.6% of tracked days. Fifty-one percent of arousals contained passive rewarming. Normothermic bout duration, general activity and activity away from the roost were positively related to ambient temperature, and activity away from the roost was negatively related to barometric pressure. Our results suggest ambient weather conditions influence winter torpor patterns of tricolored bats using thermally unstable roosts. Short torpor bout durations and potential nighttime foraging during winter by tricolored bats in thermally unstable roosts contrasts with behaviors of tricolored bats in thermally stable roosts. Therefore, tricolored bat using thermally unstable roosts may be less susceptible to WNS. More broadly, these results highlight the importance of understanding the effect of roost thermal stability on winter torpor patterns and the physiological flexibility of broadly distributed hibernating species.

Biodiversity & Conservation

Circumventing data imbalance in magnetic ground state data for magnetic moment predictions

Abstract Magnetic materials play a crucial role in the transition to more sustainable forms of energy and electric vehicles. There is an anticipated shortage in magnetic materials in the future, and as a result there is an urgent need to discover and design new magnetic materials. Computational magnetic material design using density functional theory is daunting because of the challenge in identifying magnetic ground states from a combinatorially large set of possibilities. Machine learning offers a path forward by enabling efficient surrogate models that can more readily enumerate these states, but there is a dearth of training data available, and what is available tends to be imbalanced with too much non-magnetic data. In this work we show that the discrete and previously tackled data imbalance that exists at the level of the magnetic ordering leads to an imbalanced continuous distribution with many zeros when the data is unraveled at the atomic magnetic moment level, which subsequently leads to models with low accuracy for magnetic properties. We mitigate this by using a two-part model framework. Our scheme is able to classify atoms into magnetic and non-magnetic with an F1 score and Matthew’s correlation coefficient (MCC) of ~91% and then to provide an implicit embedding representation that maps directly onto the magnitude of the magnetic moment with a mean absolute error of 0.1 μ B . Beyond screening for new magnetic materials, we demonstrate an additional practical use case of our scheme: the provision of good initial guesses for magnetic moments in first-principles electronic relaxations. Such initialization is shown to lead to faster convergence to configurations that lie closer to the ground state.

Computer Science