Robust Uncertainty Quantification for Self-Evolving Large Language Models via Continual Domain Pretraining
This work addresses the challenge of providing uncertainty quantification for self-evolving LLMs in continual learning settings, which is incremental but important for practical deployment.
The paper tackles the problem of ensuring statistical reliability for large language models undergoing continual domain pretraining, where existing conformal prediction methods fail due to unknown or shifting test distributions and produce overly large prediction sets. The result is an adaptive rejection and non-exchangeable conformal prediction framework that improves effectiveness and reliability in these scenarios, as demonstrated through extensive experiments.
Continual Learning (CL) is essential for enabling self-evolving large language models (LLMs) to adapt and remain effective amid rapid knowledge growth. Yet, despite its importance, little attention has been given to establishing statistical reliability guarantees for LLMs under CL, particularly in the setting of continual domain pretraining (CDP). Conformal Prediction (CP) has shown promise in offering correctness guarantees for LLMs, but it faces major challenges in CDP: testing data often stems from unknown or shifting domain distributions, under which CP may no longer provide valid guarantees. Moreover, when high coverage is required, CP can yield excessively large prediction sets for unanswerable queries, reducing informativeness. To address these challenges, we introduce an adaptive rejection and non-exchangeable CP framework. Our method first estimates the distribution of questions across domains in the test set using transformer-based clustering, then reweights or resamples the calibration data accordingly. Building on this, adaptive rejection CP allows the LLM to selectively abstain from answering when its confidence or competence shifts significantly. Extensive experiments demonstrate that our framework enhances both the effectiveness and reliability of CP under CDP scenarios. Our code is available at: https://anonymous.4open.science/r/CPCL-8C12/