用环结构增强图变压器,提升有机太阳能电池分子性能预测精度
RingFormer: A Ring-Enhanced Graph Transformer for Organic Solar Cell Property Prediction
- 构建原子与环层级融合的图结构,结合局部消息传递与全局注意力
- 在5个数据集上超越现有方法,CEPDB上相对提升22.77%
- 特别适合需要精准预测分子性能的研究者和材料设计人员
有机太阳能电池(OSCs)是可持续能源生产的重要方向,但高效分子的筛选依赖耗时实验。为加速进展,需开发可精准预测OSC分子性能的机器学习模型。尽管图表示学习在分子性质预测中表现良好,但其在OSC特定任务中仍不足,现有方法难以捕捉影响性能的关键环状结构特征,导致性能受限。为此,我们提出RingFormer,一种专为捕获OSC分子原子与环级结构模式而设计的图变压器框架。RingFormer构建融合原子与环结构的分层图,通过局部消息传递与全局注意力机制生成高表达性图表示,实现精确的性能预测。我们在五个精选的OSC分子数据集上进行广泛实验,结果表明RingFormer持续优于现有方法,在CEPDB数据集上相较最近对手实现22.77%的相对提升。
原文摘要 · Abstract (English)
Organic Solar Cells (OSCs) are a promising technology for sustainable energy production. However, the identification of molecules with desired OSC properties typically involves laborious experimental research. To accelerate progress in the field, it is crucial to develop machine learning models capable of accurately predicting the properties of OSC molecules. While graph representation learning has demonstrated success in molecular property prediction, it remains underexplored for OSC-specific tasks. Existing methods fail to capture the unique structural features of OSC molecules, particularly the intricate ring systems that critically influence OSC properties, leading to suboptimal performance. To fill the gap, we present RingFormer, a novel graph transformer framework specially designed to capture both atom and ring level structural patterns in OSC molecules. RingFormer constructs a hierarchical graph that integrates atomic and ring structures and employs a combination of local message passing and global attention mechanisms to generate expressive graph representations for accurate OSC property prediction. We evaluate RingFormer's effectiveness on five curated OSC molecule datasets through extensive experiments. The results demonstrate that RingFormer consistently outperforms existing methods, achieving a 22.77% relative improvement over the nearest competitor on the CEPDB dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。