arXiv:2502.02917cs.LGcs.AI2025-02被引 2

用离线强化学习实现可交互的符号回归,支持专家协作优化方程。

Interactive Symbolic Regression through Offline Reinforcement Learning: A Co-Design Framework

  • 基于强化学习设计符号网络,无需变压器解码器,提升训练效率。
  • 在SSDNC基准上超越现有算法,交互式修正使真实案例性能更优。
  • 适合有物理规律先验知识的研究者,可动态调整方程结构。

符号回归(SR)在从观测数据中发现潜在数学与物理关系方面具有巨大潜力,但表达式的组合空间庞大,给在线搜索方法和预训练变压器模型带来挑战。现有先进方法通常忽略领域专家先验知识,且不支持方程发现过程中的迭代交互。为此,我们提出符号Q网络(Sym-Q),一种大规模符号回归的交互式框架。与以往基于变压器的大型SR方法不同,Sym-Q采用无需变压器解码器的强化学习范式,通过离线强化学习利用任意树编码器进行训练,实现更高效的训练与推理。此外,我们提出协同设计机制,使基于强化学习的Sym-Q可在方程发现任何阶段与领域专家有效互动。用户可动态修改生成的表达式节点,与智能体协作优化数学表达式,使其更符合问题需求与预设物理规律,尤其适用于已有部分先验知识的情况。实验表明,预训练的Sym-Q在具有挑战性的SSDNC基准上优于现有SR算法。在真实案例中,交互式协同设计进一步提升了性能,其增益超过其他主流模型。可复现代码已开源:https://github.com/EPFL-IMOS/Sym-Q。

原文摘要 · Abstract (English)

Symbolic Regression (SR) holds great potential for uncovering underlying mathematical and physical relationships from observed data. However, the vast combinatorial space of possible expressions poses significant challenges for both online search methods and pre-trained transformer models. Additionally, current state-of-the-art approaches typically do not consider the integration of domain experts' prior knowledge and do not support iterative interactions with the model during the equation discovery process. To address these challenges, we propose the Symbolic Q-network (Sym-Q), an advanced interactive framework for large-scale symbolic regression. Unlike previous large-scale transformer-based SR approaches, Sym-Q leverages reinforcement learning without relying on a transformer-based decoder. This formulation allows the agent to learn through offline reinforcement learning using any type of tree encoder, enabling more efficient training and inference. Furthermore, we propose a co-design mechanism, where the reinforcement learning-based Sym-Q facilitates effective interaction with domain experts at any stage of the equation discovery process. Users can dynamically modify generated nodes of the expression, collaborating with the agent to tailor the mathematical expression to best fit the problem and align with the assumed physical laws, particularly when there is prior partial knowledge of the expected behavior. Our experiments demonstrate that the pre-trained Sym-Q surpasses existing SR algorithms on the challenging SSDNC benchmark. Moreover, we experimentally show on real-world cases that its performance can be further enhanced by the interactive co-design mechanism, with Sym-Q achieving greater performance gains than other state-of-the-art models. Our reproducible code is available at https://github.com/EPFL-IMOS/Sym-Q.

符号回归强化学习交互式建模物理规律

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