PEC-CDC: A prediction error-based calibration framework for robust unsupervised deep clustering

While deep clustering has made significant strides, the inherent overconfidence of Softmax-based calibration remains a critical bottleneck. Even dedicated frameworks like Calibrated Deep Clustering (CDC) fail to fully eliminate this issue, as they continue to rely on probability-based metrics. These relative metrics are constrained by inter-class normalization, which often forces high confidence scores onto ambiguous samples where discriminative features are underdeveloped. To address this limitation, we propose PEC-CDC, a novel framework that reconfigures the CDC architecture by replacing its probability-based calibration head with a regression-based Prediction Error-based Classification (PEC) head. Our core motivation is that prediction error serves as an absolute measure of distributional support independent of other classes, making it inherently resistant to overconfidence. Specifically, we utilize a frozen Teacher-Student module to shift the model's evaluation from relative confidence probability to absolute sample-cluster distribution consistency. Qualitative analysis via t-SNE and error distribution visualizations confirms that our framework establishes superior semantic boundaries and compact cluster manifolds. Extensive experiments demonstrate that PEC-CDC achieves state-of-the-art performance; notably, on the challenging CIFAR-100 dataset, it attains an Adjusted Rand Index (ARI) of 45.61%, effectively quadrupling the performance of leading baselines such as CueCo.

Authors

Institutions

Publication Details

Journal
PLoS ONE
Published
2026-06-25
DOI
https://doi.org/10.1371/journal.pone.0351959
Primary Topic
Domain Adaptation and Few-Shot Learning
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

PEC-CDC: A prediction error-based calibration framework for robust unsupervised deep clustering

Ziyang Li
PLoS ONE
Domain Adaptation and Few-Shot Learning
article

PEC-CDC: A prediction error-based calibration framework for robust unsupervised deep clustering

Ziyang Li
article en

Abstract

While deep clustering has made significant strides, the inherent overconfidence of Softmax-based calibration remains a critical bottleneck. Even dedicated frameworks like Calibrated Deep Clustering (CDC) fail to fully eliminate this issue, as they continue to rely on probability-based metrics. These relative metrics are constrained by inter-class normalization, which often forces high confidence scores onto ambiguous samples where discriminative features are underdeveloped. To address this limitation, we propose PEC-CDC, a novel framework that reconfigures the CDC architecture by replacing its probability-based calibration head with a regression-based Prediction Error-based Classification (PEC) head. Our core motivation is that prediction error serves as an absolute measure of distributional support independent of other classes, making it inherently resistant to overconfidence. Specifically, we utilize a frozen Teacher-Student module to shift the model's evaluation from relative confidence probability to absolute sample-cluster distribution consistency. Qualitative analysis via t-SNE and error distribution visualizations confirms that our framework establishes superior semantic boundaries and compact cluster manifolds. Extensive experiments demonstrate that PEC-CDC achieves state-of-the-art performance; notably, on the challenging CIFAR-100 dataset, it attains an Adjusted Rand Index (ARI) of 45.61%, effectively quadrupling the performance of leading baselines such as CueCo.

PLoS ONEVol. 21(6)
Northeast Agricultural University (CN)
Reduced inequalities
Openalex Percentile: Top 4%
Domain Adaptation and Few-Shot Learning
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.