arXiv:2604.07328cs.LG2026-04

快速预测删掉部分训练数据后模型表现,误差可控且效率接近常规训练。

How to sketch a learning algorithm

  • 基于稳定假设,通过随机复数方向的高阶导数实现算术电路局部压缩。
  • 预计算与预测耗时仅比正常训练/推理慢约 $\tilde{O}(\log(1/δ)/\varepsilon^2)$ 倍。
  • 适用于需要可解释性、隐私保护或模型行为分析的研究者。

训练数据的选择如何影响人工智能模型?这一问题对可解释性、隐私保护和基础科学至关重要。其技术核心是数据删除问题:经过合理预计算后,快速预测若从学习算法中移除特定训练数据子集,模型在给定情境下的行为。本文提出一种数据删除方案,可在深度学习场景下以趋近零的误差 $\varepsilon$ 与失败概率 $δ$ 预测模型输出。我们的预计算与预测算法分别仅比常规训练和推理慢 $\tilde{O}(\log(1/δ)/\varepsilon^2)$ 倍,存储开销相当于 $\tilde{O}(\log(1/δ)/\varepsilon^2)$ 个模型。证明基于一个称为“稳定性”的假设,该假设与强大模型的学习兼容。通过微小 GPT(microgpt)的最小实验集验证了稳定性成立。代码已公开于 https://github.com/SamSpo1/microgpt-sketch。技术上,本工作基于一种新方法:通过在随机复数方向计算高阶导数来局部压缩算术电路。前向模式自动微分使得这些导数可低成本计算。

原文摘要 · Abstract (English)

How does the choice of training data influence an AI model? This broad question is of central importance to interpretability, privacy, and basic science. At its technical core is the data deletion problem: after a reasonable amount of precomputation, quickly predict how the model would behave in a given situation if a given subset of training data had been excluded from the learning algorithm. We present a data deletion scheme capable of predicting model outputs with vanishing error $\varepsilon$ and failure probability $δ$ in the deep learning setting. Our precomputation and prediction algorithms are only $\tilde{O}(\log(1/δ)/\varepsilon^2)$ factors slower than regular training and inference, respectively. The storage requirements are those of $\tilde{O}(\log(1/δ)/\varepsilon^2)$ models. Our proof is based on an assumption that we call stability. In contrast to the assumptions made by prior work, stability appears to be fully compatible with learning powerful AI models. In support of this, we show that stability is satisfied in a minimal set of experiments with microgpt. Our code is available at https://github.com/SamSpo1/microgpt-sketch. At a technical level, our work is based on a new method for locally sketching an arithmetic circuit by computing higher-order derivatives in random complex directions. Forward-mode automatic differentiation allows cheap computation of these derivatives.

模型可解释性数据删除稳定性假设自动微分

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