English

TAO: Tolerance-Aware Optimistic Verification for Floating-Point Neural Networks

Cryptography and Security 2026-03-03 v3 Artificial Intelligence Machine Learning Systems and Control Systems and Control

Abstract

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 10210310^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.

Keywords

Cite

@article{arxiv.2510.16028,
  title  = {TAO: Tolerance-Aware Optimistic Verification for Floating-Point Neural Networks},
  author = {Jianzhu Yao and Hongxu Su and Taobo Liao and Zerui Cheng and Huan Zhang and Xuechao Wang and Pramod Viswanath},
  journal= {arXiv preprint arXiv:2510.16028},
  year   = {2026}
}

Comments

18 pages, 8 figures

R2 v1 2026-07-01T06:44:00.760Z