Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

RIMRULE: Improving Tool-Using Language Agents via MDL-Guided Rule Learning

About

Large language models (LLMs) often struggle to use tools reliably in domain-specific settings, where APIs may be idiosyncratic, under-documented, or tailored to private workflows. This highlights the need for effective adaptation to task-specific tools. We propose RIMRULE, a neuro-symbolic approach for LLM adaptation based on dynamic rule injection. Compact, interpretable rules are distilled from failure traces and injected into the prompt during inference to improve task performance. These rules are proposed by the LLM itself and consolidated using a Minimum Description Length (MDL) objective that favors generality and conciseness. Each rule is stored in both natural language and a structured symbolic form, supporting efficient retrieval at inference time. Experiments on tool-use benchmarks show that this approach improves accuracy on both seen and unseen tools without modifying LLM weights. It outperforms prompting-based adaptation methods and complements finetuning. Moreover, rules learned from one LLM can be reused to improve others, including long reasoning LLMs, highlighting the portability of symbolic knowledge across architectures.

Xiang Gao, Yuguang Yao, Qi Zhang, Kaiwen Dong, Avinash Baidya, Ruocheng Guo, Hilaf Hasson, Kamalika Das• 2025

Related benchmarks

TaskDatasetResultRank
Function CallingBFCL rand (test)
Accuracy56.6
4
Function CallingBFCL unseen (test)
Accuracy48.5
4
Multi-hop tool useToolHop rand (test)
Accuracy31.1
4
Multi-hop tool useToolHop unseen (test)
Accuracy43.1
4
Showing 4 of 4 rows

Other info

Follow for update