Generating High-Quality Prediction Intervals for Regression Tasks via Fuzzy C-Means Clustering-Based Conformal Prediction

Msaddi S., Kumbasar T.

Intelligent and Fuzzy Systems - Intelligence and Sustainable Future Proceedings of the INFUS 2023 Conference, İstanbul, Turkey, 22 - 24 August 2023, vol.759 LNNS, pp.532-539 identifier

  • Publication Type: Conference Paper / Full Text
  • Volume: 759 LNNS
  • Doi Number: 10.1007/978-3-031-39777-6_63
  • City: İstanbul
  • Country: Turkey
  • Page Numbers: pp.532-539
  • Keywords: Conformal Prediction, Fuzzy Clustering, Uncertainty Quantification
  • Istanbul Technical University Affiliated: Yes


Accurately assessing uncertainty and prediction of a regression model is essential for making informed decisions, especially in high-risk tasks. Conformal Prediction (CP) is a powerful distribution-free uncertainty quantification framework for building such models as it is capable to transform a single-point prediction of any machine learning model into a Prediction Interval (PI) with a guarantee of encompassing the true value for specified levels of confidence. On the other hand, to generate high-quality PIs, the PIs should be as narrow as possible while enveloping a certain amount of uncertainty (i.e. confidence level). The generated width of the PIs mainly depends on the nonconformity measure used within the CP. In this study, we propose two novel Fuzzy c-Means Clustering (FCM) based nonconformity measures for CP with nearest neighbors to learn distribution-free and high-quality PIs for regression. The proposed approach generates tight PIs by evaluating the degree of nonconformity of a new data point compared to the so-called calibration points via Fuzzy Sets (FSs). From the calibration dataset, we extract representative FSs via FCM and assign every test point alongside the nearest neighbors within the calibration dataset with membership grades to adapt the nonconformity measure. To evaluate the performance, we present statistical comparisons and demonstrate that the proposed FCM-based nonconformity measures result in high-quality PIs.