Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
Stochastic Grouping Conformal Prediction (SGCP) addresses the subgroup coverage disparities inherent in standard conformal prediction by learning a stochastic grouping map that allows samples to share calibration information based on similar calibration behavior. This approach yields a local score law, improving reliability across subpopulations without requiring explicit subgroup attributes, thus avoiding the worst-group bottleneck. Empirical results demonstrate SGCP's ability to significantly reduce subgroup coverage gaps while maintaining or improving prediction set efficiency compared to existing methods.
Conformal prediction can now achieve subgroup reliability without explicit group labels or sacrificing efficiency by learning a stochastic grouping map that pools calibration information locally.
Conformal prediction offers a distribution-free coverage guarantee, making it especially attractive for clinical applications. Standard conformal prediction, however, provides such guarantees only at the population level, and its prediction sets can exhibit coverage disparities across clinically important subgroups. A natural remedy is to calibrate within predefined groups. However, this can require access to sensitive subgroup attributes and is prone to a worst-group bottleneck: protecting the most difficult subgroup can inflate prediction sets for all, increasing cognitive burden on decision makers. To this end, we propose Stochastic Grouping Conformal Prediction (SGCP), a conformal framework for subgroup-reliable uncertainty quantification. It learns a stochastic grouping map that allows each sample to draw calibration information from others with similar calibration behavior, yielding a local score law that boosts reliability across subpopulations. We prove that SGCP retains the standard coverage guarantee. Experiments on synthetic and real-world benchmarks show that it consistently reduces subgroup coverage gaps while achieving smaller or comparable prediction set sizes relative to existing baselines.