mlsanityData to 7 Oct 2026

PASC: Pipeline-Aware Conformal Prediction with Joint Coverage Guarantees for Multi-Stage NLP and LLM Pipelines

arXiv 2605.18812 ↗First posted 12 May 2026cs.LG cs.CL cs.IRcited by 7

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

  1. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  2. 2017On Calibration of Modern Neural Networkscited by 4,860
  3. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  4. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
  5. 2016Distribution-Free Predictive Inference For Regressioncited by 657
  6. 2016Neural Architectures for Named Entity Recognitioncited by 548
  7. 2019Conformal Prediction Under Covariate Shiftcited by 511
  8. 2022Conformal prediction beyond exchangeabilitycited by 348
  9. 2023Robots That Ask For Help: Uncertainty Alignment for Large Language Model Plannerscited by 298
  10. 2022Confident Adaptive Language Modelingcited by 297
  11. 2022Conformal Risk Controlcited by 290
  12. 2015A Review of Relational Machine Learning for Knowledge Graphscited by 264
12 more
  1. 2021Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Controlcited by 256
  2. 2023Conformal Language Modelingcited by 163
  3. 2023Conformal Prediction With Conditional Guaranteescited by 146
  4. 2024Language Models with Conformal Factuality Guaranteescited by 128
  5. 2024Theoretical Foundations of Conformal Predictioncited by 96
  6. 2022Selection by Prediction with Conformal p-valuescited by 66
  7. 2024Mitigating LLM Hallucinations via Conformal Abstentioncited by 56
  8. 2024Conformal Prediction for Natural Language Processing: A Surveycited by 52
  9. 2021PAC Prediction Sets Under Covariate Shiftcited by 42
  10. 2022Conformal Prediction Sets with Limited False Positivescited by 28
  11. 2024Retrieval Augmented Generation for Domain-specific Question Answeringcited by 22
  12. 2026PromptPort: A Reliability Layer for Cross-Model Structured Extractioncited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 7.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025TECP: Token-Entropy Conformal Prediction for LLMs
  2. 2024ConU: Conformal Uncertainty in Large Language Models with Correctness Coverage Guarantees
  3. 2026Beyond Surface Statistics: Robust Conformal Prediction for LLMs via Internal Representations
  4. 2025Robust Uncertainty Quantification for Self-Evolving Large Language Models via Continual Domain Pretraining
  5. 2024Learning from Natural Language Explanations for Generalizable Entity Matching
  6. 2026Differentiable Conformal Training for LLM Reasoning Factuality
  7. 2026AdaMTP: An Adaptive Training Paradigm for Multi-Token Prediction
  8. 2024Improving Self Consistency in LLMs through Probabilistic Tokenization