用视觉信息提升符号回归模型的收敛与泛化能力
ViSymRe: Vision Multimodal Symbolic Regression
- 将多变量方程投影到二维空间,解决高维可视化缺陷
- 在主流基准上表现优于基线,尤其在低复杂度场景下加速推理
- 支持仅依赖数据部署,自动过滤噪声视觉特征
从观测数据中提取可解释方程以描述复杂自然现象,是人工智能的核心目标之一,该领域称为符号回归(SR)。近年来,基于Transformer的范式成为SR新趋势,解决了搜索效率低的问题。然而,数据与方程之间的模态异质性常导致模型收敛困难与泛化性能下降。本文提出ViSymRe——一种视觉符号回归框架,探索视觉模态对Transformer型SR的增强作用。针对高维场景下视觉SR不可训练的问题,提出多视角随机切片(MVRS),通过随机仿射变换将多变量方程投影至二维空间,避免了高维可视化中的变量退化、非线性交互丢失及采样复杂度指数增长等问题,实现低计算成本训练。为支持仅数据部署,设计基于生成技术的双视觉流水线架构,通过辅助视觉解码器直接从数据重建视觉特征,并利用提出的有偏交叉注意力融合模块自动抑制重构噪声的注意力权重,确保后续过程不受噪声模态影响。消融实验表明视觉模态显著提升模型收敛水平并改善各类SR指标。主流基准评估显示,ViSymRe在性能上具有竞争力,尤其在低复杂度与快速推理场景中优势明显。
原文摘要 · Abstract (English)
Extracting interpretable equations from observational datasets to describe complex natural phenomena is one of the core goals of artificial intelligence. This field is known as symbolic regression (SR). In recent years, Transformer-based paradigms have become a new trend in SR, addressing the well-known problem of inefficient search. However, the modal heterogeneity between datasets and equations often hinders the convergence and generalization of these models. In this paper, we propose ViSymRe, a Vision Symbolic Regression framework, to explore the positive role of visual modality in enhancing the performance of Transformer-based SR paradigms. To overcome the challenge where the visual SR model is untrainable in high-dimensional scenarios, we present Multi-View Random Slicing (MVRS). By projecting multivariate equations into 2-D space using random affine transformations, MVRS avoids common defects in high-dimensional visualization, such as variable degradation, non-linear interaction missing, and exponentially increasing sampling complexity, enabling ViSymRe to be trained with low computational costs. To support dataset-only deployment of ViSymRe, we design a dual-vision pipeline architecture based on generative techniques, which reconstructs visual features directly from the datasets via an auxiliary Visual Decoder and automatically suppresses the attention weights of reconstruction noise through a proposed Biased Cross-Attention feature fusion module, ensuring that subsequent processes are not affected by noisy modalities. Ablation studies demonstrate the positive contribution of visual modality to improving model convergence level and enhancing various SR metrics. Furthermore, evaluation results on mainstream benchmarks indicate that ViSymRe achieves competitive performance compared to baselines, particularly in low-complexity and rapid-inference scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。