中文

MatWheel:通过合成数据解决材料科学中的数据稀缺问题

机器学习 2025-04-15 v1 材料科学

摘要

数据稀缺和标注成本高长期以来一直是材料科学领域的持续性挑战。受其在计算机视觉等其他领域潜力的启发,我们提出了 MatWheel 框架,通过使用由条件生成模型生成的合成数据来训练材料属性预测模型。我们探索了两种情景:完全监督和半监督学习。使用 CGCNN 进行属性预测、使用 Con-CDVAE 作为条件生成模型,在 Matminer 数据库中的两个数据稀缺材料属性数据集上进行实验。结果表明,合成数据在极端数据稀缺情景下具有潜力,在两个任务中均可实现与或超过真实样本的性能。我们还发现,伪标签对生成数据质量的影响较小。未来工作将整合更先进的模型并优化生成条件,以提升材料数据飞轮的效果。

关键词

引用

@article{arxiv.2504.09152,
  title  = {MatWheel: Addressing Data Scarcity in Materials Science Through Synthetic Data},
  author = {Wentao Li and Yizhe Chen and Jiangjie Qiu and Xiaonan Wang},
  journal= {arXiv preprint arXiv:2504.09152},
  year   = {2025}
}

备注

AI4MAT-ICLR-2025: ICLR 2025 Workshop on AI for Accelerated Materials Design