ABLE: Representing and Mapping LLMs via Attribution-Based Large-model Embedding
About
The explosive growth of large language models (LLMs) has created a heterogeneous and poorly documented ecosystem, making systematic model comparison increasingly important for provenance auditing, security analysis, and model selection. Existing representation methods struggle to address this setting efficiently. Approaches analyzing internal parameters are powerful when architectures are compatible, but face scalability barriers under structural heterogeneity, while methods relying on external outputs may conflate models with similar behaviors and are difficult to align in richer output spaces across different tokenizers. To bridge this gap, we propose ABLE (Attribution-Based Large-model Embedding), a framework that leverages the interpretability space to construct model representations. By aggregating gradient-based feature attributions via a tokenizer-agnostic word-level alignment, ABLE captures model-specific input-sensitivity patterns rather than only surface-level outputs. Beyond empirical utility, we provide a stability analysis showing that, under standard regularity assumptions for differentiable Transformer-style models, ABLE induces a Lipschitz-continuous parameter-to-embedding map with finite-sample convergence guarantees. Extensive experiments on 239 open-source LLMs demonstrate that our training-free approach achieves competitive or superior performance in relation prediction, model routing, and benchmark score prediction.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Ranking correlation with full dataset evaluation | WinoGrande | Kendall Correlation0.626 | 13 | |
| Model Relation Prediction | Model Relation Prediction Dataset 135 pairs | Accuracy86.7 | 5 | |
| Model Routing | EmbedLLM benchmark | Router Accuracy67.6 | 4 | |
| Benchmark Ranking Prediction | ARC | Kendall's Tau0.664 | 3 | |
| Benchmark Ranking Prediction | HellaSwag | Kendall's Tau0.649 | 3 | |
| Benchmark Ranking Prediction | MMLU | Kendall's Tau Correlation0.655 | 3 | |
| Benchmark Ranking Prediction | TruthfulQA | Kendall's Tau Correlation0.605 | 3 | |
| Benchmark Ranking Prediction | GSM8K | Kendall's Tau Correlation0.602 | 3 |