用物理一致性统一不同精度分子数据,提升多任务学习效果
Physical Consistency Bridges Heterogeneous Data in Molecular Multi-Task Learning
- 基于分子物理规律设计一致性训练,实现不同任务间信息互通
- 高精度能量数据可显著提升结构预测准确率
- 可直接利用力和非平衡结构数据增强结构预测,适用广泛
近年来,机器学习在分子科学任务中表现出强大能力。为规模化支持多种分子属性,模型常采用多任务学习范式。然而,不同分子属性的数据往往不一致:例如平衡结构计算成本高,通常用更廉价但精度较低的方法生成,难以通过传统多任务学习解决。此外,如何有效利用其他任务的丰富数据提升特定任务也面临挑战。针对数据异构问题,我们利用分子任务间的物理规律,设计一致性训练方法,使不同任务能直接交换信息以相互提升。实验表明,高精度能量数据可显著改善结构预测;同时,一致性训练可直接利用力和非平衡结构数据提升结构预测性能,展现出整合异构数据的强大能力。
原文摘要 · Abstract (English)
In recent years, machine learning has demonstrated impressive capability in handling molecular science tasks. To support various molecular properties at scale, machine learning models are trained in the multi-task learning paradigm. Nevertheless, data of different molecular properties are often not aligned: some quantities, e.g. equilibrium structure, demand more cost to compute than others, e.g. energy, so their data are often generated by cheaper computational methods at the cost of lower accuracy, which cannot be directly overcome through multi-task learning. Moreover, it is not straightforward to leverage abundant data of other tasks to benefit a particular task. To handle such data heterogeneity challenges, we exploit the specialty of molecular tasks that there are physical laws connecting them, and design consistency training approaches that allow different tasks to exchange information directly so as to improve one another. Particularly, we demonstrate that the more accurate energy data can improve the accuracy of structure prediction. We also find that consistency training can directly leverage force and off-equilibrium structure data to improve structure prediction, demonstrating a broad capability for integrating heterogeneous data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。