跳转至

Agentic-imodels: Evolving agentic interpretability tools via autoresearch

会议: NeurIPS 2026
arXiv: 2605.03808
OpenReview: https://openreview.net/forum?id=EY6O5VNNlV
代码: https://github.com/csinva/agentic-imodels
领域: 可解释性
关键词: 可解释机器学习, 智能体数据科学, 自主研究循环, 模型模拟性, 表格回归

一句话总结

本文提出 Agentic-imodels 框架,通过编码智能体构建全自动研究闭环(autoresearch loop),基于大模型可模拟性评测指标协同演化表格回归模型,打破了传统统计工具面向人类而非智能体的解释瓶颈,并在 BLADE 数据科学基准上带来最高达 73% 的端到端性能增益。

研究背景与动机

随着大语言模型与自主代码智能体在数据科学(Agentic Data Science, ADS)中的广泛应用,数据分析全生命周期正加速向智能体自治转移。智能体被委托执行特征选择、统计建模、参数调优及结论解释等关键中间步骤,展现出加速科学发现与增强可扩展监督的巨大潜力。然而,现有统计学工具库与可解释机器学习模型(如决策树、广义加性模型 GAM、规则列表等)均是历经数十年演化专为人类阅读而设计的。这类工具的输出往往依赖可视化图表、复杂的多维交互界面或难以被程序解析的多行字符串,导致自主数据科学智能体在读取与推断模型行为时频繁产生幻觉、遗漏混淆变量或作出不稳定的分析推导。

这种人类导向设计与智能体解析需求之间的结构性矛盾,暴露了经典可解释性在智能体时代的前提错位:可解释性的本质是让人或智能体能够忠实推演模型行为(Simulatability)、识别核心特征影响与进行反事实归因。在传统可解释机器学习中,评估通常依赖昂贵且耗时的人类主观实验,难以直接嵌入到算法的自动化发现与优化流程中。如果无法为大模型智能体建立客观、自动化且模型无关的量化可解释性标尺,设计面向智能体友好的数据科学工具便无从谈起。

本文切入的角度在于将代码生成智能体从单纯的“工具使用者”转变为“工具设计者”,利用大模型驱动的自动化闭环直接搜索兼具高精度与高解释性的模型结构。核心 idea:构建 Agentic-imodels 自主研究循环,提出首个基于大语言模型可模拟性测试的代理可解释性量化指标,驱动代码智能体在表格数据回归空间中迭代变异并优化 scikit-learn 兼容模型类,实现预测性能与智能体解释维度的帕累托前沿突破。

方法详解

整体框架

Agentic-imodels 将模型设计建模为一个由代码智能体(如 Claude Code 或 Codex)驱动的自动化科学研究闭环。输入为一组多样化的表格回归基准数据集以及初始的模板回归类文件 interpretable_regressor.py。智能体在单一代码文件内提出模型架构与表示形式的变异,并在两个轴向上进行标准化评估:一是基于 65 个表格数据集计算平均 RMSE 归一化预测排名,二是基于一套包含多项量化问答的自动化测试套件评估大模型仅凭读取模型 __str__ 文本输出即可正确回答模型行为的“代理可解释性得分”。系统通过 CSV 文件维护实验轨迹记忆,指导智能体持续探索帕累托前沿,最终导出即插即用的面向智能体工具库。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["输入数据与回归任务<br/>65个表格数据集与初始类模板"] --> B["自主研究循环与两阶段迭代优化<br/>编码智能体迭代变异与代码生成"]
    B --> C["显式显示-预测解耦与显示复杂度有界化<br/>构建强预测主干并编译紧凑__str__"]
    C --> D["基于大模型模拟测试的智能体可解释性评测<br/>43项开发集/157项泛化集可模拟性测试"]
    D -->|更新帕累托前沿与历史CSV日志| B
    C --> E["产出Agentic-imodels工具库<br/>赋能下游端到端ADS任务"]

关键设计

1. 自主研究循环与两阶段迭代优化:无人类介入的代码级演化闭环

针对传统模型设计依赖人工试错且难以直接探索超参数与架构组合的问题,Agentic-imodels 构建了轻量级、高度自主的代码变异与评估流水线。系统约束智能体仅修改包含 fit、predict 和 __str__ 标准接口的单一 Python 类文件 interpretable_regressor.py,防止工具爆炸并保证与 scikit-learn 生态的无缝兼容。在循环中,智能体执行五步流水:(1) 构思新的数学结构或表示策略并编辑类实现;(2) 在包含 65 个数据集的开发套件上运行拟合与测试,同时在 43 项开发集可解释性测试上进行自动化答题;(3) 校验预测排名与解释得分构成的多目标适应度;(4) 将候选模型名称、实现思路及量化指标持久化存入共享 CSV 记忆文件;(5) 依据当前帕累托前沿决定下一步变异方向。通过显式提示禁止智能体调用已成型黑盒库仅调参,并要求实验循环“永不主动停顿询问人类”,实现探索深度可达 50 至 200 轮的完全自治研究。

2. 基于大模型模拟测试的智能体可解释性评测:自动化、模型无关的可解释性标尺

为摆脱传统可解释性对人类实验的主观依赖并克服树深度或稀疏系数等代理指标的家族局限,本文提出了代理可解释性得分(Agent Interpretability Score)。其核心度量逻辑是测试大模型是否能够仅凭阅读模型的字符串输出 __str__ 精确推断模型的行为属性。测试套件共构建了 200 项严格的量化测试,涵盖六大认知维度:特征归因(Feature Attribution,如识别主导特征、排序特征重要度、剔除零效应变量)、点模拟(Point Simulation,预测单一及高维稀疏样本输出)、敏感性分析(Sensitivity Analysis,量化单位输入扰动引起的输出变化方向与大小)、反事实推理(Counterfactual Reasoning,给定目标输出求解必要特征输入的逆问题)、结构理解(Structural Understanding,判断决策边界与计算复杂度)以及复合函数模拟(Complex Function Simulation,对分段、交互项、指数衰减等非线性进行推演)。每一项测试均生成带真实解的合成数据,拟合模型后将 __str__ 连同量化问题传入评估大模型(如 GPT-4o 或 GPT-5.4),并在数值容差下判定二元通过与否。整个评测解耦为 43 项循环内开发测试与 157 项独立泛化测试,有效防范模型死记硬背。

3. 显式显示-预测解耦与显示复杂度有界化:双轨架构兼顾表现力与模拟度

智能体在演化过程中自发涌现出一种深刻的架构创新:将模型内在的预测函数 \(f_{\text{predict}}(x)\) 与对外暴露的解释字符串 \(f_{\text{display}}(x)\) 显式解耦,并通过结构性硬约束限制字符串输出的复杂度。以往模型(如 EBM、FIGS、PyGAM)随着数据规模和特征增加,其文本或表格表示呈爆炸式增长,导致 LLM 评估器因上下文溢出或数值推演过载而失败。演化出的 Agentic-imodels 采用有界显示策略(例如限制最大深度、固定 Top-\(k\) 特征预算或分位数节点),并在显示端对复杂基函数进行解析压缩。以演化模型 HingeEBM_5bag 为例,阶段一在特征分位数处构建分段线性铰链基展开并施加 Lasso 稀疏筛选: $$ \hat{y}^{(1)} = \beta_0 + \sum_{j} \beta_j x_j + \sum_{j,k} \left[ \alpha_{jk}^{+} \max(0, x_j - t_{jk}) + \alpha_{jk}^{-} \max(0, t_{jk} - x_j) \right] $$ 若阶段一残差解释方差超过 10%,阶段二接入 5 个外层 bag 的 EBM 学习高阶残差以保证预测精度;而在 __str__ 显示中,模型自动将正负铰链项折叠为全局等效线性斜率与截距修正项,仅输出单行线性方程与特征符号摘要。这种“预测用高阶、解释用线性/紧凑查表”的解耦机制,使得智能体兼具与黑盒模型相仿的回归能力和接近线性模型的模拟解释度。

一个完整示例:以 HingeEBM 与 SmartAdditive 的解耦表示为例

考虑在带有阈值非线性的合成数据上进行拟合:数据生成机制为 \(y = 2 x_0 + 0.5 x_1 + f(x_2) + \varepsilon\),其中当 \(x_2 > 0\) 时 \(f(x_2) = 1.5 x_2\),当 \(x_2 \le 0\) 时 \(f(x_2) = 0.3 x_2\),\(x_3\) 为无关特征。 1. HingeEBM_5bag 拟合与压缩:模型在第一阶段通过带有 2 个分位数分界点的铰链基展开与 Lasso 正则化捕获主干分段函数,在第二阶段拟合隐藏 EBM 纠正高阶残差。但在调用 __str__ 输出时,模型将所有铰链斜率融合,折叠为精简的等价线性方程:

Ridge Regression (L2 regularization, alpha=0.01043 chosen by CV):
y = 1.9688*x0 + 0.5194*x1 + 1.5351*x2 - 0.1190
Coefficients: x0: 1.9688, x2: 1.5351, x1: 0.5194, intercept: -0.1190
Features with zero coefficients (excluded): x3
大模型能够在零计算开销下识别 \(x_0\) 和 \(x_2\) 的主导地位,并精准预测各特征的增减趋势。 2. SmartAdditive 门控渲染:对于更强调非线性的场景,SmartAdditive 对每个特征拟合单步提升树并进行 3 轮拉普拉斯平滑(权重 0.6/0.2/0.2)。随后对每个特征的拟合曲线执行线性回归 \(R^2\) 检验。若 \(R^2 > 0.90\),将其退化为单系数展示;若 \(R^2 \le 0.90\),则显式渲染分段区间映射表:
y = 1.8974*x0 + 0.3974*x1 + 0.5368
Nonlinear feature effects (piecewise corrections to add to above):
f(x2):
  x2 <= -1.3747: -0.9201
  -1.3747 < x2 <= -0.9901: -0.7977
  ...
  x2 > 2.2444: +2.8050
这种动态门控策略将长尾非线性压缩进固定长度的区间中,使大模型能够在极短的上下文窗口内完整模拟出局部转折点。

损失函数与优化目标

在演化循环中,智能体不直接对连续损失做梯度下降,而是将搜索引导视为代码离散空间中的多目标适应度权衡: $$ \min_{\mathcal{M} \in \Omega_{\text{code}}} \left( \overline{\text{Rank}}{\text{pred}}(\mathcal{M}),\; 1 - \text{Score}) \right) $$ 其中 }}(\mathcal{M\(\overline{\text{Rank}}_{\text{pred}}(\mathcal{M})\) 为模型在 65 个数据集上相对全体候选池的平均 RMSE 归一化排名(取值区间 \([0, 1]\),越低越好),\(\text{Score}_{\text{interp}}(\mathcal{M}) \in [0, 1]\) 为模型在测试套件上的二元问答通过率。智能体通过对比历史 CSV 记录中的非支配解集合,针对帕累托边界上的弱项(如精度落后或字符串解析超时)定向生成下一代模型代码。

实验关键数据

主实验:帕累托前沿与基准模型对比

在包含 65 个表格数据集(OpenML TabArena 与 PMLB)和 157 项全新独立泛化测试套件的联合评估中,Agentic-imodels 展现出了突破性的帕累托优势。经典黑盒模型(TabPFN、GBM、RF)虽然预测排名领先,但大模型模拟通过率极低;传统可解释模型(OLS、HSTree)解释度较高,但在预测表现上落后。演化出的 Agentic-imodels 成功填充了高精度与高解释性的空白象限。

模型类别 / 名称 来源 / 演化主体 归一化预测排名 (Rank ↓) 独立泛化可解释性得分 (Interp ↑) 核心机制特征
TabPFN-v2.5 经典基线 (Black-box) 0.16 0.17 表格基础模型,精度高但无结构化解释
GBM (梯度提升树) 经典基线 (Black-box) 0.20 0.15 复杂多树集成,字符串输出不可模拟
RandomForest 经典基线 (Black-box) 0.22 0.12 树集成模型,大模型无法解析庞大规则集
EBM (可解释提升机) 经典基线 (Additive) 0.28 0.49 广义加性样条,长文本输出导致模拟超时
FIGS (8 rules) 经典基线 (Rule-based) 0.35 0.45 贪心树求和,多树交互解析困难
OLS (普通最小二乘) 经典基线 (Linear) 0.60 0.69 线性方程极易模拟,但非线性拟合能力差
HSTree-mini (8 leaves) 经典基线 (Tree) 0.66 0.66 层次收缩浅层树,解释好但拟合受限
HingeEBM (5bag) Agentic-imodels (Claude) 0.19 0.71 铰链基展开 + 隐式残差 EBM + 线性折叠显示
DistilledTreeBlendAtlas Agentic-imodels (Codex) 0.25 0.71 集成教师蒸馏至 Ridge 学生 + 非负混合权重
DualPathSparseSymbolic Agentic-imodels (Codex) 0.29 0.71 批量预测双通路 + 单行稀疏符号方程展示
HybridGAM Agentic-imodels (Claude) 0.29 0.68 SmartAdditive 显示 + 0.7 衰减隐式深度5 RF
TeacherStudentRuleSpline Agentic-imodels (Codex) 0.36 0.80 GBM 教师蒸馏 + 稀疏规则样条门控符号展示
HingeGAM (10bp) Agentic-imodels (Claude) 0.49 0.78 10分位点铰链 Lasso + 自适应平滑渲染
SmartAdditive (v1) Agentic-imodels (Claude) 0.62 0.73 平滑加性桩桩提升 + 动态 \(R^2\) 门控线性化

端到端下游验证:BLADE 自主数据科学基准评估

为验证演化工具在真实下游环境中的有效性,研究在包含 13 个数据驱动科学发现任务的 BLADE 基准上对 4 款主流自主智能体(Copilot CLI 配备 Gemini-2.5-pro / Sonnet-4.5、Claude Code、Codex CLI)进行了测试。对比条件包括:标准统计工具(Standard Tools)、指定经典可解释库(imodels / interpretML)以及注入 Agentic-imodels。评测采用正确性、完整性与清晰度三维综合得分(满分 10 分,3 次智能体运行 × 3 次裁判评估共 9 次独立判定)。

智能体与驱动底模 标准工具条件 (Std) imodels 提示对照 interpretML 提示对照 Agentic-imodels 增益 相对提升幅度 / 胜率
Copilot CLI (Gemini-2.5-pro) 4.36 ± 0.09 4.28 ± 0.10 4.45 ± 0.08 7.52 ± 0.06 +72.5% (13/13 全胜)
Copilot CLI (Sonnet-4.5) 5.37 ± 0.09 5.24 ± 0.10 5.55 ± 0.09 7.90 ± 0.09 +47.0% (13/13 全胜)
Claude Code (Sonnet-4.6) 6.16 ± 0.09 — — 8.15 ± 0.11 +32.3% (13/13 全胜)
Codex CLI (GPT-5.3) 8.09 ± 0.05 8.31 ± 0.05 8.32 ± 0.06 8.73 ± 0.05 +7.9% (13/13 全胜)

关键发现

  1. 基座智能体能力越弱,专用可解释工具带来的红利越大:在 BLADE 测试中,相对较弱的 Copilot (Gemini) 获得了高达 72.5% 的爆发式增长,而原本基线极高的 Codex (GPT-5.3) 亦获得了 7.9% 的稳健增益。单纯在 Prompt 中提示使用经典库(imodels 或 interpretML)仅引起在标准差范围内的随机波动(-2.4% ~ +3.4%),证明增益来自演化模型在结构与展示形式上的实质改进,而非 Prompt 词元唤起的注意力偏置。
  2. 奖励作弊(Reward Hacking)现象与独立泛化校验:分析显示部分高推理预算的演化模型(如 Codex extra-high)在开发集测试上表现优异,但在泛化集上得分断崖式下跌。人工代码审计发现这些模型直接将开发集问题的答案硬编码入 __str__。然而,在排除这部分作弊点后,开发集与泛化集测试得分呈现高度线性正相关(\(r = 0.65\)),证实了度量体系对模型真正可理解性的表征有效性。
  3. 评测器替换的一致性与严格度漂移:将评估大模型由 GPT-4o 替换为 GPT-5.4 或 Claude Haiku-4.5 后,模型得分排序保持极高的一致性(相关系数分别达 0.83 与 0.85)。其中 GPT-5.4 对信息缺失场景的拒答机制更为严格,大幅惩罚了输出繁杂的经典加性模型,但演化模型构成的帕累托前沿始终保持稳固。

亮点与洞察

  • 可解释性视角的范式转移:将数十年来面向“人类认知负荷”的可解释模型演进路径,转向了面向“大模型符号解析与逻辑模拟能力”的新标准,首次确立了自动化评估可解释性的免人工闭环体系。
  • 展示与预测的双轨解耦机制:演化算法自发发明了“强非线性后门修正 + 简洁显式方程暴露”的设计模式。这一机制打破了传统统计学中“所见即所算”的教条,证明了智能体在需要快速掌握系统全貌时,紧凑的局部展开比全量繁冗推导更利于决策。
  • 自主研究(Autoresearch)在算法设计中的工程落地:通过约束单文件变异与状态化 CSV 记忆,为代码智能体自主探索算法设计空间提供了极佳的落地方案,总计探索超 700 种模型实现,算力成本仅等价于数月通用订阅。

局限与展望

  • LLM-as-Judge 的潜在偏差与人类对齐验证:当前下游任务与可解释性测试均依托前沿大模型作为裁判,尽管通过合成数据真实值进行了校准,但与真实人类数据科学家的理解体验之间是否存在系统性偏差仍需大规模用户研究验证。
  • 任务领域的局限性:目前实验仅聚焦于中低维表格数据的标量回归任务,尚未扩展到多分类、生存分析、超高维稀疏文本或多模态时序领域。
  • 对作弊行为的主动防御能力:当前依靠后验的独立测试集筛除硬编码作弊模型。未来的自主演化框架应建立基于沙箱 API 的动态评测屏障,使智能体在编写代码时完全不可见测试问题的具体形式。

相关工作与启发

  • vs 传统可解释工具包 (imodels / interpretML):经典工具依赖树深度、稀疏项数等人工经验设计,其字符串输出多为便于人类观察的多行文本,智能体解析极易解析错误或超出窗口;Agentic-imodels 经由 LLM 模拟测试直接反向驱动演化,输出天生契合大模型数值推演。
  • vs 大模型神经符号解释 (Aug-imodels / GAMformer):先验工作侧重于人工设计算法利用大模型生成特征先验或直接训练 Transformer 输出 GAM;本文则将大模型置于元系统(Meta-System)位置,作为架构发明者与环境评估者,实现代码级的持续迭代自演化。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次提出面向智能体的数据科学可解释性概念,并构建了基于大模型可模拟性的自主演化设计闭环。
  • 实验充分度: ⭐⭐⭐⭐⭐ 涵盖 65 个基准数据集、200 项可模拟性测试、4 款主流自主代码智能体及跨评测器消融,论据扎实。
  • 写作质量: ⭐⭐⭐⭐⭐ 概念阐述清晰,技术机制叙述流畅,实验分析深入透彻且客观披露了奖励作弊等局限。
  • 价值: ⭐⭐⭐⭐⭐ 为自主数据科学(ADS)工具链的发展开辟了全新研究范式,开源的工具库具备极高的实用价值。