Abstract
Choosing a network architecture for zero-knowledge (ZK) proofs of inference needs a circuit-cost estimate that can be checked against the prover, yet NAS work on ZK has relied on untested cost proxies. We study ezkl, a Halo2-based prover, on CIFAR-10 MLPs, CNNs and ResNet-style networks. (1) Cost model. A parametric model of assigned cells, with coefficients fitted on 180 trained, calibrated networks and evaluated on 54 held-out ones, matches ezkl-calibrated logrows exactly for 39 and within one for 51 (Spearman 0.84; cross-validated over all 234 networks: 70% exact, 95% within one). The mismatches have a mechanism: on 72 dumped ezkl settings, logrows is exactly the maximum of four resource requirements (rows, constants, lookup table, range-check table; 72 of 72 sampled networks at logrows 19–23), of which cell count is one. (2) A negative result. Width-matched, ReLU and x² networks have the same calibrated logrows (all 5 skeletons; 45 of 50 matched pairs in the sweep), contradicting the assumption that polynomial activations are cheaper to prove in ezkl; unnormalized x³ + x networks calibrate to a circuit up to 16× larger at equal cell count because their range-check table spans 2²² + 1 values. (3) Benchmark. We release zkml-bench, 474 networks with two training seeds and proxy scores. NEAR predicts accuracy only for ReLU networks (ρ = +0.70) and is indistinguishable from zero on all three polynomial families; a Zen-style score (our variant, not the published Zen-Score protocol) is anti-correlated on polynomial networks; a sign-pattern NASWOT variant is negatively correlated on polynomial networks; ZiCo’s apparent success on x³ + x disappears once failed training runs are removed. Between 39% and 60% of polynomial networks fail to train. (4) Search. With budgets set on ezkl-calibrated circuit size, a tuned joint accuracy-cost proxy almost always reduces to NEAR-only and shows no consistent gain over it, while NEAR-only beats random selection; we withdraw an earlier claim of a joint-proxy gain. Filtering by the cost model’s estimate removes most but not all budget violations. Scope: CIFAR-10, one ezkl version, calibration on trained networks with real inputs, real proofs for seven architectures.
Editorial notes
Points the editor recorded on accepting the paper. They are published with the article; the authors were not required to address them before publication.
- Section 6.9, paragraph 'Why ⌈log2 C⌉ is not always logrows': replace 'none of the 0 ReLU, x² or x²+x networks has such a range check' with the actual count of dumped non-x³+x networks (59, from Table 15).
- Section 7, 'NAS benchmarks' paragraph: 'Layer-level ezkl measurements exist (see below)' should point to the ZK circuit optimization paragraph above, which discusses Akor et al.
- Section 1: the sentence on the 28 s to 970 s proving-time range appears in both 'The problem' and 'What we show empirically'; remove one occurrence.
- Sections 3.1–3.2, 4.4 and 5: fix the rendering artefacts 'AssumptionAssumption 1', 'Eq. Eq. (1)', 'Table 2 andTable 9' and 'Section 5 andSection 6'.
- Section 6.1, 'Released data': the scripts are said to regenerate 'Tables 4–10', but Section 9 says they regenerate every table; extend the range to cover Tables 11–15 if those are also generated.
1. Introduction
The problem.
ZK-ML systems (ezkl, Orion, RISC Zero) let anyone prove neural network inference was correct — without revealing inputs. This enables private biometric verification (e.g., Worldcoin), on-chain AI oracles, and compliant ML auditing. But proving is slow, and its cost depends on architectural choices that are hard to predict: in our measurements real proving time ranges from s to s over the 7 trained networks we proved (Table 10).
What we show empirically.
Real proving time varies by more than an order of magnitude (28 s to 970 s) across the 7 trained networks we proved (Table 10). A width-matched comparison (Table 2 andTable 9) shows that this spread does not come from the ReLU-versus- choice: at identical widths the two have the same calibrated (in all five skeletons, and in 45 of 50 matched pairs across the trained sweep). Differences come from layer type, width, and from ezkl’s range calibration, which raises for unnormalized networks. We therefore report a negative result for the assumption that polynomial activations are cheaper to prove in ezkl. We also report that the zero-cost proxies we tested do not predict accuracy across activation families, that polynomial networks often fail to train, and that, under budgets set on calibrated circuit size, a joint accuracy-cost proxy does no better than NEAR alone.
Why this requires NAS.
Architecture search for ZK inference needs to trade accuracy against circuit cost, and cost estimates must be validated against the prover. We evaluate a cost model against ezkl calibrations of trained networks and evaluate six zero-cost proxies, plus a NASWOT variant, on a benchmark that includes polynomial and BN-free networks, where the standard proxies were not designed to work.
Our contributions.
-
Cost model with a held-out evaluation (Section 3, Table 14): a parametric model of assigned cells, fitted on 180 trained, calibrated networks and evaluated on 54 disjoint ones.
-
Measurements (Section 5 andSection 6.7): a width-matched ReLU result replicated on untrained and trained networks, and the calibration effect.
-
Benchmark and proxy study (Section 6): 474 networks, two seeds, per-architecture results released; proxy correlations with Holm-adjusted significance, confidence intervals, partial correlations, and analyses with and without failed networks and with a common training recipe.
-
Search under calibrated budgets (Section 6): a protocol with a tuning/evaluation split; the joint proxy does not beat NEAR-only, and the cost model’s estimate removes most budget violations.
Paper organization.
Section 2 reviews ZK-SNARK cost and contrasts with MPC. Section 3 states the cost model and the proxy analysis. Section 4 introduces zkml-bench. Section 5 reports width-matched ezkl cost measurements. Section 6 presents empirical results. Section 7 surveys related work. Section 9 concludes.
2. Background
2.1 ZK-SNARK Circuit Cost Model
In Halo2/PLONK over BN254 (as used by ezkl), the circuit is a rectangular table of rows. Cost is dominated by three components:
-
Linear layers: each multiply-accumulate contributes about one assigned cell (fitted coefficient , Section 6), giving cells per layer.
-
Lookup-based activations: non-linearities such as ReLU are evaluated through lookup tables whose size grows with the activation’s integer range; the table is shared across the circuit. At the widths we test, the resulting per-element cost is small relative to the multiply-accumulate rows (Section 5).
-
Polynomial activations (): evaluated with multiplications and no lookup; the fitted per-element cost of the power operator is of the same order as ReLU’s (Section 6).
For a fixed quantization scale and lookup range, the circuit structure is determined at compile time from the architecture specification and does not depend on weight values. In ezkl the scale and range, and hence , are chosen by a calibration pass on data, which can raise (Section 5); our cost model therefore predicts the circuit size up to this calibration effect.
2.2 Contrast with MPC Cost Model
CryptoNAS (Ghodsi et al. 2020) optimizes for MPC (garbled circuits). In MPC: linear layers cost via secret sharing (effectively free), while ReLU requires oblivious transfer (expensive). In ZK-SNARKs, linear layers are not free: multiply-accumulates dominate the assigned cells at the widths we test (Section 5), so an architecture chosen to minimize the ReLU count need not be cheap to prove. We do not evaluate CryptoNAS itself.
3. Cost Model and Proxy Analysis
This section states what we can and cannot establish about ezkl circuit size. The claims are deliberately modest: AssumptionAssumption 1 is a structural statement and a parametric form; its quantitative content is tested empirically in Section 5 andSection 6, not proved.
3.1 A Parametric Cost Model
ezkl compiles an ONNX graph into a PLONKish circuit. Fix the quantization scale and the lookup range (both are chosen by ezkl’s calibration pass). Each node has an operator type , input and output tensor shapes, and attributes (kernel size, stride, padding, groups).
Assumption 1 (Structure of the cost model). *For fixed the circuit layout (columns, selectors, lookup and permutation structure) is a function of the graph structure only and does not depend on the weight values, which enter only as cell assignments. Define as the total number of assigned cells (ezkl’s total_assignments). We use the parametric form
where is the set of Gemm, MatMul and Conv nodes, with for a Gemm node and for a Conv node with output spatial size and groups; is the number of output elements; are coefficients; and is an unmodelled residual.*
What is and is not claimed.
The first sentence of the assumption follows from how PLONKish layouts are built (weights are witness values), but we do not prove it for ezkl’s implementation; it is the premise of the model. It is supported empirically: assigned cells and calibrated were identical across three re-drawn weight and calibration-data seeds in every width-matched skeleton (Table 2). We do not derive the coefficients or the accounting from assigned cells to rows analytically. Instead the coefficients are fitted by non-negative least squares on a fit split of trained, calibrated networks and evaluated on a disjoint split (Section 6), and the relation between assigned cells and is checked empirically (Section 5 andSection 6): the identity holds in the unnormalized width-matched skeletons but for only 163 of 234 trained networks (the calibrated is lower in 52 and higher in 19), and we do not assume it. Empirically (Section 6, 72 dumped circuits at 19–23), ezkl reports rows and the calibrated is the largest of four requirements (rows, constants, lookup table, range-check table), so holds only when another requirement binds exactly one level above the rows requirement . The structural proxy used by the search is the same form with hand-set coefficients; the fitted version is reported alongside it. is reported only through the fit residuals.
3.2 Activation Costs at Equal Multiply-Accumulate Counts
Lemma 1 (Bounded ratio in the idealized model). *Let two networks have identical width profile (hence identical and element counts ) and differ only in per-element activation cost, , with all other terms of Eq. Eq. (1) equal. With the average fan-in,
decreasing in .*
Proof. Divide by ; the ratio is a decreasing function of equal to at and tending to as . ◻
The lemma is deliberately weak. It shows that, in a model where activation costs are fixed constants, networks with equal MACs differ in cell count by at most ; it does not show that FLOPs cannot approximate cell count within a constant factor (for fixed it can). Whether the factor is large is an empirical question: our measurements find (Section 5). The one place where equal-MAC networks differ substantially in ezkl is through calibration, not through cell count: unnormalized networks have the same assigned cells but a larger by up to , i.e. a circuit up to larger, than the other activations at the same width. That is an observation about ezkl’s calibration, and we state it as such.
3.3 Proxy Failure Modes for Polynomial Activations
Proposition 2 (NASWOT degeneracy for activations). For any network whose activations are (more generally with even), the Gram matrix of the binary patterns used by NASWOT has rank one on almost every input set, so is degenerate.
Proof. For the indicator equals one for every non-zero value, which holds almost surely under continuous random inputs. Every sample therefore maps to the all-ones code, is the all-ones matrix, and . ◻
Remarks.
(i) The proposition is mathematics only: standard NASWOT scores the outputs of nn.ReLU modules and returns for any network without them, so the counts in our sweep (all polynomial networks, including the pure- ones) are produced by the implementation and do not test the proposition. (ii) The argument does not cover or , whose sign patterns are not constant. (iii) To avoid the confound we also evaluate a heuristic generalization that records at the input of every activation module; for ReLU it equals the standard score, and for polynomial networks it is defined. This variant is not part of the NASWOT paper.
A Zen-style score (a variant of Zen-NAS, not the published protocol).
Lin et al. (Lin et al. 2021) (Algorithm 1) compute the Zen-Score on the feature map before global average pooling, after removing the residual links, and add a term that sums over the batch-norm layers. We did not implement that protocol. Our score is with and , and it differs from the published score in four ways: (i) it is computed on the network output (logits), not the pre-pooling feature map; (ii) residual links of ResNet-style networks are kept; (iii) it has no batch-norm-variance term (our networks use no BatchNorm, and we do not substitute a LayerNorm or GroupNorm statistic) and instead adds as a second term; and (iv) the perturbation and input scales are fixed as stated. Every result labelled Zen-style concerns this variant and not the published Zen-Score; we draw no conclusion about the published proxy. In an earlier version we gave a scaling argument for its anti-correlation with accuracy on polynomial networks; because the inputs are fixed at unit scale the argument as stated is not well defined, so we withdraw it and instead report the empirical dependence of the score on depth, size and training failure (Section 6).
4. The zkml-bench Benchmark
4.1 Search Space
Activation types.
ReLU, , , . ReLU is the baseline (lookup-based); is the simplest polynomial; is a signed polynomial with better expressivity; approximates GELU-style behavior.
Normalization.
none, LayerNorm, GroupNorm. BatchNorm is excluded because its inference form divides by , which we expect to be costly in ZK but did not measure.
Convolution type.
standard conv, depthwise separable (DWS). DWS reduces parameters and multiply-accumulates.
Architecture families.
MLP, CNN, ResNet-style with skip connections, DepthwiseSepConv-based.
4.2 Metrics
Evaluated on CIFAR-10, the primary dataset of NAS-Bench-201 (Dong & Yang 2020) (earlier CIFAR-100 and ImageNet-16-120 results are withdrawn; Section 6.1). Recorded per architecture: mean test top-1 accuracy after 30 epochs over two seeds, the zero-cost proxy scores and the structural cost estimate at the seed-0 initialization, and, for the calibrated subset, ezkl’s calibrated and assigned cells on the trained network.
4.3 Zero-Cost Proxies Evaluated
| Proxy | Reference | Data-free | Poly-compatible |
|---|---|---|---|
| Synflow | Tanaka et al. 2020 | ✓ | ✓ |
| NEAR | Husistein et al. 2025 | (1 batch) | ✓ |
| Zen-style (variant) | after Lin et al. 2021 | ✓ (Gaussian) | ✓ |
| ZiCo | Li et al. 2023 | (1 batch) | ✓ |
| MeCo | Jiang et al. 2023 | ✓ | ✓ |
| NASWOT | Mellor et al. 2021 | No (Proposition 2) | |
| Structural (ours) | This work | ✓ | ✓ |
Zero-cost proxies evaluated in zkml-bench.
4.4 Algorithm: Joint Proxy-Guided ZK-ML NAS
Algorithm 1 ranks candidate architectures by a scalarized score of an accuracy proxy and a circuit-cost estimate. Both terms are -scored within the candidate pool :
where is the cell count from the cost model of Eq. Eq. (1). is not fixed a priori: in the experiments it is chosen from by maximizing the best accuracy among the selected architectures on a tuning split that is disjoint from the evaluation split (Section 6). Setting recovers NEAR-only selection.
Candidate architectures , budget on , selection size , weight architectures to train estimated by the cost model, or calibrated by ezkl compute at initialization and from the cost model the architectures with the largest
Properties. The cost term is a deterministic, training-free model (AssumptionAssumption 1); how closely it matches ezkl-calibrated is measured in Section 6. The budget in line 1 is applied either to the model’s estimate or, when calibration has been run, to the calibrated ; the search experiments use the latter, because the former can under- or over-estimate the circuit, and in them the cost term is ezkl’s calibrated cell count (an oracle cost) rather than the model’s estimate. The cost is one NEAR evaluation and one cost-model evaluation per candidate, versus an ezkl calibration or proof per candidate.
5. What Drives ezkl Circuit Cost
This section reports width-matched measurements of ezkl circuit size. Skeletons are identical across columns and only the activation changes. For each network we run ezkl (version 9.5.0) gen_settings and calibrate_settings (target=resources) with random Gaussian calibration inputs and read the calibrated and the row count; no proofs are generated. Weights are random (seed ) and untrained, so calibration ranges reflect untrained networks. Polynomial networks are exported without clamping. Each cell is a single run.
| Skeleton | ReLU | |||
|---|---|---|---|---|
| MLP 128-64-32 | 0.407M (19) | 0.408M (19) | 0.408M (19) | 0.404M (23) |
| MLP 256-128 | 0.823M (20) | 0.825M (20) | 0.825M (20) | 0.818M (23) |
| MLP 512-256 | 1.709M (21) | 1.712M (21) | 1.713M (21) | 1.697M (23) |
| CNN 16-32 | 1.734M (21) | 1.832M (21) | 1.857M (21) | 1.879M (23) |
| CNN 16-32-64 | 2.795M (22) | 2.910M (22) | 2.938M (22) | 2.963M (23) |
Width-matched ezkl circuit size: total assigned cells (calibrated ) for the same skeleton with different activations (no normalization, CIFAR-10 input). Weights and calibration data were re-drawn for three seeds; all assignment counts and are identical across the three seeds, so one seed is shown.
ReLU and have the same circuit size at these widths.
We use ezkl’s total_assignments as the circuit-size measure : for all runs of ReLU, and networks (five skeletons, three seeds), equals the calibrated exactly. The network has the same calibrated as the ReLU network in all skeleton-seed pairs, and more assignments in the MLPs and – more in the CNNs. At these widths the multiply-accumulate rows dominate, and we find no evidence that ReLU is more expensive than . This contradicts a claim made in an earlier version of this work. The large cost differences between architectures in Table 10 therefore come from layer type (depthwise and standard convolutions versus MLP layers) and width, not from the ReLU/ choice, and the lowest-cost architectures in that table are small polynomial MLPs because they are small MLPs.
Unnormalized networks calibrate to a larger circuit.
In every unnormalized skeleton, in all three seeds, the network has about the same number of assignments as the others (– ReLU) but is calibrated at , against – for the other activations; ezkl’s calibration also selected a lower input scale ( versus ). At equal assigned cells this is a circuit to times larger. On trained networks calibrated with real inputs (Section 6.7), 20 of the 24 calibrated unnormalized networks in the sweep are at and the rest between and ; with LayerNorm (15 networks) they range from to . Dumped ezkl settings show the mechanism (Section 6): the calibration adds a range check over , a table of entries, which needs .
Repeated draws.
Re-drawing weights and calibration data (three seeds, untrained) did not change any assignment count or in Table 2, and Table 9 gives the same picture for trained networks and real inputs.
Hypotheses we did not test.
From reading ezkl’s circuit layout code (ezkl/src/circuit/ops/layouts.rs) we expect that reshape and concatenation add no rows, that the reciprocal square root in LayerNorm is a costly operation, and that max-pooling costs more than average-pooling. We did not measure these and make no claim about them.
6. Empirical Results
6.1 Experimental Setup
Benchmark. We regenerated the architecture sweep for this version. It contains 474 CIFAR-10 networks: 78 MLPs, 72 CNNs, 60 depthwise-separable CNNs, 132 ResNet-style and 132 depthwise ResNet-style networks, each with ReLU, , or activations (111, 121, 121 and 121 networks) and no normalization, LayerNorm or GroupNorm. CIFAR-100 and ImageNet-16-120 results from an earlier version are withdrawn: the underlying files could not be recovered and the CIFAR-100 ReLU subset had only three networks.
Training. 30-epoch SGD (momentum , weight decay , cosine schedule, batch , random crop with padding and horizontal flip), learning rate , or with gradient clipping for polynomial networks without normalization. The sweep is run by sweep_gpu.py, which follows this recipe with two differences from the earlier CPU pipeline: the dataset and the augmentation live on the GPU, and the random-crop padding adds zeros in normalized space, whereas the earlier pipeline padded raw images with black pixels (which normalize to a nonzero constant). The two pipelines are therefore not identical, and accuracies are not comparable across them: for the 8 architectures shared with the earlier version the two-seed sweep mean is higher than the earlier three-seed 30-epoch mean for 8 of 8 (by +0.8 to +5.6 points; +5.6 for the LayerNorm MLP). All comparisons in this paper use the new sweep only. Every architecture is trained with two seeds (0 and 1). Zero-cost proxies and the cost estimate are computed once, at the seed-0 initialization. A network is called failed if training diverged or its mean test accuracy is below .
ezkl. All calibrations in this section use ezkl 9.5.0 (calibrate_settings, target=resources), real CIFAR-10 test images as calibration inputs, and the seed-0 trained weights, on a 24-logical-core AMD Threadripper PRO 3945WX host shared with other jobs. One rule decides which networks enter every calibration-based analysis: networks whose seed-0 training failed (nothing meaningful to calibrate) and networks whose estimated exceeds (irrelevant to the budgets studied) are excluded. 255 networks were calibrated; 21 of them met the exclusion rule (they were calibrated in an early pass before the rule was applied) and are dropped, leaving 234, of which 0 failed inside ezkl. The dropped and skipped lists are released.
Released data. Per-architecture results for both seeds (accuracy, proxies at initialization, cost estimate), the calibration records with the per-operator features used for fitting, and every summary number in this section are in the repository, together with the scripts that regenerate Tables 4–10. Trained network weights (about 0.7 GB) are not released.
6.2 Label Noise and Training Failures
Across the two seeds the median per-architecture standard deviation of test accuracy is 0.24 points (90th percentile 1.53; the maximum, 45.4, is a network that trained in one seed and failed in the other), and the Spearman correlation between the two seeds’ accuracies is 0.95. Label noise is therefore small relative to differences between architectures. Training failure is large and depends on the recipe: a network is failed if training diverged in at least one seed or its mean accuracy is below ; 170 of the 474 networks failed, and 153 of them have mean accuracy below . Failures by activation are 0 of 111 (ReLU), 50 of 121 (), 73 of 121 () and 47 of 121 (); by normalization, 37 none, 49 LayerNorm and 84 GroupNorm (polynomial networks with normalization use the larger learning rate, so normalization does not prevent failure here). The accuracy labels of polynomial networks are therefore partly trainability labels, and we report results with and without the failed networks.
6.3 Multi-Seed and Training-Budget Validation
To check that 30-epoch accuracy is a usable ranking signal, we retrained 26 architectures (the eight architectures of the earlier version’s timing table plus activation-matched controls for ReLU, , and ) with three seeds (42, 43, 44) at 30 and at 100 epochs, each with a fresh cosine schedule, using the same sweep_gpu.py pipeline as the sweep; all 156 of 156 planned runs completed (8 diverged and are included at chance level) and the per-run outputs are released (Table 3). Across architecture means the 30- to 100-epoch Spearman correlation is 0.949 (); across the 78 paired runs it is 0.952, and the three seed-wise correlations range from 0.942 to 0.964. Accuracy rises by 6.13 points on average from 30 to 100 epochs, so the longer budget changes accuracy levels more than ordering; some deep polynomial networks learn late, so the ordering is not exact. The largest seed standard deviation at 30 epochs is 15.08 points. This job replaces an earlier three-seed validation run on a different GPU whose raw output was not retained; its numbers are used only for the pipeline comparison in Section 6.1.
| Architecture | 30 epochs | 100 epochs |
|---|---|---|
| cnn_relu_none_16_32_64 | ||
| cnn_relu_none_16_32_64_128 | ||
| cnn_relu_none_64_128_256 | ||
| cnn_x2_none_16_32_64 | ||
| cnn_x2_none_16_32_64_128 | ||
| cnn_x2px_none_16_32_64 | ||
| cnn_x2px_none_16_32_64_128 | ||
| cnn_x3px_none_16_32_64 | ||
| cnn_x3px_none_16_32_64_128 | ||
| dws_relu_none_16_32_64 | ||
| dws_relu_none_32_64_128 | ||
| dws_relu_none_32_64_128_256 | ||
| dws_x2_none_16_32_64 | ||
| dws_x2_none_32_64_128_256 | ||
| dws_x2px_none_16_32_64 | ||
| dws_x2px_none_32_64_128_256 | ||
| dws_x3px_none_16_32_64 | ||
| dws_x3px_none_32_64_128_256 | ||
| mlp_relu_layernorm_128x64x32 | ||
| mlp_relu_none_128x64x32 | ||
| mlp_x2_layernorm_128x64x32 | ||
| mlp_x2_none_128x64x32 | ||
| mlp_x2px_layernorm_128x64x32 | ||
| mlp_x2px_none_128x64x32 | ||
| mlp_x3px_layernorm_128x64x32 | ||
| mlp_x3px_none_128x64x32 |
Three-seed (42, 43, 44) test top-1 (mean sample SD) after 30 and after 100 epochs (a fresh cosine schedule for each budget) for the 26 architectures of the earlier validation, rerun with sweep_gpu.py; per-run outputs are released. Runs that diverged are included at their (chance-level) accuracy.
6.4 Zero-Cost Proxies Across Activation Families
Table 4 gives Spearman between each proxy and mean accuracy per activation family with bootstrap intervals and Holm-adjusted significance over the 25 activation-family tests it contains. NEAR is a reliable predictor only for ReLU networks (, CI ); on the polynomial families its intervals include zero ( -0.12 , +0.01, -0.14). The Zen-style score (our variant, Section 3; not the published Zen-Score) is negatively correlated with accuracy in every family in this analysis (ReLU -0.23, -0.42, -0.31, -0.54), but not robustly: with a common training recipe its ReLU correlation is +0.19 (Table 6) and its partial correlation on ReLU is +0.02. ZiCo is the one proxy with a significant positive coefficient on a polynomial family (, +0.45), but once failed networks are removed (Table 5) it is -0.38 on and -0.50 on : its apparent success on comes from separating networks that train from networks that do not, not from ranking accuracy among those that train. Removing failed networks changes other conclusions too: Holm-significant entries lost are NASWOT-sign on ; NASWOT-sign on ; NASWOT-sign on ; Zen-style on ; ZiCo on ; entries gained are Synflow on ; Synflow on ; Zen-style on ; ZiCo on ; ZiCo on . The negative correlations of the Zen-style score and of the sign-pattern NASWOT variant are therefore partly trainability effects. MeCo and Synflow are weak or inconsistent.
Two controls address confounds. Table 6 restricts to normalized networks, where every activation was trained with the same recipe, and the picture is the same (NEAR on ReLU +0.73, on -0.02, +0.06, -0.07). Table 7 removes, by rank regression, the effects of architecture type, normalization and parameter count: NEAR’s partial correlation on ReLU is +0.22 (against +0.70 unadjusted), so much of NEAR’s apparent power on ReLU networks reflects architecture type and size. We therefore do not claim that any tested proxy is a reliable accuracy predictor across activation families, and we do not attribute the failure to the activation function alone.
| Proxy | ReLU () | () | () | () |
|---|---|---|---|---|
| NEAR | [+0.57,+0.80] | [-0.29,+0.05] | [-0.17,+0.19] | [-0.30,+0.03] |
| Synflow | [+0.03,+0.40] | [-0.41,-0.04] () | [-0.14,+0.43] () | [-0.38,+0.03] () |
| ZiCo | [+0.07,+0.40] | [-0.29,+0.09] | [+0.29,+0.60] | [-0.22,+0.16] |
| MeCo | [-0.02,+0.32] | [-0.26,+0.08] () | [-0.16,+0.27] () | [-0.24,+0.14] () |
| Zen-style (variant) | [-0.42,-0.04] | [-0.54,-0.27] () | [-0.49,-0.11] () | [-0.64,-0.42] () |
| NASWOT (ReLU patterns) | [+0.44,+0.75] | n/a | n/a | n/a |
| NASWOT (pre-activation signs) | [+0.44,+0.75] | [-0.43,-0.11] | [-0.45,-0.12] | [-0.49,-0.19] |
Spearman between proxy score at initialization and mean top-1 accuracy over two training seeds (CIFAR-10, regenerated sweep), per activation family, with bootstrap CI over architectures. marks Holm-adjusted over the 25 activation-family tests in this table. A dash or “n/a” means fewer than finite scores.
| Proxy | ReLU () | () | () | () |
|---|---|---|---|---|
| NEAR | [+0.57,+0.80] | [-0.38,+0.09] | [-0.21,+0.39] | [-0.31,+0.20] |
| Synflow | [+0.03,+0.40] | [-0.64,-0.27] () | [-0.62,+0.01] () | [-0.58,-0.15] () |
| ZiCo | [+0.07,+0.40] | [-0.69,-0.26] | [-0.60,-0.12] | [-0.58,-0.12] |
| MeCo | [-0.02,+0.32] | [-0.42,+0.04] | [-0.33,+0.26] | [-0.40,+0.11] |
| Zen-style (variant) | [-0.42,-0.04] | [-0.42,+0.12] | [-0.70,-0.20] | [-0.59,-0.12] |
| NASWOT (ReLU patterns) | [+0.44,+0.75] | n/a | n/a | n/a |
| NASWOT (pre-activation signs) | [+0.44,+0.75] | [-0.46,+0.00] | [-0.51,+0.05] | [-0.47,-0.02] |
As Table 4 after excluding networks whose training diverged or ended at chance level (mean top-1 ; 0 of 111 ReLU, 50 of 121 , 73 of 121 , 47 of 121 excluded). Holm adjustment is applied within this table.
| Proxy | ReLU () | () | () | () |
|---|---|---|---|---|
| NEAR | [+0.58,+0.82] | [-0.23,+0.18] | [-0.17,+0.29] | [-0.27,+0.13] |
| Synflow | [-0.09,+0.47] | [-0.65,-0.34] () | [-0.54,+0.01] () | [-0.70,-0.40] () |
| ZiCo | [+0.08,+0.51] | [-0.30,+0.14] | [+0.22,+0.60] | [-0.25,+0.18] |
| MeCo | [-0.05,+0.38] | [-0.34,+0.10] () | [-0.10,+0.41] () | [-0.19,+0.28] () |
| Zen-style (variant) | [-0.07,+0.44] | [-0.67,-0.34] () | [-0.52,+0.07] () | [-0.65,-0.32] () |
| NASWOT (ReLU patterns) | [+0.38,+0.75] | n/a | n/a | n/a |
| NASWOT (pre-activation signs) | [+0.38,+0.75] | [-0.44,-0.03] | [-0.31,+0.14] | [-0.48,-0.12] |
As Table 4, restricted to networks with LayerNorm or GroupNorm. All activations in this subset were trained with the same recipe (lr ), removing the learning-rate difference that applies to norm-free polynomial networks (lr with clipping). Holm adjustment is applied within this table.
| Proxy | ReLU | |||
|---|---|---|---|---|
| NEAR | ||||
| Synflow | ||||
| ZiCo | ||||
| MeCo | ||||
| Zen-style (variant) | ||||
| NASWOT (ReLU patterns) | – | – | – | |
| NASWOT (pre-activation signs) |
Partial Spearman correlation of proxy and accuracy after removing, by rank regression, the effects of architecture type (MLP, CNN, DWS-CNN, ResNet-style), normalization and parameter count. : unadjusted for the partial correlation (descriptive only; the Holm-adjusted tests are in Table 4).
6.5 NASWOT
Standard NASWOT scores the outputs of nn.ReLU modules. It returns on all 363 of the 363 polynomial networks because they contain none; for the networks the degeneracy is also mathematically correct (Proposition 2), but the count does not test the proposition. To remove the confound we score the sign of each activation module’s input instead (Section 3). That variant equals the standard score on ReLU networks () and is defined for every polynomial network, yet it is negatively correlated with accuracy there ( -0.27, -0.29, -0.35; Table 4). The failure of NASWOT on polynomial networks is therefore not only a missing-module artifact.
| Family | standard | sign variant | |
|---|---|---|---|
| ReLU | 111 | 0 | 0 |
| 121 | 121 | 0 | |
| 121 | 121 | 0 | |
| 121 | 121 | 0 |
Networks with a NASWOT score: standard implementation (scores nn.ReLU outputs) versus the pre-activation-sign variant.
6.6 The Zen-Style Score
Among the 110 polynomial networks without normalization, the Zen-style score is weakly negatively correlated with accuracy (), positively correlated with depth () and with parameter count (), and the failure rate is 43% in the top Zen-style quartile against 11% in the bottom quartile (14% overall). Excluding failed networks, the correlation with accuracy is -0.19 (). Among the 39 ReLU networks without normalization it is -0.42, with parameter count +0.40. This is descriptive: it is consistent with this variant rewarding size and depth, which are not what determines accuracy here, but we do not claim a mechanism.
6.7 ezkl Circuit Size and Timings on Trained Networks
Width-matched comparison. Table 9 repeats the width-matched comparison of Section 5 on trained networks calibrated with real inputs. The result is unchanged: ReLU and networks have the same in all five unnormalized skeletons, and unnormalized networks calibrate to in every skeleton. Across all 50 ReLU/ pairs in the sweep that share a skeleton and were calibrated, 45 have the same , in 4 the network is smaller and in 1 it is larger; the median ratio of assigned cells (/ReLU) is 1.03.
Accounting. equals the calibrated for 163 of the 234 trained networks (70%): for 146/194 ReLU, and networks (ReLU 59/77, 42/55, 45/62) and 17/40 of the networks. It is not exact in either direction: the calibrated is lower than in 52 networks and higher in 19. The next subsection explains both directions.
Timings. Table 10 replaces the timings of the earlier version, whose ezkl version, settings, weight state and hardware were not recorded. It covers 7 trained seed-0 networks at 19–22 with real inputs, one ezkl version and one host. Mock proofs were repeated three times (spread at most 14% of the median) and take 2–49 s. Real proofs were run for all 7 networks, three times for and once for , and take 28–970 s. The proving times in the earlier version (up to s for a network) are not reproduced: the comparable network here proves in a fraction of that time, and we withdraw the earlier timings. The quantized circuits agree with the float models on 700 of 700 real test-image predictions. The host is shared with other jobs, so times are indicative; the load average before and after each run is in the released file.
| Skeleton | ReLU | |||
|---|---|---|---|---|
| MLP 128-64-32 | 0.404M (19) | 0.407M (19) | 0.408M (19) | 0.402M (23) |
| MLP 256-128 | 0.823M (20) | 0.824M (20) | 0.824M (20) | 0.813M (23) |
| MLP 512-256 | 1.703M (21) | 1.710M (21) | 1.711M (21) | 1.688M (23) |
| CNN 16-32 | 1.732M (21) | 1.832M (21) | 1.855M (21) | 1.879M (23) |
| CNN 16-32-64 | 2.795M (22) | 2.909M (22) | 2.938M (22) | 2.956M (23) |
Width-matched ezkl circuit size on trained seed-0 networks calibrated with real CIFAR-10 inputs: assigned cells (calibrated ). “fail”: calibration did not complete or errored (recorded in the released file).
| Architecture | top-1 (2 seeds) | mock | prove | verify | proof KB | q-agree | |
|---|---|---|---|---|---|---|---|
| mlp_x2px_none_128x64x32 | 54.5 | 19 | 2.5 [2.5,2.5] | 28.3 [28.2,28.6] | 0.20 [0.20,0.21] | 15.4 | 100/100 |
| mlp_x2px_layernorm_128x64x32 | 56.5 | 19 | 3.6 [3.5,4.0] | 57.7 [55.4,59.2] | 0.12 [0.10,0.13] | 27.2 | 100/100 |
| dws_relu_none_16_32_64 | 65.4 | 20 | 16.1 [16.0,16.5] | 203.1 [193.2,205.1] | 0.22 [0.22,0.22] | 28.9 | 100/100 |
| dws_relu_none_32_64_128 | 71.3 | 21 | 30.7 [30.2,31.0] | 360.8 [354.2,373.1] | 0.43 [0.42,0.48] | 29.0 | 100/100 |
| dws_relu_none_32_64_128_256 | 75.8 | 22 | 46.2 [45.4,49.1] | 749.9 | 1.56 | 29.0 | 100/100 |
| cnn_relu_none_16_32_64 | 77.9 | 22 | 41.0 [40.4,42.1] | 712.8 | 1.16 | 29.0 | 100/100 |
| cnn_relu_none_16_32_64_128 | 82.5 | 22 | 42.6 [41.7,44.1] | 970.1 | 2.06 | 29.0 | 100/100 |
ezkl timings on trained seed-0 networks with real CIFAR-10 inputs under one recorded configuration (ezkl 9.5.0, AMD Ryzen Threadripper PRO 3945WX 12-Cores, 24 logical cores, shared host). Median [min,max] over repeats; seconds. Mock proofs: . Real proofs: for , for . Last column: quantized-circuit predictions that agree with the float model on real test images.
6.8 Budget-Constrained Architecture Search
Protocol. The pool is the 234 calibrated trained networks; for a budget it contains those with calibrated . A method selects networks to train, and the outcome is the best mean accuracy among them. The joint proxy’s cost term is ezkl’s calibrated assigned cells, an oracle cost that is at least as informative as the model’s estimate, so the comparison favors the joint proxy. Each of 300 random splits of the pool tunes over for the joint proxy on one half (strict improvement is required to move away from a smaller , so ties go to , i.e. NEAR-only) and evaluates all methods on the other. Table 11 reports regret against the best network in the evaluation half. Networks whose seed-0 training failed are not in the pool, so no method is penalized for selecting one; 2 pool members fail in seed 1 or have mean accuracy below and are kept at their mean accuracy. This is a limitation.
Result. Under calibrated budgets the tuned joint proxy almost always reduces to NEAR-only: is chosen in 95% of tuning splits at , and Joint ties NEAR-only in almost every split at (win/tie/loss against NEAR-only: 0/300/0, 0/300/0, 0/300/0, 19/280/1 at ). Because ties favor by construction, we also report the untuned test (Table 12): the mean regret of each fixed on the evaluation halves is at its minimum at 0.0, 0.0, 0.0, 0.25 for , with regret 0.02, 0.04, 0.33, 3.61 at against 0.02, 0.04, 0.33, 2.93 at the best fixed . Table 12 gives the paired per-split differences: for the joint proxy is better than NEAR-only in at most 25% of splits and worse in at most 25%, and in 0 of 16 budget– cells does the interval exclude zero. We therefore withdraw the earlier claim that the joint proxy improves on NEAR-only. NEAR-only does beat random selection by a wide margin at every budget (regret 0.02 vs. 1.58 at , 0.04 vs. 4.18 at , 0.33 vs. 7.47 at and 3.61 vs. 8.63 at accuracy points), and ranking by cost alone is poor (1.77 to 19.35). The pools are mixed: ReLU networks are 28–33% of each, and among the highest-NEAR networks at there are 7, 7, 7 and 3 polynomial networks. Because is a large share of the smaller pools, we repeat the comparison with : regret is then 0.88, 0.39, 4.17 and 7.07 for NEAR-only against 3.43, 6.62, 11.78 and 12.91 for random selection, and the tuned joint proxy is again no better than NEAR-only (win/tie/loss 4/290/6, 0/294/6, 0/300/0, 2/266/32; slightly worse at ). Untuned at the best fixed is 0.5, 0.25, 0.25 and 0.0 at (regret 0.33, 0.19, 4.12, 7.07 against 0.88, 0.39, 4.17, 7.07 at ). A small therefore lowers regret at some budgets and sizes and not at others, and we find no consistent gain from the cost term.
Where the cost model helps. The cost model is useful when ezkl is not run. Table 13 filters by the model’s estimated : 0–21% of the networks it admits exceed the calibrated budget, and it rejects 5–25% of the networks that are feasible. Choosing high-NEAR networks with no budget violates the calibrated budget in every case, whereas choosing inside the estimated budget does so for a fraction of 0.0, 0.4, 0.1 and 0.0 of the selections at . The estimate thus removes most but not all budget violations; it does not replace calibration.
| pool | oracle | Joint | NEAR-only | ZK-cost-only | Random | J v N | |
|---|---|---|---|---|---|---|---|
| 36 | 65.4 | 0.02 [0.00,0.00] | 0.02 [0.00,0.00] | 1.77 [0.00,5.17] | 1.58 [0.44,3.15] | 0/300/0 | |
| 72 | 71.3 | 0.04 [0.00,0.00] | 0.04 [0.00,0.00] | 7.10 [0.45,13.86] | 4.18 [1.59,6.96] | 0/300/0 | |
| 111 | 82.5 | 0.33 [0.00,3.50] | 0.33 [0.00,3.50] | 19.07 [10.59,25.09] | 7.47 [4.53,11.10] | 0/300/0 | |
| 175 | 82.5 | 3.39 [0.00,7.34] | 3.61 [0.00,7.34] | 19.35 [11.50,26.02] | 8.63 [4.97,11.92] | 19/280/1 |
Budget-constrained search under ezkl-calibrated budgets ( from calibration of trained networks). Each of 300 random splits of the pool tunes for the joint proxy on one half and evaluates all methods on the other, choosing architectures to train; the outcome is the best accuracy among them. Entries are mean regret in accuracy points against the best architecture in the evaluation half, with the – percentile range over splits; Random averages draws per split. Last column: splits where Joint beats / ties / loses to NEAR-only (regret).
| pool | |||||||
|---|---|---|---|---|---|---|---|
| 36 | 0.02 | +0.00 [+0.00,+0.00] (0/0) | +0.00 [+0.00,+0.00] (0/0) | +0.18 [+0.00,+3.08] (0/14) | +0.75 [+0.00,+5.84] (0/27) | +0.97 [+0.00,+5.84] (0/31) | |
| 72 | 0.04 | +0.00 [+0.00,+0.00] (0/0) | +0.02 [+0.00,+0.00] (0/1) | +0.95 [+0.00,+5.16] (0/28) | +5.94 [+0.00,+14.54] (0/94) | +6.34 [+0.45,+14.54] (0/98) | |
| 111 | 0.33 | +0.07 [+0.00,+1.37] (0/3) | +0.78 [+0.00,+6.53] (0/25) | +5.32 [+0.00,+15.02] (0/88) | +17.66 [+10.48,+25.77] (0/99) | +17.90 [+10.59,+25.09] (0/100) | |
| 175 | 3.61 | -0.68 [-5.98,+0.00] (18/0) | -0.38 [-5.98,+3.39] (16/3) | +2.14 [+0.00,+10.91] (0/42) | +14.46 [+8.22,+22.72] (0/100) | +14.77 [+9.27,+24.40] (0/100) | |
| 36 | 0.88 | -0.27 [-3.43,+0.14] (10/3) | -0.56 [-3.55,+2.21] (25/16) | +2.05 [-3.43,+8.90] (16/59) | +2.23 [-3.43,+9.03] (20/54) | +2.61 [-3.43,+9.03] (20/57) | |
| 72 | 0.39 | -0.20 [-2.58,+0.00] (6/0) | +0.10 [-2.58,+2.87] (7/7) | +6.00 [+0.00,+15.06] (1/75) | +8.87 [+0.74,+15.06] (1/99) | +9.40 [+0.74,+15.06] (1/99) | |
| 111 | 4.17 | -0.05 [-2.44,+0.00] (3/2) | +0.41 [+0.00,+6.53] (1/10) | +7.53 [+0.00,+23.07] (0/76) | +16.78 [+9.78,+25.54] (0/100) | +17.14 [+9.78,+25.54] (0/100) | |
| 175 | 7.07 | +0.27 [-5.63,+5.75] (6/11) | +0.05 [-5.63,+5.75] (9/14) | +0.69 [-5.63,+10.40] (24/33) | +14.80 [+4.85,+20.38] (0/100) | +15.37 [+4.85,+20.38] (0/100) |
Untuned test of the joint proxy. For each fixed , the per-split paired difference in regret (accuracy points) relative to (NEAR-only) on the evaluation halves: mean, – percentile range over 300 splits, and in parentheses the percentage of splits in which it is better / worse (negative difference = better than NEAR-only). The third column is NEAR-only’s mean regret. Upper block , lower block .
| feasible | admitted | exceeds | rejected | violation (est. budget) | violation (no budget) | |
|---|---|---|---|---|---|---|
| 36 | 33 | 0.18 [0.05,0.31] | 0.25 [0.11,0.40] | 0.0 [0.0,0.2] | 1.0 | |
| 72 | 72 | 0.21 [0.12,0.30] | 0.21 [0.12,0.29] | 0.4 [0.1,0.8] | 1.0 | |
| 111 | 126 | 0.17 [0.11,0.25] | 0.05 [0.02,0.10] | 0.1 [0.0,0.3] | 1.0 | |
| 175 | 165 | 0.00 [0.00,0.00] | 0.06 [0.02,0.09] | 0.0 [0.0,0.0] | 1.0 |
Filtering by the cost model’s estimated (no ezkl run) against the calibrated budget, on the 234 calibrated trained networks. “feasible”: calibrated . “admitted”: estimated . “Exceeds”: share of admitted networks whose calibrated . “Rejected”: share of feasible networks the estimate rejects. “Violation”: share of the highest-NEAR networks, chosen inside the estimated budget (or with no budget), whose calibrated . bootstrap intervals over networks.
6.9 Cost-Model Fit and Evaluation
Table 14 fits the coefficients of Eq. Eq. (1) by non-negative least squares on 180 trained networks and evaluates on the 54 networks that remain, after the exclusion rule, from a -network validation sample drawn earlier (a stratified sample of MLP, CNN and DWS-CNN architectures; no ResNet-style networks), which are excluded from the fit. Because that evaluation split contains no ResNet-style networks while 82 of the fit networks are ResNet-style, we also report repeated -fold cross-validation over all 234 calibrated networks: the hand-set proxy matches the calibrated exactly for 68% and within one for 93% (Spearman 0.87), the fitted model for 70% and 95% (0.89), and the fitted model is exact for 100% of the 16 ResNet-style and 66% of the 66 depthwise ResNet-style networks. On the evaluation split the hand-set proxy matches the calibrated exactly for 37 and within one for 50 of 54 (Spearman 0.80, RMSE 0.19 in cells); the fitted model for 39 and 51 (0.84, 0.11). Fitted coefficients are close to one assigned cell per multiply-accumulate (standard convolution 0.94, depthwise 1.11, Gemm 0.99) and zero per matrix-multiply or convolution node, so the hand-set per-node overhead of cells is not supported by the data. Per-element costs are 14.82 (ReLU), 12.30 (power), 5.57 (normalization), 2.28 (multiplication) and 3.40 (addition), so ReLU and the polynomial power operator cost the same order of magnitude per element, consistent with Section 5. Agreement is weaker for networks (exact on 3/7 of the evaluation networks), where calibration, not cell count, sets the circuit size.
Why is not always . In ezkl’s settings the reported row count is half of the assigned cells: the ratio of assigned cells to rows lies between 1.9999 and 2.0000 in all 234 calibrated networks. The calibrated is then the largest of four resource requirements, not a function of rows alone: , where const is ezkl’s total_const_size (weights stored as constants: for MLPs it equals the parameter count, median const/parameters 1.0, while for convolutional networks it is 126–363 times larger; we did not investigate why), lookup span is the width of the calibrated lookup range and range-check span is the largest range-check table. The rule reproduces the calibrated exactly for 72 of 72 dumped networks, a random sample of up to networks per activation family among those with fewer than M assigned cells plus chosen earlier to include matching and mismatching cases; none is excluded by the exclusion rule, and their range over 19–23, so the rule is validated for small circuits only (Table 15; ReLU 19/19, 19/19, 21/21, 13/13). The resource that binds is the row count in 16 networks, the constants in 60, the lookup table in 15 and a range-check table in 12. This explains the anomaly: 12 of the 13 dumped networks have a range check of (a span of values, just above a -entry table, so it needs ; ezkl’s calibration also selects the lower input scale ) regardless of cell count (the unnormalized networks of the sweep are at 23 and, for four of them, at 24–25), and none of the 0 ReLU, or networks has such a range check, and it explains why parameter-heavy but MAC-light networks are set by their constants rather than their cells. The rule is an empirical fit to settings we dumped, not a derivation from ezkl’s source, and our cost model predicts only the cell count, which fixes the rows requirement but not the constants, lookup-table or range-check requirements. Because ezkl’s rows are half the cells, the rows requirement is in 72 of 72 dumps; is then the calibrated exactly when another requirement binds one level higher (the constants requirement equals in 53 of 72 dumps, which is why the cell-count estimate is right for most networks), and the calibrated is lower than when the rows requirement alone binds. in Eq. Eq. (1) is reported only through these residuals, and we do not claim an exact formula.
| exact | within 1 | RMSE | |||
|---|---|---|---|---|---|
| Hand-set proxy, evaluation split | 54 | 37 | 50 | 0.80 | 0.19 |
| Fitted (NNLS), evaluation split | 54 | 39 | 51 | 0.84 | 0.11 |
| Fitted (NNLS), fit split | 180 | 125 | 171 | 0.89 | 0.11 |
| fitted, eval, ReLU | 20 | 16 | 20 | 0.94 | 0.10 |
| fitted, eval, | 11 | 9 | 11 | 0.96 | 0.07 |
| fitted, eval, | 16 | 11 | 16 | 0.94 | 0.15 |
| fitted, eval, | 7 | 3 | 4 | 0.12 | 0.09 |
| 5-fold CV, all 234 networks: hand-set | 234 | 68% | 93% | 0.87 | – |
| 5-fold CV, all 234 networks: fitted | 234 | 70% | 95% | 0.89 | – |
| CV fitted, resnet | 16 | 100% | – | – | – |
| CV fitted, dws-resnet | 66 | 66% | – | – | – |
| CV fitted, mlp | 52 | 83% | – | – | – |
| CV fitted, cnn | 53 | 76% | – | – | – |
| CV fitted, dws-cnn | 47 | 44% | – | – | – |
Cost model of Eq. Eq. (1) against ezkl-calibrated on trained networks with real inputs. Coefficients are fitted by non-negative least squares (relative-error weighting) on 180 networks and evaluated on the 54 networks of the earlier validation set, which are excluded from the fit. “exact” and “within 1” count networks whose predicted equals or is within one of the calibrated ; is Spearman between predicted and calibrated ; RMSE is in of assigned cells.
Fitted coefficients (cells per MAC / per node / per element): macs_conv_standard=0.941, macs_conv_depthwise=1.11, macs_gemm=0.992, n_matmul_conv_nodes=0, elem_relu=14.8, elem_pow=12.3, elem_mul=2.28, elem_add_sub=3.4, elem_norm=5.57, elem_pool_reduce=0.
| Family | rule exact | |
|---|---|---|
| ReLU | 19 | 19 |
| 19 | 19 | |
| 21 | 21 | |
| 13 | 13 | |
| all | 72 | 72 |
Test of the four-resource rule on ezkl’s dumped settings.json for 72 trained networks (a random sample of up to per activation among networks under M assigned cells, plus chosen earlier; range 19–23 ). The binding resource (ties counted for each) was rows in 16, constants in 60, lookup table in 15 and range-check table in 12 networks.
7. Related Work
ZK-ML provers.
ezkl (zkonduit 2022) is the primary Halo2-based system (our evaluation target). Orion/Giza uses Cairo/STARK (no trusted setup, larger proofs). RISC Zero and SP1 use RISC-V zkVMs. None include NAS tooling or benchmark data. A recent survey of ZK-based verifiable machine learning (Peng et al. 2025) names limited circuit expressiveness and high proving cost among the main obstacles; architecture-level search for proving cost is one way to address the latter.
ZK circuit optimization.
TeleSparse (Maheri et al. 2025) applies post-hoc sparsification and neural teleportation (to narrow activation ranges and hence lookup tables) to reduce proof time by at roughly accuracy cost. This is complementary to our approach: TeleSparse compresses a fixed ReLU architecture after-the-fact; this work instead measures how circuit cost depends on the architecture and evaluates proxies for accuracy and cost (we do not claim a search method that improves on NEAR-only). ZKML (EuroSys 2024) and ZEN (2021) optimize compiler passes; to our knowledge none searches over architecture space. Two NAS works are close in name but not in goal: PASNet (Peng et al. 2023) searches over polynomial-versus-ReLU activations for two-party-computation private inference, where the cost model is communication and latency rather than ZK circuit size; and zkNAS (Wang et al. 2026) uses zero-knowledge proofs to let a server prove that an outsourced architecture search was run correctly, rather than searching for architectures that are cheap to prove. Closest in measurement scope, Akor et al. (Akor et al. 2026) profile eight CNN layer types (activations, pooling, normalization and linear layers) in ezkl at two precision settings over experiments, reporting proof time, proof size, circuit complexity and peak memory. Our Section 5 is complementary rather than a replacement: it compares whole networks that differ only in activation, at matched width, and relates a graph-level cost model to ezkl’s calibrated circuit size, and it feeds a NAS benchmark. We have read their abstract but have not reproduced their measurements, so we make no claim about whether our ReLU-versus- result agrees with their layer-level activation measurements.
MPC/FHE NAS.
CryptoNAS (Ghodsi et al. 2020) minimizes ReLU count for MPC (garbled circuits). In MPC, linear layers are free via secret sharing; in ZK-SNARKs they are not. In our fitted cost model multiply-accumulates cost about one assigned cell each while a ReLU element costs about 14.82 cells (Section 6), so at the widths we test the multiply-accumulate cells dominate and ReLU count would be a poor proxy for circuit size; we did not evaluate CryptoNAS itself. LinGCN (NeurIPS 2023) linearizes nonlinear layers for homomorphically encrypted inference, where multiplicative depth is the relevant cost; in our ZK setting the cost model depends on circuit size across all layers.
Training-free NAS proxies.
NEAR (Husistein et al. 2025), AZ-NAS (Lee & Ham 2024), ZiCo (Li et al. 2023), and MeCo (Jiang et al. 2023) were developed and evaluated on standard ReLU-based NAS benchmarks (e.g. NAS-Bench-201 and NATS-Bench). In our benchmark NEAR is on ReLU networks but on networks (Table 4), and NASWOT is degenerate on networks (Proposition 2). We are not aware of earlier evaluations of these proxies on ZK-cost search spaces.
NAS benchmarks.
NASBench-201 (Dong & Yang 2020) is the de facto standard, using ReLU+BatchNorm exclusively. NATS-Bench (Dong et al. 2021) likewise focuses on standard activations. Neither includes polynomial activations, BN-free designs, or ZK cost as a metric. Layer-level ezkl measurements exist (see below), but we are not aware of an architecture-level NAS benchmark that records ZK circuit cost alongside accuracy for polynomial and BN-free networks.
8. Limitations and Discussion
Scope.
All experiments use CIFAR-10 inputs, MLPs, CNNs and ResNet-style networks, one ezkl version (9.5.0), and one host. The largest calibrated circuit in the analyses has and the largest proved circuit . We make no claim about transformers, ImageNet-scale inputs, other proving backends, or hardware other than the one used. We expect a ResNet-50 on ImageNet to need on the order of ; this is an extrapolation, not a measurement.
Label quality and failures.
Accuracy is measured after 30 epochs with two seeds (median standard deviation 0.24 points; seed-pair Spearman 0.95). A three-seed, 100-epoch validation of 26 architectures (Section 6.3) gives a 30-to-100-epoch Spearman of across architecture means, supporting 30 epochs as a ranking signal, but late learning in some deep polynomial networks means the ordering is not exact. 153 networks, mostly polynomial, end at chance level (170 fail by the definition of Section 6.1); polynomial networks with normalization used the larger learning rate and fail often, so accuracy labels for polynomial networks partly measure trainability under this recipe. We report results with and without the failed networks and with a common recipe, but a different recipe could change the polynomial results.
Calibration on trained networks, and pools.
Circuit-size results use trained seed-0 weights and real test images for calibration. ezkl’s calibration depends on the network and the calibration data, and we did not study its sensitivity to the number or choice of calibration images. Networks whose seed-0 training failed and networks with estimated are not in the search pools (2 pool members fail in seed 1 or have mean accuracy below and are kept at their mean accuracy). Search results therefore describe a pool of trainable, calibratable networks (ReLU networks are 28–33% of each pool), and they do not penalize methods for selecting networks that would fail.
Timings.
Real-proof timings cover seven architectures, were run on a host shared with other jobs, and use one real proof at logrows 22 (Table 10). They are indicative; circuit size (, assigned cells) is the measure we rely on.
Reproducibility.
Per-architecture results for both seeds, calibration records with fitting features, and the scripts that generate every table are released. Trained weights (about 0.7 GB) are not.
What we do not claim.
We make no claim about the published Zen-Score (our Zen-style score is a variant that differs in four respects, Section 3). We do not claim that polynomial activations are cheaper to prove, that the joint accuracy-cost proxy improves search, that the cost model is exact, or that any tested proxy predicts accuracy across activation families.
9. Conclusion
We evaluated ZK circuit cost and zero-cost accuracy proxies against ezkl on trained CIFAR-10 networks. A parametric cell-count model fitted on 180 networks matches ezkl-calibrated exactly for 39 of 54 held-out networks and within one for 51, with weak agreement for . Width-matched, ReLU and networks have the same circuit size, which contradicts the common assumption that polynomial activations are cheaper to prove in ezkl; the largest cost differences we found come from layer type, width and ezkl’s range calibration. NEAR predicts accuracy for ReLU networks only, a Zen-style score (our variant of Zen-NAS) and a NASWOT variant are anti-correlated on polynomial networks, and under budgets set on calibrated circuit size a joint accuracy-cost proxy does not improve on NEAR-only. The cost model is most useful as a pre-filter: it removes most, not all, budget violations without running ezkl.
The zkml-bench code, per-architecture results for both seeds, calibration records, and the scripts that regenerate every table are at https://github.com/npow/zk-circuit-aware-nas (tag v2). Trained weights (about 0.7 GB) are not released.
References
-
Abdelfattah, M. S., Mehrotra, A., Dudziak, Ł., and Lane, N. D. Zero-cost proxies for lightweight nas. In International Conference on Learning Representations, 2021.
-
Dong, X. and Yang, Y. NAS-Bench-201: Extending the scope of reproducible neural architecture search. In International Conference on Learning Representations, 2020.
-
Akor, G. C., Ahakonye, L. A. C., Lee, J.-M., and Kim, D.-S. Benchmarking CNN components in EZKL: A layer-level analysis for EVM-compatible deployment. In International Conference on Artificial Intelligence in Information and Communication (ICAIIC), pages 645–650, 2026. doi:10.1109/ICAIIC68212.2026.11454315.
-
Dong, X., Liu, L., Musial, K., and Gabrys, B. NATS-Bench: Benchmarking NAS algorithms for architecture topology and size. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021.
-
Ghodsi, Z., Veldanda, A. K., Reagen, B., and Garg, S. CryptoNAS: Private inference on a relu budget. In Advances in Neural Information Processing Systems, 2020.
-
Husistein, R. T., Reiher, M., and Eckhoff, M. NEAR: A training-free pre-estimator of machine learning model performance. In International Conference on Learning Representations, 2025.
-
Jiang, T., Wang, H., and Bie, R. MeCo: Zero-shot NAS with one data and single forward pass via minimum eigenvalue of correlation. In Advances in Neural Information Processing Systems, 2023.
-
Lee, J. and Ham, B. AZ-NAS: Assembling zero-cost proxies for network architecture search. In IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2024.
-
Li, G., Yang, Y., Bhardwaj, K., and Marculescu, R. ZiCo: Zero-shot nas via inverse coefficient of variation on gradients. In International Conference on Learning Representations, 2023.
-
Lin, M., Wang, P., Sun, Z., Chen, H., Sun, X., Qian, Q., Li, H., and Jin, R. Zen-NAS: A zero-shot nas for high-performance image recognition. In International Conference on Computer Vision, 2021.
-
Mellor, J., Turner, J., Storkey, A., and Crowley, E. J. Neural architecture search without training. In International Conference on Machine Learning, 2021.
-
Maheri, M. M., Haddadi, H., and Davidson, A. TeleSparse: Practical privacy-preserving verification of deep neural networks. Proceedings on Privacy Enhancing Technologies, 2025(4):861–880; arXiv:2504.19274.
-
Peng, H., Zhou, S., Luo, Y., Xu, N., Duan, S., Ran, R., Zhao, J., Wang, C., Geng, T., Wen, W., Xu, X., and Ding, C. PASNet: Polynomial architecture search framework for two-party computation-based secure neural network deployment. In ACM/IEEE Design Automation Conference (DAC), 2023; arXiv:2306.15513.
-
Peng, Z., and others. A survey of zero-knowledge proof based verifiable machine learning. arXiv preprint arXiv:2502.18535, 2025.
-
Tanaka, H., Kunin, D., Yamins, D. L. K., and Ganguli, S. Pruning neural networks without any data by iteratively conserving synaptic flow. In Advances in Neural Information Processing Systems, 2020.
-
Wang, H., Zhang, D., Jiang, T., Wang, D., Guo, F., Wang, J., and Guo, Y. zkNAS: Secure and efficient outsourced-NAS with zero-cost proxies. Springer Nature Singapore, 2026. doi:10.1007/978-981-92-0375-8_31.
-
zkonduit. ezkl: zero-knowledge machine learning inference toolchain (software). https://github.com/zkonduit/ezkl.
Disclosure of AI use
Claude (Anthropic; Claude Code with Sonnet and Opus-class models) was used throughout this work: writing and debugging the code in the linked repository, running experiments, drafting and revising the manuscript, and simulated peer review. The human author directed the research, reviewed the results, and is responsible for the content.
Reproducibility scope: the per-architecture results for the regenerated CIFAR-10 sweep (two seeds), the ezkl calibration records on trained networks, the ezkl timing records, the multi-seed validation outputs and the scripts that generate every table are in the linked repository. Trained network weights (about 0.7 GB) are not released.