用张量方法统一优化材料设计,兼具可解释性与高精度预测。
Tensor Methods: A Unified and Interpretable Approach for Material Design
- 采用张量补全技术构建统一模型,自动提取可解释的物理因子。
- 在非均匀采样数据下,相比传统机器学习提升5%整体预测准确率。
- 能复现已知物理规律,适合材料研发人员发现新设计模式。
材料设计需针对特定性能进行优化,但设计参数增多时搜索空间呈指数增长,难以全面合成与评估。传统计算方法如有限元分析(FEA)计算成本过高,而现有机器学习代理模型常缺乏可解释性,且在非均匀采样数据上表现不佳。本文提出使用张量补全方法作为统一方案,在保持预测能力的同时,自然获得可解释的张量因子。实验表明,该方法能通过因子复现物理现象,说明其预测符合物理规律,有助于实验者发现潜在新模式。在非均匀采样场景下,某些具备低秩约束的张量方法表现出更好泛化能力,最优模型相较基线机器学习方法在聚合R²上提升达5%,部分分布外区域误差减半。
原文摘要 · Abstract (English)
When designing new materials, it is often necessary to tailor the material design to have some desired properties. As the set of material design parameters grows, the search space grows exponentially, making the actual synthesis and evaluation of all combinations of designs virtually impossible. Even using traditional computational methods, such as Finite Element Analysis (FEA), becomes too computationally heavy to search this design space. Recent methods use machine learning (ML) surrogate models to more efficiently determine optimal material designs; unfortunately, these methods often (i) are notoriously difficult to interpret and (ii) under perform when the training data comes from a non-uniform sampling of the entire design space. In this work, we suggest the use of tensor completion methods as an all-in-one approach for interpretability and predictions. We observe classical tensor methods are able to compete with traditional ML methods in predictions, with the added benefit of their interpretable tensor factors (which are given for free). In our experiments, we are able to rediscover physical phenomena via the tensor factors, indicating that our predictions are aligned with the physics of the problem. This also means these factors could be used by experimentalists to identify potentially novel patterns, given we are able to rediscover existing ones. We also study the effects of both types of surrogate models (traditional ML \& tensor-based) when we encounter training data from a non-uniform sampling of the design space. We observe some more specialized tensor methods that are able to give better generalization in these non-uniform sampling scenarios, due to the low-rank constraint. We find the best generalization comes from a tensor model, which is able to improve upon the baseline ML methods by up to 5\% on aggregate $R^2$, and halve the error in some out of distribution sections.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。