ICML 2026poster0 citations

CGRiC: Compositional Risk Certification for Structured LLM Outputs

Ibne Farabi Shihab, SANJEDA AKTER, Anuj Sharma

Abstract

Large language models increasingly generate structured outputs, including citation-grounded summaries, multi-step reasoning chains, and tool-augmented responses, where correctness is inherently compositional: a single flawed claim can invalidate an otherwise accurate response. Existing certification methods treat outputs as atomic units, forcing a binary choice between unsafe acceptance and wasteful rejection. We introduce \textbf{Claim Graph Risk Control (CGRiC)}, a framework that decomposes responses into dependency graphs of verifiable claims and assigns calibrated per-claim risk bounds via information-lift statistics. By composing these bounds, CGRiC provides explicit guarantees on the probability that any incorrect claim passes verification undetected. When this composed risk exceeds a target threshold, the system triggers localized repairs rather than full abstention, preserving correct content while fixing problematic claims. Our approach explicitly models extraction noise and verifier imperfection, and exploits conditional independence structure for tighter certificates when validated. Empirically, CGRiC achieves target risk levels while reducing abstention by 31\% compared to atomic baselines across QA, summarization, and reasoning tasks.

LLMTheoryGraphs
BibTeX
@inproceedings{
shihab2026cgric,
title={{CGR}iC: Compositional Risk Certification for Structured {LLM} Outputs},
author={Ibne Farabi Shihab and Sanjeda Akter and Anuj Sharma},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=MPwD9hhJoa}
}