PASC: Pipeline-Aware Conformal Prediction with Joint Coverage Guarantees for Multi-Stage NLP and LLM Pipelines
On the map: Retrieval-augmented generation → Generation and context integration
Abstract (arXiv)
Modern NLP and LLM systems are pipelines: named entity recognition (NER) -> entity disambiguation (NED) -> entity typing, retrieval-augmented generation (retriever -> reader), and agentic chains of planner -> tool -> critic. Errors compound across stages, but existing uncertainty quantification methods either calibrate each stage independently (no joint coverage) or apply a Bonferroni union bound (joint coverage, but conservative). We present PASC (Pipeline-Aware Split Conformal), which reduces multi-stage joint coverage to a single scalar conformal prediction problem on the joint maximum nonconformity score. PASC provides a finite-sample distribution-free guarantee that all K stages are simultaneously covered with probability at least 1 - alpha, and is nearly tight up to a 1/(n+1) factor. On a three-stage NER -> NED -> entity-typing pipeline over CoNLL-2003, PASC achieves 96.4% end-to-end coverage versus 93.4% for Bonferroni and 86.5% for independent CP, at identical average prediction set size (1.083). Under distribution shift to WNUT-17 Twitter and WikiNEuRal Wikipedia data, PASC empirically maintains the target coverage in the tested shift settings while independent CP collapses to 59%. PASC requires a single quantile computation, runs 1.7x faster than Bonferroni, and scales to K = 6 stages where independent CP drops to 0.53 end-to-end coverage. The same joint-maximum-score reduction applies directly to compound LLM systems and agent pipelines.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Placed in Generation and context integration, one of 15 directions in Retrieval-augmented generation. Builds on 24 of its 44 references that are in the map. First posted on arXiv 12 May 2026.
Builds on
24 of 44 references.
References · 24
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2016Distribution-Free Predictive Inference For Regressioncited by 657
- 2016Neural Architectures for Named Entity Recognitioncited by 548
- 2019Conformal Prediction Under Covariate Shiftcited by 511
- 2022Conformal prediction beyond exchangeabilitycited by 348
- 2023Robots That Ask For Help: Uncertainty Alignment for Large Language Model Plannerscited by 298
- 2022Confident Adaptive Language Modelingcited by 297
- 2022Conformal Risk Controlcited by 290
- 2015A Review of Relational Machine Learning for Knowledge Graphscited by 264
12 more
- 2021Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Controlcited by 256
- 2023Conformal Language Modelingcited by 163
- 2023Conformal Prediction With Conditional Guaranteescited by 146
- 2024Language Models with Conformal Factuality Guaranteescited by 128
- 2024Theoretical Foundations of Conformal Predictioncited by 96
- 2022Selection by Prediction with Conformal p-valuescited by 66
- 2024Mitigating LLM Hallucinations via Conformal Abstentioncited by 56
- 2024Conformal Prediction for Natural Language Processing: A Surveycited by 52
- 2021PAC Prediction Sets Under Covariate Shiftcited by 42
- 2022Conformal Prediction Sets with Limited False Positivescited by 28
- 2024Retrieval Augmented Generation for Domain-specific Question Answeringcited by 22
- 2026PromptPort: A Reliability Layer for Cross-Model Structured Extractioncited by 3
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026Autonomous Research Agents: A Survey of AI Scientists and the Verification Gapcited by 6
- 2026When Can Conformal Risk Control Certify LLM Outputs? Bounds, Impossibility, and Adaptation for Structured Generationcited by 4
- 2026Post-Hoc Trajectory-Risk Certification for Modular LLM-Based Security Agentscited by 1
- 2026Mitigating Fabrication in Multi-Stage LLM Pipelines for Hiring: An Empirical Evaluation of Prompt Guardrails and Human-in-the-Loop Checkpointscited by 1
- 2026Spawn Freely, Act Sparingly: Progressive Risk Vesting for Recursive LLM-Agent Treescited by 1
- 2026The Exceedance Design Effect: Effective Sample Size for Thresholds under Clusteringcited by 0
- 2026Uncertainty Quantification for LLM Agents: A Taxonomy, an Evaluation Protocol, and an Empirical Studycited by 0
- 2026Uncertainty Quantification for LLM Agents: A Taxonomy, an Evaluation Protocol, and an Empirical Study7 Sep 2026
- 2026Spawn Freely, Act Sparingly: Progressive Risk Vesting for Recursive LLM-Agent Trees1 Sep 2026
- 2026The Exceedance Design Effect: Effective Sample Size for Thresholds under Clustering21 Aug 2026
- 2026Post-Hoc Trajectory-Risk Certification for Modular LLM-Based Security Agents4 Aug 2026
- 2026Mitigating Fabrication in Multi-Stage LLM Pipelines for Hiring: An Empirical Evaluation of Prompt Guardrails and Human-in-the-Loop Checkpoints22 Jul 2026
- 2026Autonomous Research Agents: A Survey of AI Scientists and the Verification Gap29 Jun 2026
- 2026When Can Conformal Risk Control Certify LLM Outputs? Bounds, Impossibility, and Adaptation for Structured Generation27 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2025TECP: Token-Entropy Conformal Prediction for LLMs
- 2024ConU: Conformal Uncertainty in Large Language Models with Correctness Coverage Guarantees
- 2026Beyond Surface Statistics: Robust Conformal Prediction for LLMs via Internal Representations
- 2025Robust Uncertainty Quantification for Self-Evolving Large Language Models via Continual Domain Pretraining
- 2024Learning from Natural Language Explanations for Generalizable Entity Matching
- 2026Differentiable Conformal Training for LLM Reasoning Factuality
- 2026AdaMTP: An Adaptive Training Paradigm for Multi-Token Prediction
- 2024Improving Self Consistency in LLMs through Probabilistic Tokenization