---
title: Atomistic Language Models (ALMs) Overview
url: https://www.emergentmind.com/topics/atomistic-language-models-alms
type: topic
---

# Atomistic Language Models (ALMs) Overview

Searching arXiv for recent papers on Atomistic Language Models and closely related benchmarks/tokenization methods.
Atomistic Language Models (ALMs) are language-model-based systems that operate directly on atomistic structure representations rather than solely on textual surrogates. In the materials setting, ALMs unify a pretrained atomistic encoder, a causal large-language backbone, and a denoising diffusion decoder through purely continuous projectors so that a single model can understand atomistic structures, generate materials from natural language, and optimize crystal structures as instructed by text [2606.21395]. Closely related work extends the same atom-level modeling program to biomolecules and molecular graphs, including atom-by-atom protein generation from SELFIES sequences and atom-level tokenization of local molecular environments for molecular LLMs [2308.09482], [2512.03080].

## 1. Conceptual scope and motivation

The immediate motivation for ALMs in materials science is the mismatch between the centrality of three-dimensional atomic structure and the limitations of conventional language-model pipelines. In the formulation introduced for crystalline materials, atomistic structure and natural language had long been modeled separately, with language models either calling atomistic models as tools or being fine-tuned on lossy textual encodings that discard atomistic information [2606.21395]. ALMs are designed to replace that separation with native multimodality.

This problem setting is sharpened by the representational demands of crystallography. Crystallographic Information Files (CIFs) are standard structure representations, but successful manipulation of CIFs requires more than syntax recovery: it requires geometric operations on lattices and atomic positions, consistency of chemical formulae, and tolerance-aware structural equivalence. AtomWorld formalizes this challenge as the “motor skills” problem for LLMs, defining benchmark tasks for reading, writing, and editing three-dimensional atomic structures encoded in CIFs [2510.04704]. The benchmark’s framing suggests that ALMs are not merely larger text models for materials science; they are intended to couple textual competence with spatial and atomistic reasoning.

The term also has a broader domain scope. In biomolecular work, an ALM can denote a model that generates proteins atom by atom from a chemical language representation, including noncanonical residues and protein–drug conjugates, rather than restricting generation to amino-acid vocabularies [2308.09482]. A plausible implication is that “ALM” is best understood as a family of multimodal or atom-level language models whose defining property is direct access to atomic structure, whether the substrate is a crystal, a small molecule, or a protein-like macromolecule.

## 2. Representations and cross-modal interfaces

The materials ALM architecture is defined by three components and two continuous interfaces. A frozen pretrained atomistic encoder $\mathcal{E}$ maps a crystal’s 3D coordinates, lattice, and elements into per-atom embeddings; a causal large-language backbone $\phi$ processes text and “soft-token” atoms; and a denoising diffusion decoder $\mathcal{D}$ reconstructs atomic coordinates and lattice from continuous latent steering signals [2606.21395]. The specific backbone reported is Qwen3-8B, and the diffusion decoder is MatterGen. The input-side projector $P_{\mathrm{in}}$ lifts encoder outputs into the LLM embedding space, while the output-side bridge uses a producer implemented as a Q-Former and a consumer implemented as IP-Adapter cross-attention injected into each GemNet-T block of the diffusion network. The architecture is explicitly described as using purely continuous projectors, with no quantization.

This continuous bridge is important because crystallographic manipulation is fundamentally geometric. For CIF-based tasks, fractional coordinates $(u_i,v_i,w_i)$ are converted to Cartesian positions by
$$
\mathbf{r}_i = u_i\,\mathbf{a} + v_i\,\mathbf{b} + w_i\,\mathbf{c},
$$
and interatomic distances are computed as Euclidean norms in Cartesian space [2510.04704]. Structural editing therefore requires models to invert the fractional–Cartesian mapping when prompts specify Cartesian displacements or distances. AtomWorld’s evaluation further operationalizes structural fidelity through maximum displacement after Hungarian matching and RMSD, with max\_dist reported as particularly sensitive because many editing tasks move only a single atom [2510.04704].

A distinct but related representational strategy appears in AtomDisc, which quantizes local atomic environments into structure-aware discrete tokens embedded directly in an LLM’s token space [2512.03080]. There, a molecule is represented as a graph, a pre-trained GIN encoder maps each atom and its two-hop neighborhood into a $300$-dimensional embedding, and a learned codebook of size $512$ assigns a symbolic token $\langle\!\text{atom\_}k\rangle$ by nearest-neighbor search. This contrast is structurally important: the materials ALM emphasizes continuous bridging from atomistic embeddings into diffusion steering, whereas AtomDisc emphasizes discrete atom-level vocabulary extension. This suggests two major design regimes for atomistic language modeling: continuous latent interfacing and discrete structure-aware tokenization.

## 3. Training procedures and controllable generation

The materials ALM is trained in three stages. Stage 1 performs soft-token alignment by training only $P_{\mathrm{in}}$ to describe structures in text under causal language-model cross-entropy. The reported data comprise approximately $1.35$ million structure-to-description pairs from Robocrystallographer and GPT-Narratives, optimized with AdamW using learning rate $1\mathrm{e}{-3}\to$ cosine decay, batch size $32$, and $12$k steps [2606.21395]. The result is that $P_{\mathrm{in}}$ learns to map per-atom embeddings into LLM token space without quantization.

Stage 2, termed ALM Core, adds multimodal instruction tuning. A LoRA adapter is attached to $\phi$, the atomistic encoder remains frozen, and only LoRA plus $P_{\mathrm{in}}$ are trained on a five-bucket mixture: describe, property\_apps, arXiv abstract generation, CAMEL Q\&A, and MaScQA multiple choice [2606.21395]. Hyperparameters include LoRA rank $128$, $\alpha=256$, learning rates $2\mathrm{e}{-4}$ for LoRA and $2\mathrm{e}{-5}$ for $P_{\mathrm{in}}$, batch size $256$, and $12$k steps. This stage is the point at which the model becomes a genuinely mixed-domain language model rather than a narrowly aligned structural describer.

Stage 3 specializes the backbone for editing and generation. ALM Edit uses strong conditioning, full $\phi$ finetuning, and MatterGen CSP mode, while ALM Gen uses weak conditioning, LoRA rank $8$ on $\phi$, and MatterGen Base [2606.21395]. Both variants optimize a shared diffusion objective with classifier-free guidance dropout $p=0.2$, and both add auxiliary losses on the producer output: composition-count BCE, contrastive decorrelation, and directional cross-entropy on raise/lower prompts for ALM Edit. The reported diffusion hyperparameters include $T=1000$, guidance scale $g=0.5$ for Edit and $g=1.0$ for Gen, producer token count $M=16$, context window $N=128$, and atom token count $K=8$.

Inference-time control is provided by Text-to-Crystal Feynman–Kac (T2C-FK), which replaces a single diffusion chain with an $N$-particle bootstrap SMC that reweights partial trajectories by a reward on the Tweedie $\hat{x}_0$ estimate [2606.21395]. The sampler can enforce exact stoichiometry or custom property objectives. Its stoichiometry reward combines Hungarian negative log-likelihood, count $L_1$ gap, and ratio JS-divergence, and at $t=0$ atomic numbers are overridden by Hungarian assignment for exact element counts. In the limit $N\to\infty$, the procedure implements a Feynman–Kac posterior of the form $p^*(x_0)\propto p(x_0)\exp(r(x_0)/\tau)$. A plausible implication is that ALMs inherit controllability not only from prompt conditioning but also from explicit inference-time posterior shaping.

## 4. Benchmarks, metrics, and empirical behavior

ALM Bench is introduced as the first benchmark for text-conditioned crystal generation and optimization, with a seven-task suite containing $7\,000+$ prompt–crystal pairs [2606.21395]. Its tasks include directional editing, crystal structure prediction, application consistency, polymorph generation, doping/substitution, strain, and text-to-structure recovery. Metrics include direction-correct rate using MatterSim-relaxed property changes, Match@1 and Match@K with RMSE@1/20 via pymatgen StructureMatcher, SUN and MSUN for stable/unique/novel yields, de novo metrics $U$, $V_{\mathrm{struct}}$, and $V_{\mathrm{chem}}$, and judge scores for application consistency on a $0$–$2$ scale.

The reported performance highlights are substantial. ALM Core breaks the “GNN–LLM wall” on property prediction, with LLM4Mat-Bench MAD/MAE $\ge 5$ on $4/9$ slices and MAE $5$–$100\times$ better than prior text LLMs [2606.21395]. ALM Edit sets new state of the art on MP-20 and MPTS-52 crystal structure prediction, achieving best RMSE, including $0.021$ Å on MP-20, and Match@20 of $83.2\%$. ALM Edit also beats GPT-4o, 4.1, and 5.2 baselines on ALM Bench directional editing and application tasks. ALM Gen achieves top SUN yields on MP-20 at $7.8\%$ and competitive MSUN on LeMat-GenBench at $35.2\%$, improving upon unsteered MatterGen by more than $2\times$.

AtomWorld complements ALM Bench by exposing the failure modes of current frontier models on crystallographic operations [2510.04704]. Across ten AtomWorld actions and five CIF-perception subtasks, easy actions such as change, remove, and add achieve approximately $85$–$100\%$ success with negligible max\_dist, whereas move, move\_towards, and insert\_between fall to roughly $50$–$80\%$ success with mean max\_dist of $0.1$–$0.5$ Å. Hard actions including swap, delete\_below, and rotate\_around fall below $30\%$ success, and rotate\_around often exceeds $5$ Å max\_dist. CIF-Repair places closed models above $90\%$ success and open-source models around $50$–$80\%$; CIF-Gen reaches approximately $90$–$100\%$ for standard prototypes such as NaCl and CaF$_2$, but drops to approximately $60\%$ for non-standard stoichiometries in the same prototype. For the Chemical Competence Score, effect sizes are reported as approximately $0.77$ for Qwen3-4B, $1.14$ for Qwen3-32B, and $0.99$ for Llama3-70B. StructProp remains difficult: Gemini 2.5 Pro reaches only approximately $70\%$ success on $10$ manually labeled cases, and most models remain below $50\%$ [2510.04704].

The dominant failure modes are numeric imprecision in coordinate arithmetic, mis-indexing, syntax lapses, context-window limitations for long supercell expansions, and reliance on memorized CIF snippets rather than true geometric understanding [2510.04704]. These observations delimit the present boundary of ALM capability: language-conditioned materials generation is advancing rapidly, but fine-grained crystallographic manipulation remains error-prone.

## 5. Molecular and biomolecular variants

In biomolecular modeling, the ALM of “Atom-by-atom protein generation and beyond with language models” treats every biomolecule as a single sequence of SELFIES tokens, with a vocabulary on the order of thirty tokens covering atoms, bonds, rings, branching, and four special tokens [2308.09482]. SELFIES is used because it captures complete chemical graph information, including atom identity, bond order, ring closures, branching, aromatics, and stereochemistry; no additional learned graph embeddings are required. The model is a GPT-style decoder-only Transformer trained with standard autoregressive cross-entropy, typically using roughly four attention heads, embedding dimension approximately $256$–$512$, feed-forward inner dimension approximately $4\cdot d_{\text{model}}$, and on the order of $1$–$10$ million total parameters. Sequences are up to $1\,664$ tokens, and random SMILES/SELFIES atom-order permutations expand approximately $10\,000$–$50\,000$ raw structures into approximately $250\,000$ training examples for each task.

That model demonstrates that atom-level sequence modeling can recover multiple hierarchical layers of protein organization from atom graphs alone [2308.09482]. When $1\,000$ novel proteins are sampled, approximately $68\%$ parse successfully into continuous N$\to$C backbones with recognizable sidechains and unique primary sequences, AlphaFold2 predictions return per-residue pLDDT mostly in the $70$–$90$ range, and the generated samples exhibit helices, strands, and turns in proportions matching the PDB training set. The same framework is extended beyond canonical amino acids by training on augmented proteins with random small-fragment attachments at every sidechain, after which generated proteins match training distributions in molecular weight, LogP, topological polar surface area, and fragment counts, while also producing novel sidechain chemistries. In the antibody–drug conjugate setting, training on approximately $250\,000$ combined protein–linker–warhead sequences yields about $90\%$ of samples passing backbone-plus-sidechain parsing, with pLDDT $\ge 80$, while generated warheads match LogP, QED, and SA distributions of ZINC.

AtomDisc occupies a neighboring but distinct position in the molecular ALM landscape [2512.03080]. Its core claim is that quantizing atom-level local environments into structure-aware tokens injects an interpretable inductive bias into molecular LLMs. The codebook has size $512$; embeddings are initialized by $k$-means on a random subset of $500$k atom embeddings; after $10$ epochs on $7$ million atoms from PubChem, nearly all $512$ codes are actively used. The tokens are projected into the $4096$-dimensional embedding space of LLaMA-2-7B by a two-layer MLP plus RMSNorm, and training proceeds through codebook training, projector alignment, multi-task instruction pretraining, and supervised fine-tuning. Reported downstream results include an average ROC-AUC of $84.7$ on seven MoleculeNet classification datasets, QM9 frontier-orbital MAE of $0.0035$ averaged over HOMO, LUMO, and gap, and substantial gains over UniMoT on forward prediction and retrosynthesis generation tasks. Because each atom token corresponds to a tight cluster of local environments, AtomDisc also exposes structure–property associations: tokens $319$ and $338$ distinguish different hydroxyl environments, and swapping them in model input shifts predicted aqueous solubility by approximately $0.3$ pK units, with Wasserstein distance $0.40$ and significance $p<10^{-10}$ [2512.03080]. This suggests that atom-level tokenization can serve not only as an input representation, but also as an attribution mechanism.

## 6. Limitations, ambiguities, and research directions

Current materials ALMs have explicit domain limits. Their encoder and decoder focus on inorganic, crystalline materials; amorphous, biomolecular, or defect-rich systems require new encoders and decoders [2606.21395]. Continuous projectors avoid codebooks, but may be less efficient at extreme scaling than VQ-VAE-like quantization for certain modalities. Strong scaling laws are anticipated, but balancing natural-language retention against strong cross-modal steering already requires careful LoRA-versus-full-finetune trade-offs. Inference-time stoichiometric and symmetry enforcement through T2C-FK adds approximately $8\times$ compute.

AtomWorld proposes several concrete routes toward more robust ALMs [2510.04704]. These include numeric-aware attention, multimodal chain-of-thought via Visualization-of-Thought with simple 3D sketches or coordinate tables inline, diffusion-based LLMs for valid-structure generation and repair, domain-focused pretraining with millions of CIF files, structure–property pairs, and geometric operation sequences, and task-specific fine-tuning on before-CIF/prompt/after-CIF triples with losses that penalize site mismatches. The same work also reports that a preliminary tool-augmented workflow using code-graph RAG and pymatgen APIs boosts remove success from $84\%$ to $100\%$ and insert\_between from $46\%$ to $83\%$, while rotate\_around remains difficult at approximately $18\%$ success. This suggests that near-term progress may depend as much on symbolic geometry and code execution as on larger language backbones.

A persistent misconception is that atomistic language modeling can be reduced to CIF string completion or retrieval of memorized crystal templates. AtomWorld’s nonstandard-stoichiometry failures and rotation/editing errors argue against that reduction, indicating that syntactic fluency and genuine geometric competence are separable [2510.04704]. Another source of confusion is terminological rather than technical: a separate line of work uses the language of “atoms” for internal sparse representation units in mechanistic interpretability, introducing the atomic inner product and reporting $99.9\%$ sparse reconstruction across layers on average and more than $99.8\%$ of atoms satisfying a uniqueness condition, compared with $0.5\%$ for neurons and $68.2\%$ for features [2509.20784]. That usage is conceptually distinct from chemical or materials ALMs.

Taken together, the literature defines ALMs less as a single architecture than as a research program. In materials, the defining features are native multimodality, continuous or structure-aware interfaces to atomistic representations, and evaluation on text-conditioned generation and optimization [2606.21395]. In molecules and proteins, the same program appears as atom-level tokenization, atom-by-atom generation, and direct discovery of structure–property associations [2308.09482], [2512.03080]. The central open question is therefore no longer whether language models can ingest atomistic information, but whether they can do so with the numerical precision, structural faithfulness, and controllability required for autonomous scientific workflows.

Source: https://www.emergentmind.com/topics/atomistic-language-models-alms