用大模型动态检索+规则推理,提升难预测钙钛矿空间群的准确率。
Predicting Space Groups of Double Perovskites by LLM with Dynamic Few-Shot Learning
- 基于动态少样本检索和规则引导推理,平衡主流与罕见空间群预测
- 在罕见空间群上准确率提升3.26个百分点,整体性能优于现有方法
- 适合材料科学中数据不均衡场景下的结构预测任务
双钙钛矿(DPs)具有广泛的组分可调性,但稳定结构的空间群(SGs)预测仍具挑战,因可用数据集对主流SG类严重倾斜。本文提出基于大模型代理的框架DyRIS,从给定组分预测排名的SG候选。DyRIS采用增强多样性的动态少样本提示机制,减少高频SG的主导影响;结合基于B/B'阳离子有序、量化指标及主流SG偏差控制的规则推理,优化并排序前3名候选。在3,528个热力学筛选的DP条目上评估,训练数据比为0.5时,DyRIS在整体准确率上表现优异,并取得最佳总体Top-1宏平均F1分数,且在所有少数类SG指标上领先。相比CrabNet,DyRIS在少数类Top-1准确率提升3.26个百分点;在少数类Top-3准确率上超越最强的PyCaret基线。消融实验表明,多样性检索、量化指标、主流SG偏差控制及B/B'有序信息均贡献显著。额外实验显示,最终规则推理步骤难以被传统分类器或排序模型替代。结果表明,结合检索式大模型推理与晶体学领域知识,在不平衡材料数据集中具有预测空间群的潜力。
原文摘要 · Abstract (English)
Double perovskites (DPs) offer broad compositional tunability, but predicting the space groups (SGs) of stable structures remains difficult because available datasets are often strongly imbalanced toward dominant SG classes. We refer to dominant SG classes as major SGs and underrepresented classes as minor SGs. We introduce Dynamic and Diversity-enhanced Few-shot Retrieval and Rule-Guided Inference for Space-Group Prediction (DyRIS), an LLM-agent-based framework that predicts ranked SG candidates from a given DP composition. DyRIS uses diversity-enhanced dynamic few-shot prompting to retrieve relevant in-context examples while limiting the dominance of frequently represented SGs. It further incorporates rule-guided inference based on B/B' cation ordering, quantitative indicators, and major-SG bias control to refine and rank the final Top-3 SG candidates. We evaluate DyRIS on 3,528 thermodynamically filtered DP entries and compare it with composition-based and descriptor-based baselines. At a training-data ratio of 0.5, DyRIS achieves competitive overall accuracy while obtaining the best Overall Top-1 macro-F1 score and the best performance across all Minor-SG metrics. DyRIS improves Minor-SG Top-1 accuracy by 3.26 percentage points relative to CrabNet and achieves higher Minor-SG Top-3 accuracy than the strongest PyCaret-based baseline. Ablation studies show that diversity-enhanced retrieval, quantitative indicators, major-SG bias control, and B/B' ordering information each contribute to prediction performance. Additional experiments show that the final rule-guided inference step is not easily replaced by conventional classifier- or ranker-based models. These findings demonstrate the potential of combining retrieval-based LLM reasoning with crystallographic domain knowledge for SG prediction in imbalanced materials datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。