Aligning LLMs for Crystal Material Design: Test-Time Reinforcement Learning and Model Merging

Pengyu Wang, Yang Han, Xin Chen, Kai Yu, Tong Zhu, Lu Chen, Bo Chen


Abstract
Conditional crystal generation requires Large Language Models (LLMs) to produce Crystal Strings with discrete and continuous variables while satisfying composition, symmetry, and property constraints. Desired regions are often sparse in supervised data, while retraining for each objective preference is costly. We present a verifier-grounded framework combining label-free Test-Time Reinforcement Learning (TTRL) and Model Merging (MM). TTRL performs one batch-level, partially transductive adaptation per property using composition, symmetry, and learned band-gap rewards, without target reference structures. Property-specific experts are interpolated to navigate bounded symmetry–band-gap trade-offs without retraining intermediate policies. On MP-20, TTRL improves symmetry control, band-gap accuracy, and compositional fidelity over High-Precision SFT and Standard RL, and is competitive with diffusion and preference-aligned baselines. In a held-out high-band-gap region, TTRL generalizes to test prompts unseen during adaptation, reducing the mean squared error between surrogate-predicted and target band gaps by 52% relative to standard reinforcement learning under best-of-ten sampling. An independent band-gap predictor shows consistent gains, while density functional theory calculations further assess selected candidates. Similar gains with Qwen3.5-9B provide cross-backbone evidence. Model Merging recovers competitive two-objective trade-offs from two endpoint experts without additional RL, enabling efficient adaptation and preference control for structured scientific generation.
Anthology ID:
2026.findings-emnlp.2238
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
47898–47919
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.2238/
DOI:
Bibkey:
Cite (ACL):
Pengyu Wang, Yang Han, Xin Chen, Kai Yu, Tong Zhu, Lu Chen, and Bo Chen. 2026. Aligning LLMs for Crystal Material Design: Test-Time Reinforcement Learning and Model Merging. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 47898–47919, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Aligning LLMs for Crystal Material Design: Test-Time Reinforcement Learning and Model Merging (Wang et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.2238.pdf