让用户验证云端神经网络输出是否可信,不依赖硬件或确定性计算。
TAO: Tolerance-Aware Optimistic Verification for Floating-Point Neural Networks
- 用操作符级容差区域替代逐位相等,实现对浮点计算结果的合理验证。
- 实测在多种硬件上阈值比理论上限紧100~1000倍,对抗攻击成功率0%。
- 适合关注模型服务可信度的开发者和部署在异构硬件上的推理用户。
神经网络越来越多地运行在用户无法控制的硬件上(如云GPU、推理市场)。然而,机器学习即服务(MLaaS)很少透露实际运行的内容,也无法保证输出忠实反映输入。用户无法应对服务降级问题(如模型替换、量化、图重写或广告嵌入变化)。验证输出困难,因为异构加速器上的浮点(FP)执行本质上是非确定性的。现有方法要么对真实浮点神经网络不实用,要么重新引入厂商信任。本文提出TAO:一种容差感知的乐观验证协议,接受在有原则的操作符级可接受区域内的输出,而非要求逐位相等。TAO结合两种误差模型:(i) 声称的操作符级IEEE-754最坏情况边界;(ii) 在不同硬件上校准的紧密经验百分位轮廓。差异触发基于Merkle锚定、阈值引导的争议博弈,递归划分计算图直至只剩一个操作符,此时裁决简化为轻量级理论边界检查或小规模诚实多数投票。未被挑战的结果在挑战窗口后完成,无需可信硬件或确定性内核。我们实现了TAO作为PyTorch兼容运行时和已部署于Ethereum Holesky测试网的合约层。运行时注入图结构,计算各操作符边界,并以微小开销(在Qwen3-8B上仅0.3%)运行未经修改的供应商内核。在A100、H100、RTX6000、RTX4090上的卷积网络、Transformer和扩散模型中,经验阈值比理论边界紧10^2–10^3倍,且基于边界的对抗攻击成功率为0%。TAO在真实异构机器学习计算中实现了可扩展性与可验证性的统一。
原文摘要 · Abstract (English)
Neural networks increasingly run on hardware outside the user's control (cloud GPUs, inference marketplaces). Yet ML-as-a-Service reveals little about what actually ran or whether returned outputs faithfully reflect the intended inputs. Users lack recourse against service downgrades (model swaps, quantization, graph rewrites, or discrepancies like altered ad embeddings). Verifying outputs is hard because floating-point(FP) execution on heterogeneous accelerators is inherently nondeterministic. Existing approaches are either impractical for real FP neural networks or reintroduce vendor trust. We present TAO: a Tolerance Aware Optimistic verification protocol that accepts outputs within principled operator-level acceptance regions rather than requiring bitwise equality. TAO combines two error models: (i) sound per-operator IEEE-754 worst-case bounds and (ii) tight empirical percentile profiles calibrated across hardware. Discrepancies trigger a Merkle-anchored, threshold-guided dispute game that recursively partitions the computation graph until one operator remains, where adjudication reduces to a lightweight theoretical-bound check or a small honest-majority vote against empirical thresholds. Unchallenged results finalize after a challenge window, without requiring trusted hardware or deterministic kernels. We implement TAO as a PyTorch-compatible runtime and a contract layer currently deployed on Ethereum Holesky testnet. The runtime instruments graphs, computes per-operator bounds, and runs unmodified vendor kernels in FP32 with negligible overhead (0.3% on Qwen3-8B). Across CNNs, Transformers and diffusion models on A100, H100, RTX6000, RTX4090, empirical thresholds are $10^2-10^3$ times tighter than theoretical bounds, and bound-aware adversarial attacks achieve 0% success. Together, TAO reconciles scalability with verifiability for real-world heterogeneous ML compute.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。