atlas

Research map / Large language models / Instruction and fine-tuning

Fine-Tuned Language Models Generate Stable Inorganic Materials as Text

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 35 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Fine-tunes LLaMA-2 models (7B/13B/70B) with LoRA on text-encoded crystal structures under multitask prompts, generating stable inorganic materials that outperform CDVAE on M3GNet/DFT stability.

Problem

Generating stable 3D inorganic crystal structures is hard because atoms mix discrete element identities with continuous coordinates and require symmetry handling; prior generative models are specialized diffusion or graph architectures.

Approach

Encode crystals as newline-separated text strings with LLaMA-2's digit-level tokenizer, combine with text prompts, and parameter-efficiently fine-tune a pretrained LLM (LoRA) with a multitask curriculum (generation, text-conditioning, infilling) and random translation augmentation, with no crystal-specific architecture.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune LLaMA-2 70B rather than use CDVAEUnconditional crystal generation · MP-20Metastable (M3GNet): 28.8% -> 49.8%Helped
Fine-tune LLaMA-2 70B rather than use CDVAEUnconditional crystal generation · MP-20Stable (DFT): 5.4% -> 10.6%Helped
Lower sampling temperature from τ=1.0 to τ=0.7 (ablation)LLaMA-2 70B · MP-20Metastable (M3GNet): 35.4% -> 49.8%Helped
Lower sampling temperature from τ=1.0 to τ=0.7 (ablation)LLaMA-2 13B · MP-20Metastable (M3GNet): 33.4% -> 38.0%Helped
Lower sampling temperature from τ=1.0 to τ=0.7 (ablation)LLaMA-2 7B · MP-20Metastable (M3GNet): 35.1% -> 35.0%No clear effect
Use fractional rather than absolute coordinates (ablation)LLaMA-2 7B · MP-20Structural validity: 90.8% -> 91.4%Helped
Use fractional rather than absolute coordinates (ablation)LLaMA-2 7B · MP-20Compositional validity: 80.5% -> 83.2%Helped
Add atom-order permutations (ablation)LLaMA-2 7B · MP-20Structural validity: 92.5% -> 89.2%Hurt
Add atom-order permutations (ablation)LLaMA-2 7B · MP-20Compositional validity: 82.9% -> 81.7%Hurt
Use the shared LLaMA-2 tokenizer instead of special crystal tokens (ablation)LLaMA-2 fine-tuning · Crystal training examplesTraining loss (Epoch 5): 0.588 -> 0.385Helped
Guide element swaps with the infilling language model rather than uniform samplingTemplate-based material mutation and relaxation · Materials sampled from the test setStable compoundsHelped
Increase model scaleLLaMA-2 trained on atomistic data · Translation symmetry evaluationIncrease in Perplexity under TransformationHelped

Surprising findings

Limitations

Experimental setup

Datasets: MP-20 Materials Project

Benchmarks: MP-20

Metrics: structural validity compositional validity coverage recall coverage precision wdist density wdist num elements metastable rate (M3GNet) stable rate (DFT) IPT diversity

Baselines: CDVAE LM-CH LM-AC

Builds on: LoRA CDVAE M3GNet LLaMA-2

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy