English

Capturing LLM Capabilities via Evidence-Calibrated Query Clustering

Artificial Intelligence 2026-05-19 v1 Machine Learning

Abstract

Query clustering organizes queries into groups that reflect shared latent capability demands, enabling capability-aware LLM evaluation. Existing clustering methods, which primarily rely on semantic taxonomies or embeddings, often fail to capture such latent capability requirements due to a misalignment between surface-level semantics and actual model performance. We propose ECC, an algorithm that calibrates prior semantic embeddings using limited posterior model comparisons to bridge the gap between surface-level semantics and latent capability requirements. ECC characterizes each cluster through a capability profile parameterized by a Bradley-Terry model and uses trainable mixture weights to accommodate queries with mixed capability demands, jointly learning a flexible, capability-aware clustering structure that supports query-specific inference of LLM capabilities. Extensive quantitative and qualitative evaluations demonstrate that ECC significantly improves LLM capability ranking quality, outperforming human-labeled and embedding-based baselines by an average of 17.64 and 18.02 percentage points, respectively, and proves effective in downstream tasks such as query routing.

Keywords

Cite

@article{arxiv.2605.17110,
  title  = {Capturing LLM Capabilities via Evidence-Calibrated Query Clustering},
  author = {Fangzhou Wu and Sandeep Silwal and Qiuyi Zhang},
  journal= {arXiv preprint arXiv:2605.17110},
  year   = {2026}
}

Comments

45 pages