Search papers, labs, and topics across Lattice.
This paper introduces ARIA, a causal-aware framework designed to enhance the reasoning capabilities of Large Language Models (LLMs) in materials discovery by addressing the issue of contextual tunneling, where models overly rely on narrow evidence. By employing a three-tier cascade approach that includes direct causal reasoning, physics-informed analogical transfer, and explicit parametric fallback, ARIA effectively improves the accuracy of forward prediction and inverse design tasks for two-dimensional materials. The framework not only mitigates the limitations of traditional LLMs but also generates auditable causal traces, promoting trustworthy AI applications in materials science.
Contextual tunneling in LLMs can be overcome, leading to more reliable and physically grounded materials discovery through the innovative ARIA framework.
Generative models have revolutionized the process of materials discovery, yet they often fail to satisfy underlying physical causality. Through an analysis of Large Language Models (LLMs) augmented with knowledge graphs derived from current literature, we uncover a phenomenon termed contextual tunneling, where models "over-anchor" on narrow, retrieved evidence while suppressing global physical reasoning. To address this problem, we introduce ARIA, a causal-aware framework that conditions knowledge use on mechanistic completeness. ARIA routes each query through a three-tier cascade: (i) direct causal reasoning when complete evidence chains of Process-Structure-Property (PSP) are available, (ii) physics-informed analogical transfer for sparse or novel material systems, and (iii) explicit parametric fallback when external evidence is incomplete. As a proof of concept, we construct a Knowledge Graph (KG) containing 2,839 extracted PSP relations from peer-reviewed articles in the materials literature and evaluate ARIA on forward prediction and inverse design tasks for two-dimensional (2D) materials. ARIA mitigates contextual tunneling, improves over unaugmented and naive KG-augmented baselines, and provides further gains when an online literature search is used for evidence enrichment. Crucially, ARIA produces auditable causal traces, enabling physically grounded and trustworthy AI-assisted materials discovery.
