Selection-Safe Conformal Early Exit: Distribution-Free Coverage under Adaptive Depth
Organizations: Department of Computer Science, BRAC University · Department of Computer Science, Iowa State University
Abstract
Calibrating a conformal prediction set at each early exit does not preserve coverage once a data-dependent exit is selected. We construct one terminal path score from prefix maxima, calibrate a single split-conformal quantile, and return nested prefix sets that all contain the valid terminal set. The guarantee is selection-safe without an alpha/L correction and holds simultaneously for every post-calibration label-free stopping policy that preserves terminal-set containment. We also bound the efficiency cost of the path maximum by the empirical tail of the early-over-terminal score excess. Across eight encoder task-backbone cells, pathwise selection attains macro coverage 0.901 versus 0.814 for naive layerwise selection, with average set size 1.41 and latency fraction 0.71. On CIFAR-100 with a 24-exit ViT-L/16 (n_cal=500), first-eligible stopping at budget b=4 attains coverage 0.902, set size 2.76, and latency fraction 0.57, and the same calibration supports deployment-time budget changes without refitting. In a joint exit-count/calibration-size sweep at n=100, pathwise set size moves only from 3.02 to 3.11 as exits increase from 4 to 24, while Bonferroni degenerates to the full label space for 12 or more exits. Paired comparison against MinSE valid selection at (n,L)=(100,24) gives a set-size difference of 0.80 [0.47, 1.13]. Full-vocabulary decoder experiments remain a negative boundary, with sets of 7,162-9,620 tokens. Pathwise calibration is most useful when candidate exits are many relative to calibration size, when budgets change after calibration, and when shallow heads return actionable sets.
Figures & tables
| Method | Controlled target | Selected validity | Avg. set | Latency/full | Exit/full |
|---|---|---|---|---|---|
| Naive layerwise CP | fixed-layer coverage | ||||
| Bonferroni CP | simultaneous coverage | ||||
| EENN-AVCS ( Jazbec et al., 2024a ) | AVCS coverage | ||||
| MinSE valid selection ( Hegazy et al., 2025 ) | selected-set coverage | ||||
| FyS supervised ( Jazbec et al., 2024b ) | – risk | n/a | |||
| Pathwise conformal | selected-set coverage |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Protected object | Guarantee | Main requirement |
|---|---|---|---|
| Layerwise CP | set at fixed exit | marginal coverage per exit | exit fixed in advance |
| EENN-AVCS ( Jazbec et al., 2024a ) | nested exit sequence | anytime-valid predictive set | posterior-stability construction |
| Fast-yet-Safe ( Jazbec et al., 2024b ) | adaptive point or distribution | expected or high-probability risk | specified loss and calibrated policy |
| Valid selection ( Hegazy et al., 2025 ) | chosen conformal set | post-selection coverage | stability correction |
| Pathwise CP | returned prefix set | exact marginal coverage | exchangeability and frozen path |
| Task | Backbone | gap | RHS | observed inflation | RHS / observed | ||||
|---|---|---|---|---|---|---|---|---|---|
| SST-2 | BERT | ||||||||
| SST-2 | DistilBERT | ||||||||
| MNLI | BERT | ||||||||
| MNLI | DistilBERT | ||||||||
| RTE | BERT | ||||||||
| RTE | DistilBERT |
| Stage | Operation |
|---|---|
| Train | Fit the backbone and exit heads on training data. |
| Design | Choose exits and increasing score maps without using final calibration or test labels; a provisional budget may guide this search. |
| Calibrate | Compute and the single quantile on . |
| Execute | Update candidatewise and construct after every executed exit. |
| Stop | Return an eligible prefix set; if none is eligible, return , or when is empty. |
| Target | Guarantee | Required condition | Not implied |
|---|---|---|---|
| Selected set | marginal coverage | exchangeable terminal scores; frozen path | feature-conditional validity |
| Realized calibration | Beta or DKW lower floor | independent calibration draw; stated continuity or conservative ties | per-input coverage |
| Group shift | coverage within each fixed group | within-group exchangeability and valid quantile | arbitrary covariate shift |
| Hybrid output | error | joint point-error control and valid fallback | error when both branches use |
| PAC size | required minimum | |||
|---|---|---|---|---|
| Fraction above floor | – | |||
| Group method | easy | medium | hard | shifted marginal |
| Pooled | ||||
| Mondrian |
| coverage (mean SD) | Beta floor | DKW floor | set size | worst draw | |
|---|---|---|---|---|---|
| Shift | calibration | hard | hard cov. | shifted cov. | set | exit/full |
|---|---|---|---|---|---|---|
| MNLI label | pooled | |||||
| MNLI label | Mondrian | |||||
| BoolQ label | pooled | |||||
| BoolQ label | Mondrian | |||||
| SST-2 difficulty | pooled | |||||
| SST-2 difficulty | Mondrian |
| commit | cond. commit err. | joint point err. | fallback share | fallback cov. | joint fallback miss | overall cov. | latency/full | |
|---|---|---|---|---|---|---|---|---|
| Task | Backbone | returned path set | empty terminal | mean positive inflation |
|---|---|---|---|---|
| SST-2 | BERT | |||
| SST-2 | DistilBERT | |||
| MNLI | BERT | – | ||
| MNLI | DistilBERT | – | ||
| RTE | BERT | |||
| RTE | DistilBERT |
| Task | Backbone | familywise interval | |
|---|---|---|---|
| SST-2 | BERT | ||
| SST-2 | DistilBERT | ||
| MNLI | BERT | ||
| MNLI | DistilBERT | ||
| RTE | BERT | ||
| RTE | DistilBERT |
| Comparator | validity | latency/full | avg. set |
|---|---|---|---|
| MinSE valid selection | |||
| EENN-AVCS | |||
| Bonferroni CP |
| Task | false removal at | |||||
|---|---|---|---|---|---|---|
| SST-2 | ||||||
| MNLI | ||||||
| RTE | ||||||
| BoolQ |
| Score or schedule | coverage range | avg. set | exit/full | latency/full |
|---|---|---|---|---|
| – | ||||
| Design affine | – | |||
| Design ECDF | – | |||
| APS | – | |||
| RAPS | – | |||
| Sparse schedule | – |
| mean cov. | path floor | fraction above | 12-exit floor | ||
|---|---|---|---|---|---|
| Item | Configuration |
|---|---|
| Accelerator | NVIDIA A SXM , GB, single GPU |
| CUDA / PyTorch | / |
| Precision | FP inference |
| Batch size | |
| Maximum encoded length | tokens |
| Warm-up | complete executions per method |
| Task | Backbone | path cov. | returned set | empty terminal | naive cov. | gap | latency/full |
|---|---|---|---|---|---|---|---|
| SST-2 | BERT | ||||||
| SST-2 | DistilBERT | ||||||
| MNLI | BERT | – | |||||
| MNLI | DistilBERT | – | |||||
| RTE | BERT | ||||||
| RTE | DistilBERT |
| Task | Backbone | full-depth acc. | singleton rate | singleton acc. |
|---|---|---|---|---|
| SST-2 | BERT | |||
| SST-2 | DistilBERT | |||
| MNLI | BERT | |||
| MNLI | DistilBERT | |||
| RTE | BERT | |||
| RTE | DistilBERT |
| Policy | budget | coverage (95% CI) | avg. set (95% CI) | latency/full (95% CI) | empty-terminal |
|---|---|---|---|---|---|
| First eligible | |||||
| First eligible | |||||
| First eligible | |||||
| First eligible | |||||
| First eligible | |||||
| Design restriction |
| Pathwise | Bonferroni | EENN-AVCS | MinSE valid selection | paired MinSE Path | ||
|---|---|---|---|---|---|---|
| Dataset | independently trained path | full-depth acc. | Path | MinSE valid selection | Bonferroni | ||
|---|---|---|---|---|---|---|---|
| CIFAR-100 | ViT-B/16 | ||||||
| CIFAR-100 | ViT-L/16 | ||||||
| ImageNet-1k | ViT-B/16 | ||||||
| ImageNet-1k | ViT-L/16 |
| DistilGPT2 | Pythia-1.4B | Pythia-2.8B | |
|---|---|---|---|
| Full-depth accuracy | |||
| Selected coverage | |||
| Average exit/full | |||
| Average set size | |||
| Early-exit rate |
Conformal Prediction
Conformal prediction wraps predictive models with calibrated sets or intervals that provide statistical coverage guarantees.
See all 352 papers