atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Optimistic Verifiable Training by Controlling Hardware Nondeterminism

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Low-rank compression and merging · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes an optimistic verifiable training scheme that controls GPU hardware nondeterminism by training in higher precision, rounding intermediate computations, and logging the trainer's rounding decisions with an adaptive threshold so auditors can replicate training exactly.

Problem

Optimistic verifiable training fails because nondeterminism between GPU types makes exact replication of training impossible, so weight-comparison audits are non-robust; proof-based alternatives are too expensive to scale.

Approach

Trainer trains in FP64 (higher than target FP32), rounds after each intermediate computation, and logs rounding directions for outputs far enough from the rounding boundary (adaptive threshold tau); the auditor applies the logged corrections via a rev operation so that weights match bit-for-bit across GPUs, and an interactive Merkle-tree verification game localizes disputes.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use rounding logs and auditor corrections to synchronize intermediate rounding decisionsOptimistic verifiable training · ResNet-50 and GPT-2 across A40, Titan XP, and RTX 2080 Ticross-GPU training replicationHelped
remove auditor rounding corrections while retaining FP64 training and FP32 rounding (ablation)Optimistic verifiable training · ResNet-50 and GPT-2 cross-GPU trainingcross-GPU training replicationHurt
use standard FP32 training instead of hardware-nondeterminism control (ablation)ResNet-50 and GPT-2 · cross-GPU trainingweight divergenceHurt
use efficient encoding instead of naive log encoding (ablation)Optimistic verifiable training · ResNet-50 and GPT-2, 1 training stepstorage: 456 MB -> 105 MB; 92 MB -> 22 MBHelped
add Zip compression to efficiently encoded rounding logs (ablation)Optimistic verifiable training · ResNet-50 and GPT-2, 1 training stepstorage: 105 MB -> 96 MB; 22 MB -> 20 MBHelped
reduce rounding amount from b=32 to b=26 (ablation)Optimistic verifiable training · ResNet-50 and GPT-2test performanceNo clear effect
reduce rounding amount from b=32 to b=26 (ablation)Optimistic verifiable training · ResNet-50 and GPT-2, 1 training stepstorage: 96 MB -> 91 MB; 20 MB -> 18 MBHelped
reduce rounding amount from b=32 to b=26 (ablation)Optimistic verifiable training · ResNet-50 and GPT-2training timeHurt
increase rounding aggressiveness to b=27 (ablation)Optimistic verifiable training · ResNet-50 and GPT-2auditor rev corrections: 0Helped
apply optimistic verifiable training instead of the zero-knowledge proof systemLogistic Regression training · Logistic Regression, one training passstorage: 350 MB -> 106 KBHelped
apply optimistic verifiable training instead of the zero-knowledge proof-of-training protocolVGG-11 training · VGG-11, one training iterationtraining and auditing time: 22 min. -> 6 sec.Helped
use GPU nondeterminism instead of data ordering as the source of model variationResNet-50 · model divergence comparisonl2 weight difference: $133.2 \pm 9$ -> $1.1 \pm 0.07$Helped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 Shakespeare corpus

Benchmarks: CIFAR-10 Shakespeare corpus

Metrics: exact weight replication across GPUs l2 weight divergence storage cost (MB/GB) training time (s) number of rev corrections test accuracy perplexity test loss

Baselines: proof-of-learning zero-knowledge logistic regression verifiable training zkML VGG-11 proof-of-training MPC-based GPT-2 secure inference

Builds on: verification game proof-of-learning Merkle tree

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy