arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.11816cs.LGcond-mat.soft

分子CT:统一不同尺度分子的几何与表示学习

Molecular CT: Unifying Geometry and Representation Learning for Molecules at Different Scales

Changping Laboratory(昌平实验室);Shenzhen Bay Laboratory(深圳湾实验室);Peking University(北京大学);Huawei Technologies Co., Ltd.(华为技术有限公司)

Jun Zhang, Yao-Kun Lei, Yaqiang Zhou, Yi Isaac Yang, Yi Qin Gao

AI总结:

本文提出分子构型变换器(Molecular CT),结合关系感知编码器与几何学习单元,实现跨尺度分子表示学习,以轻量结构在基准上取得相当或更优性能。

摘要:

深度学习正在改变分子物理学的许多领域,并已展现出为具有挑战性的分子建模问题提供新解决方案的巨大潜力。伴随这一趋势,对与分子系统兼容的、富有表现力且多功能的神经网络架构的需求日益增长。为此,我们引入了一种新的深度神经网络架构——分子构型变换器(Molecular Configuration Transformer,简称Molecular CT)。Molecular CT由关系感知编码器模块和计算上通用的几何学习单元组成,因此能够考虑粒子间的关系约束,同时可扩展到不同的粒子数量,并对平移-旋转变换具有不变性。其计算效率和通用性使Molecular CT适用于多种分子学习场景,尤其适合跨不同分子系统的可迁移表示学习。作为示例,我们展示了Molecular CT能够对不同尺度的分子系统进行表示学习,并在常见基准上使用比基线模型更轻量的结构取得了相当或更好的结果。

英文摘要:

Deep learning is changing many areas in molecular physics, and it has shown great potential to deliver new solutions to challenging molecular modeling problems. Along with this trend arises the increasing demand of expressive and versatile neural network architectures which are compatible with molecular systems. A new deep neural network architecture, Molecular Configuration Transformer (Molecular CT), is introduced for this purpose. Molecular CT is composed of a relation-aware encoder module and a computationally universal geometry learning unit, thus able to account for the relational constraints between particles meanwhile scalable to different particle numbers and invariant with respect to the trans-rotational transforms. The computational efficiency and universality make Molecular CT versatile for a variety of molecular learning scenarios and especially appealing for transferable representation learning across different molecular systems. As examples, we show that Molecular CT enables representational learning for molecular systems at different scales, and achieves comparable or improved results on common benchmarks using a more light-weighted structure compared to baseline models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑