使用多头注意力回归模型预测烧结的NdFeB磁铁的性能
Qichao Liang1,2, Qiang Ma3, Hao Wu4
1Department of Rare Earth, Jiangxi University of Science and Technology, Ganzhou, 341000, China. qichaoliang1984@163.com.
Scientific reports
|November 21, 2024
概括
一个新的多头注意力回归模型提高了合成合铁 (NdFeB) 磁铁的效率. 这种机器学习方法提高了数据处理和材料性质的预测准确性.
科学领域:
- 材料科学 材料科学 材料科学
- 数据科学数据科学数据科学
- 化学工程是化学工程的重要组成部分.
背景情况:
- 烧结-铁- (NdFeB) 磁铁的制备是复杂的,耗时的,昂贵的.
- 传统的机器学习模型面临着大型,高维数据集和超参数调整的挑战.
- 神经网络提供强大的非线性建模,但缺乏可解释性.
研究的目的:
- 开发一个更高效和可解释的机器学习模型,用于NdFeB磁铁合成.
- 为了增强对材料结构属性关系的理解.
- 为了加快材料性能的优化.
主要方法:
- 收集了1200个高质量的 NdFeB 磁铁实验数据点.
- 开发了一个多头注意力回归模型,将注意力机制集成到神经网络中.
- 利用并行数据处理来加速训练和推理.
主要成果:
- 实现了0.97的残余和0.84的强制性确定系数.
- 证明对特征工程和超参数调整的依赖减少.
- 展示了神经网络模型的增强解释性.
结论:
- 多头注意力回归模型显著提高了NdFeB磁铁属性预测的效率和准确性.
- 这种方法为基于机器学习的材料建模提供了有价值的见解.
- 这项研究为先进的多式联接 NdFeB 磁铁模型铺平了道路.


