Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Zirui Wang, Yusen Hou, Shaofeng Liang, Bowen Tian, Yanlin Zhang, Wenshuo Chen, Yutao Yue• 2026

Related benchmarks

TaskDatasetResultRank
Ranking correlation with full dataset evaluationWinoGrande
Kendall Correlation0.626
13
Model Relation PredictionModel Relation Prediction Dataset 135 pairs
Accuracy86.7
5
Model RoutingEmbedLLM benchmark
Router Accuracy67.6
4
Benchmark Ranking PredictionARC
Kendall's Tau0.664
3
Benchmark Ranking PredictionHellaSwag
Kendall's Tau0.649
3
Benchmark Ranking PredictionMMLU
Kendall's Tau Correlation0.655
3
Benchmark Ranking PredictionTruthfulQA
Kendall's Tau Correlation0.605
3
Benchmark Ranking PredictionGSM8K
Kendall's Tau Correlation0.602
3
Showing 8 of 8 rows

Other info

Follow for update