ProbeLLM: Automating Principled Diagnosis of LLM Failures
About
Understanding how and why large language models (LLMs) fail is becoming a central challenge as models rapidly evolve and static evaluations fall behind. While automated probing has been enabled by dynamic test generation, existing approaches often discover isolated failure cases, lack principled control over exploration, and provide limited insight into the underlying structure of model weaknesses. We propose ProbeLLM, a benchmark-agnostic automated probing framework that elevates weakness discovery from individual failures to structured failure modes. ProbeLLM formulates probing as a hierarchical Monte Carlo Tree Search, explicitly allocating limited probing budgets between global exploration of new failure regions and local refinement of recurring error patterns. By restricting probing to verifiable test cases and leveraging tool-augmented generation and verification, ProbeLLM grounds failure discovery in reliable evidence. Discovered failures are further consolidated into interpretable failure modes via failure-aware embeddings and boundary-aware induction. Across diverse benchmarks and LLMs, ProbeLLM reveals substantially broader, cleaner, and more fine-grained failure landscapes than static benchmarks and prior automated methods, supporting a shift from case-centric evaluation toward principled weakness discovery.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Code Generation | MBPP | MA0.68 | 12 | |
| Language Understanding | SuperBLUE | MA.0.61 | 12 | |
| Multitask Language Understanding | MMLU | Mean Accuracy (MA)73 | 12 | |
| Truthfulness | TruthfulQA | MA.81 | 12 | |
| Failure Diagnosis | MMLU | Macro Similarity Type34 | 8 | |
| Failure Diagnosis | SuperGLUE | Macro Similarity Score36 | 8 | |
| Failure Diagnosis | MBPP | Macro Similarity Score50 | 8 | |
| Failure Diagnosis | TruthfulQA | Macro Similarity Type38 | 8 | |
| Automated Probing | HellaSwag | Error Rate75 | 3 | |
| Automated Probing | MBPP | Error Rate (%)0.9 | 3 |