arXiv:2409.15616cs.LG2024-09被引 16

用强化学习自动生成可解释的聚合物特征,提升性能预测准确率

Reinforcement Feature Transformation for Polymer Property Performance Prediction

  • 通过三级强化学习框架自动构建特征,实现生成与筛选的交互优化
  • 在多个聚合物数据集上显著提升预测性能,最高达9.2%的准确率增益
  • 方法可解释性强,适合材料研发中需要理解特征作用的场景

聚合物性能预测旨在预测聚合物的特定属性,已成为评估其性能的有效手段。然而,现有机器学习模型因聚合物数据集质量低,难以有效学习聚合物表征,从而影响整体性能。本研究聚焦于通过重构最优且可解释的描述符表示空间,提升聚合物性能预测效果。传统特征工程和表征学习方法存在人力成本高或不可解释的问题,难以完全解决该任务。为此,本文提出可追溯的分组强化生成视角(Traceable Group-wise Reinforcement Generation Perspective)。具体地,将表示空间重构视为一个交互过程,结合嵌套生成与选择机制:生成创建有意义的描述符,选择则消除冗余以控制描述符规模。采用三级马尔可夫决策过程进行级联强化学习,自动完成描述符与操作选择及跨域融合。通过分组生成策略探索并增强级联智能体的奖励信号。实验验证了所提框架的有效性,在多个聚合物数据集上表现出色。

原文摘要 · Abstract (English)

Polymer property performance prediction aims to forecast specific features or attributes of polymers, which has become an efficient approach to measuring their performance. However, existing machine learning models face challenges in effectively learning polymer representations due to low-quality polymer datasets, which consequently impact their overall performance. This study focuses on improving polymer property performance prediction tasks by reconstructing an optimal and explainable descriptor representation space. Nevertheless, prior research such as feature engineering and representation learning can only partially solve this task since they are either labor-incentive or unexplainable. This raises two issues: 1) automatic transformation and 2) explainable enhancement. To tackle these issues, we propose our unique Traceable Group-wise Reinforcement Generation Perspective. Specifically, we redefine the reconstruction of the representation space into an interactive process, combining nested generation and selection. Generation creates meaningful descriptors, and selection eliminates redundancies to control descriptor sizes. Our approach employs cascading reinforcement learning with three Markov Decision Processes, automating descriptor and operation selection, and descriptor crossing. We utilize a group-wise generation strategy to explore and enhance reward signals for cascading agents. Ultimately, we conduct experiments to indicate the effectiveness of our proposed framework.

聚合物预测强化学习特征生成可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。