UNATE: UNsupervised ATomic Embedding for crystal structures property prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Solà-Garcia, Laura, Solé, Àlex, Ruiz-Hidalgo, Javier
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910255290515456
author Solà-Garcia, Laura
Solé, Àlex
Ruiz-Hidalgo, Javier
author_facet Solà-Garcia, Laura
Solé, Àlex
Ruiz-Hidalgo, Javier
contents Accurately predicting crystal properties is critical for accelerating materials discovery, but it is often limited by scarce labeled data and costly theoretical calculations. To alleviate this, we propose UNATE (Unsupervised Atomic Embedding), a framework that leverages structural information extracted from unlabeled crystal structures. UNATE integrates an unsupervised denoising autoencoder with self-supervised contrastive learning to learn robust atomic representations, which are then used as input features for downstream property prediction. Experimental results show that replacing raw atomic numbers with UNATE-pretrained node embeddings yields a 2.7\% improvement over the full-data baseline. Notably, the benefits become more pronounced in scenarios with limited labeled data, reaching improvements of up to 10\% when only 25\% of the labeled data is used.
format Preprint
id arxiv_https___arxiv_org_abs_2605_25866
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle UNATE: UNsupervised ATomic Embedding for crystal structures property prediction
Solà-Garcia, Laura
Solé, Àlex
Ruiz-Hidalgo, Javier
Machine Learning
Materials Science
Classical Physics
Accurately predicting crystal properties is critical for accelerating materials discovery, but it is often limited by scarce labeled data and costly theoretical calculations. To alleviate this, we propose UNATE (Unsupervised Atomic Embedding), a framework that leverages structural information extracted from unlabeled crystal structures. UNATE integrates an unsupervised denoising autoencoder with self-supervised contrastive learning to learn robust atomic representations, which are then used as input features for downstream property prediction. Experimental results show that replacing raw atomic numbers with UNATE-pretrained node embeddings yields a 2.7\% improvement over the full-data baseline. Notably, the benefits become more pronounced in scenarios with limited labeled data, reaching improvements of up to 10\% when only 25\% of the labeled data is used.
title UNATE: UNsupervised ATomic Embedding for crystal structures property prediction
topic Machine Learning
Materials Science
Classical Physics
url https://arxiv.org/abs/2605.25866