Auditing Domain Alignment in Latent Representations for Cross-Domain Materials Machine Learning

16 January 2026, Version 1
This content is an early or alternative research output and has not been peer-reviewed by Cambridge University Press at the time of posting.

Abstract

Machine learning models are increasingly applied to heterogeneous materials datasets spanning different synthesis routes, measurement protocols, and structural classes. Although multi-task and representation-learning approaches are commonly used to improve predictive performance, the latent representations learned by such models are rarely examined directly, allowing domain misalignment to remain hidden until model transfer fails. Here, we present a practical, representation-level framework for auditing domain alignment in latent spaces learned by multi-task neural networks for materials property prediction. Using porous carbon materials and metal–organic frameworks as a representative cross-domain case study, we analyze a shared latent space trained across multiple target properties and regularization strategies. Global and local dimensionality-reduction visualizations are combined with quantitative measures of domain separability and latent–target correlation structure to assess alignment quality. We find that domain-alignment objectives induce measurable geometric reorganization and partial large-scale mixing in latent space, while domain identity remains locally recoverable and physically interpretable monotonic associations between latent dimensions and target properties are preserved. These diagnostics reveal structural differences in learned representations that are not apparent from conventional performance metrics alone and enable model comparison based on latent-space geometry rather than predictive accuracy. The proposed auditing framework is model-agnostic and can be readily integrated into existing materials machine learning workflows, providing a practical tool for improving robustness and interpretability in cross-domain materials modeling.

Keywords

Materials informatics
Representation learning
Latent space analysis
Domain alignment
Unsupervised learning
Metal-organic frameworks
Porous carbon materials
Heterogeneous Materials Datasets

Supplementary materials

Title
Description
Actions
Title
Supplementary Information for "Auditing Domain Alignment in Latent Representations for Cross-Domain Materials Machine Learning"
Description
This file contains an extended description of the methodology, stability of the network encoding and UMAP & PCA images of Baseline and Best_Aligned Encoders.
Actions

Comments

Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting and Discussion Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.