跳转至

ECSEL: Explainable Classification via Signomial Equation Learning

会议: ICML 2026
arXiv: 2601.21789
代码: https://github.com/AdiaLumadjeng/ecsel (有)
领域: 可解释机器学习 / 符号回归 / 内在可解释分类器
关键词: signomial函数、符号回归、可解释分类、L1稀疏正则、闭式归因

一句话总结

ECSEL 把"每个类别一个 signomial(带实数指数的幂律和)函数 + softmax"作为分类器,配合 L1 稀疏正则与多阶段优化,既能在 AI Feynman 等符号回归 benchmark 上以远低于 SOTA 的算力恢复 95.86% 的目标方程,又能在 11 个分类数据集上与 XGBoost/MLP 打平,同时所有特征归因都由模型参数闭式给出。

研究背景与动机

领域现状:当前可解释 AI 主要有两条路线。一是 post-hoc 解释(LIME、SHAP、Integrated Gradients),在黑盒模型外再训一个替代模型来解释预测;二是 inherently interpretable 模型(决策树、GAM、稀疏线性模型),结构本身就是解释。符号回归(SR)则属于第二类的极端形态——直接产出一条人类可读的方程。

现有痛点:通用 SR 方法(GP、PySR、DGSR、NeSymRes)把搜索空间设成"任意函数形式",导致两个问题:(1) 算力极大,DGSR 在一个方程上平均要 612s,且经常超时;(2) 高维数据上性能崩坏。而 post-hoc 解释又被 Rudin 等人批评为"对高风险决策不可靠"。

核心矛盾:通用 SR 的表达能力 没有被benchmark 兑现——作者发现 AI Feynman 100 个物理方程里 45 个本身就是 signomial(\(\sum_k \alpha_k \prod_j x_j^{\beta_{k,j}}\) 形式的幂律和)。也就是说,benchmark 早就在喊"我有结构",但通用方法非要在巨大空间里盲搜。

本文目标:(1) 把 signomial 作为一类正经的"模型族"而不是优化目标;(2) 让 signomial 既能做 SR、又能做分类;(3) 让"全局/决策边界/局部"三层解释都从模型参数 闭式 推出,不再需要采样。

切入角度:signomial 在对数空间里就是线性函数(\(\log z = \sum_j \beta_j \log x_j + \log\alpha\)),所以指数 \(\beta_{k,j}\) 直接编码了"特征对输出的弹性"(economics 里的 elasticity)。这是天然的"参数即解释"结构。

核心 idea:用"每类一个 signomial + softmax + L1 正则"换掉 deep classifier,把"训练成本"换成"零成本的解释"。

方法详解

整体框架

ECSEL 要解决的问题是:既能做符号回归、又能做分类,而且分类器的"解释"不靠事后采样、直接从参数读出来。它的做法是把"每个类别一个 signomial 函数 + softmax"当成分类器本体。给定特征向量 \(x \in \mathbb{R}^m\),先经一次仿射变换把每维压到 \([1, 10]\)(signomial 的幂律要求底数为正),再为每个类 \(c\) 学一个由 \(K\) 个幂律项加性组成的分数函数 \(z_c(x) = \sum_{k=1}^{K} \alpha_{c,k} \prod_{j=1}^{m} x_j^{\beta_{c,k,j}}\),参数是系数 \(\alpha_{c,k} \in \mathbb{R}\) 与指数 \(\beta_{c,k,j} \in \mathbb{R}\),超参 \(K\) 控制复杂度(\(K=1\) 为单幂律,\(K>1\) 为多幂律加性组合)。多类问题在 \(\{z_c\}\) 上接 softmax、二类接 sigmoid 给出概率,SR 版本只把 cross-entropy 换成 MSE。作者还为这套结构补了理论靠山——Signomial 通用近似定理:经 \(\log\) 变换映到正 orthant 上就是指数线性函数,再套 Stone-Weierstrass,可证 signomial 在 \(\mathbb{R}^m_{>0}\) 紧子集上对连续函数稠密,从而与神经网络同属"万能近似器",只是天然偏好乘性幂律关系。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["特征向量 x"] --> B["仿射映射到 [1, 10]<br/>保证幂律底数为正"]
    B --> C["类别专属 signomial + L1 指数稀疏化<br/>每类一个幂律和得分 z_c"]
    C --> D["softmax / sigmoid 输出概率"]
    E["多阶段 staged optimization<br/>K=1 用 L-BFGS-B;K>1 走 Adam→精修→L-BFGS"] -->|"训练系数 α 与指数 β"| C
    C -->|"训完直接读参数"| F["闭式三层解释族<br/>全局弹性 / 决策边界 / 局部归因"]

关键设计

1. 类别专属 signomial + L1 指数稀疏化:让得分函数本身就是一条可读方程并自动选特征

传统 GAM、稀疏线性模型只能做加性组合,捕捉不到 e-commerce 里 PageValue 与 ExitRate 之间那种乘除交互;黑盒模型虽能捕捉交互,却又要回头靠 SHAP 来解释。ECSEL 让每个类 \(c\) 拥有独立的一组 \(\{\alpha_{c,k}, \beta_{c,k,j}\}\),其得分函数本就是一条人类可读的"乘除分式"方程,乘性的幂律结构天然表达这类弹性交互。关键的稀疏化技巧是把 L1 罚项只加在 指数 上:训练目标 \(\mathcal{L} = -\frac{1}{N}\sum_i \log p_{y_i}(x_i) + \lambda \sum_{c,k,j} |\beta_{c,k,j}|\) 把不相关特征的 \(\beta\) 推向 0,而 \(\beta=0\) 等价于 \(x_j^0 = 1\),即"把那一项里这个特征抹掉",于是产出真正稀疏的方程。这和稀疏系数 \(\alpha\) 有本质差别——稀疏 \(\beta\) 是在每一项内部做"特征选择",稀疏 \(\alpha\) 只是淘汰整项的"项选择",前者粒度更细。又因为 signomial 在 \(\log\) 后是线性的,这种乘性结构还顺带保证了后面的归因可以闭式给出。

2. 多阶段 staged optimization:让非凸指数空间可靠收敛

signomial 理论上很美,但指数 \(\beta\) 可以取任意实数(含负值、分数),梯度对 \(\beta\) 形如 \(z_{c,k}(x) \cdot \log x_j\),量级极易爆炸,直接 Adam 一把梭往往要么卡在局部极小、要么发散。ECSEL 因此按 \(K\) 分情形优化:\(K=1\) 时整个目标是低维光滑函数,用 L-BFGS-B 直接求解;\(K>1\) 时空间高维非凸,走三段策略——① Adam 配强 L1 做"结构发现",让各项分化出不同角色;② 减弱 L1 做"精修";③ 用最优 Adam 点初始化 L-BFGS 做最后抛光,并对多个随机种子 multi-start。同时对幂律项做 \(\log\) 域变换加特征缩放以保数值稳定。这套"先用噪声梯度跳出局部、再用二阶法收敛"的流程正是把理论上漂亮的 signomial 落地成"实际能跑"的关键,也直接把 SR 恢复率从 DGSR 的 59% 抬到 95.86%。

3. 闭式三层解释族:全局弹性 / 决策边界 / 局部归因都化为参数代数式

SHAP、LIME 之所以慢(KernelSHAP 在 OSI 上要 28.5s),是因为它们在用 Monte Carlo 采样去逼近一个本应有解析式的量;signomial 的 \(\log\) 线性结构让这些量都能闭式写出,训完模型后任何解释查询都只是参数代数运算、零额外计算。具体有四类:(a) 全局弹性 \(E_{c,j}(x) = \partial \log z_c / \partial \log x_j = \sum_k \frac{z_{c,k}(x)}{z_c(x)} \beta_{c,k,j}\)\(K=1\) 时退化为常数 \(\beta_{c,j}\);(b) counterfactual——把 \(x_j\) 乘以 \(q\) 后新分数直接是 \(z_c^{\text{new}}(x) = \sum_k q^{\beta_{c,k,j}} z_{c,k}(x)\),无需重新预测;(c) 决策边界灵敏度 \(\partial(z_c - z_{c'})/\partial \log x_j\)\(K=1\) 时为 \(z_c \beta_{c,j} - z_{c'} \beta_{c',j}\),直接读出"哪个指数差驱动了类间竞争";(d) 局部归因 利用 \(\log z_{c,k}(x) = \log z_{c,k}(b) + \sum_j \beta_{c,k,j} \log(x_j/b_j)\)\(K=1\) 时是 精确 的 SHAP 式分解,\(K>1\) 时退化为一阶线性化 \(\phi_j \approx G_{c,j}(x^*)(\log x_j - \log x_j^*)\)。作者进一步正式证明(Theorem 3.2)ECSEL 满足 G1-G3、D1-D2、L1-L2 全部七条解释性性质,把"声称可解释"升级为"可证明可解释"。

损失函数 / 训练策略

分类用 cross-entropy 加指数 L1:\(\mathcal{L} = -\frac{1}{N}\sum_i \log p_{y_i}(x_i) + \lambda \sum_{c,k,j} |\beta_{c,k,j}|\);SR 改用 MSE 版本 \(\mathcal{L}_{\text{SR}} = \frac{1}{N}\sum_i (y_i - z(x_i))^2 + \lambda \sum_{k,j}|\beta_{k,j}|\)\(\lambda\) 是关键超参(PaySim 上取 \(2 \times 10^4\))。优化器对 \(K=1\) 用 L-BFGS-B,\(K>1\) 用 Adam (warm) + Adam (refine) + L-BFGS (polish) 三段式,超参由 Optuna TPE 在 30 个 trial 内搜得。

实验关键数据

主实验

符号回归(45 个 AI Feynman signomial 子集 + Livermore/Jin/Korns/DGSR 合成集,5 个随机种子 42-46)

方法 符号恢复率 平均耗时(秒/方程)
NeSymRes 56% 126.3
NGGP 58.54% 468.7
DGSR (SOTA) 59.10% 612.9
ECSEL 95.86% 86.4

分类(11 个 binary/multi-class benchmark,5-fold CV,代表性 3 个数据集)

数据集 方法 Acc. F1 少数类 Recall
Ilpd LR 71.55 58.45 3.03
Ilpd XGBoost 72.41 63.03 6.06
Ilpd ECSEL 75.86 74.39 42.42
Compas XGBoost 68.18 68.08 62.54
Compas ECSEL 68.47 68.36 62.82
Transfusion XGBoost 80.06 78.72 38.89
Transfusion ECSEL 79.33 77.95 41.67

ECSEL 在 11 个里 4 个拿 F1 第一(Seeds/Hearts/ILPD/Compas),9 个数据集上和最优方法差距 \(<1\) 个百分点;ILPD 上 F1 比 XGBoost 高 11.36,少数类召回直接 +36 个点。

消融实验 / 解释器对比(OSI e-commerce 数据集)

方法 解释器 计算时间(秒) Top-3 特征
ECSEL 精确指数 0.1 PVER, SI, PV
LR LinearSHAP 0.1 PVER, Mo, PR
LR LIME 5.3 PVER, Mo, PR
RF TreeSHAP 1.5 PVER, PV, SI
RF LIME 32.0 PVER, PV, ER
XGBoost TreeSHAP 0.1 PVER, Mo, SI
XGBoost LIME 7.7 PVER, PR, ER
MLP KernelSHAP 28.5 PVER, PR, Mo

关键发现

  • 结构红利非常大:DGSR 在 AI Feynman signomial 子集上虽然是 SOTA,但因为架构不允许限制函数形式,恢复率只有 59%;ECSEL 直接 hardcode signomial 形式,恢复率涨 37 个点,耗时降到 1/7。
  • 少数类召回是隐形优势:ILPD 上 XGBoost 少数类 recall 才 6%(基本只猜多数类),ECSEL 直接 42%;fraud detection PaySim 上 ECSEL F1 79.08%,超过此前 DSC 的 78%,且 precision 高达 94.27%。
  • 解释成本零摊销:把训练时间多花一点(OSI 上 5.5s vs LR 0.1s),换来推理时 SHAP/LIME 完全不需要。MLP 上 KernelSHAP 要 28.5s 才能跑完测试集解释,ECSEL 0.1s。
  • 学到的方程有 领域意义:PaySim 上 \(\beta_{\text{OBO}} = 1.42\) 揭示"欺诈者超线性地针对高价值账户"——这是黑盒模型给不出来的可执行 insight;OSI 上自动 surface 出 PVER(PageValue/ExitRate)这个组合特征作为 dominant predictor。

亮点与洞察

  • "参数即解释"的彻底贯彻:很多 inherently interpretable 模型(如 GAM)声称可解释但还是要画 partial dependence;ECSEL 把全局弹性、counterfactual、决策边界、局部归因全部都化为 \(\beta\)\(z_{c,k}\) 的代数式,文章 Theorem 3.2 把 7 条性质形式化证明了一遍——这是"声称可解释" → "可证明可解释"的升级。
  • 从"benchmark observation"到"算法设计":作者从一个非常实证的观察(AI Feynman 100 个方程里 45 个是 signomial)反推出方法。这种"benchmark 已经在告诉我答案,但通用方法不去听"的思路可以迁移到其他领域——比如 LLM benchmark 里也常有大量结构化任务被通用模型"过度通用化"地处理。
  • L1 加在指数而非系数:这个小细节关键。\(\beta_j = 0\) 等价于 \(x_j^0 = 1\),等价于"这一项里这个特征不存在",所以稀疏 \(\beta\) 等价于"每一项里都做特征选择"。如果稀疏 \(\alpha\),只能淘汰整项,粒度更粗。这种"在乘性结构里搞稀疏"的思路对其他乘性模型(如 KAN、NAM)都有借鉴价值。

局限与展望

  • 作者承认:\(K\) 必须提前指定,是分类时常规超参但在 SR 里是真限制;高次单变量多项式(Nguyen)上仍打不过特化方法。
  • 自己看出的局限:(1) 要求所有特征 \(> 0\),需要先做 \([1, 10]\) 仿射映射;负值或类别特征处理不优雅;(2) \(K>1\) 时局部归因从精确退化为一阶线性化,"严格可解释性"打了折扣;(3) 多阶段优化里的超参(Adam 步数、L1 退火 schedule)会显著影响最终方程的"美感",重现性是个挑战;(4) 对离散/类别特征几乎没有讨论,这限制了在表格数据外的应用。
  • 改进思路:把 \(K\) 做成可学习的(neural ODE 风格的"按需增长"),或者把指数限制到有理数子集以增强 exact 符号恢复率;探索 group-L1 等让不同类共享特征选择结构;用 Mixture-of-signomials 处理多模态分布。

相关工作与启发

  • vs DGSR/NeSymRes/gplearn(通用 SR):他们在巨大空间里搜任意函数;ECSEL 锁定 signomial 子空间。代价是放弃了非幂律结构(如 \(\sin\)\(\exp\) 等只是参考函数族里没有的),换来 37 个点的恢复率提升和 7 倍加速。
  • vs GAM / Neural Additive Models(NAM):GAM/NAM 是 加性 的可解释模型,无法捕捉乘性交互;ECSEL 是 乘性 可解释模型,自然处理 elasticity 类经济/生物特征。两者互补,未来可以合并成 "Generalized Additive + Multiplicative Models"。
  • vs SHAP/LIME:post-hoc 方法在任意模型上事后采样估计;ECSEL 直接从参数闭式给出,且 G1 ≈ global SHAP,G3 ≈ LIME,L1, L2 ≈ additive SHAP。意义是把 SHAP 的"估计量"变成 signomial 上的"恒等式"——快、确定、可证明。
  • vs KAN(Kolmogorov-Arnold Networks):KAN 也声称可解释,用可学习样条 + symbolification;ECSEL 的 signomial 是更受约束但天然闭式的子空间。两者可以看作"可解释模型"光谱上的不同点。

评分

  • 新颖性: ⭐⭐⭐⭐ 把 signomial 从优化对象提升为模型类是关键 reframing,但每个组件单独都不算新。
  • 实验充分度: ⭐⭐⭐⭐⭐ 45 个 SR 方程 + 11 个分类数据集 + 2 个真实 case study + 与 4 类基线和 5 种解释器全面对比,充分。
  • 写作质量: ⭐⭐⭐⭐ 结构清晰、定理与 property 编号严格;少量公式编号在分类章节略密集。
  • 价值: ⭐⭐⭐⭐⭐ 在金融/医疗等高 stakes 场景给出"无需 post-hoc"的真·可解释分类器,且 PaySim/OSI 两个案例展示了实际能落地的 insight。