Towards Pareto-Optimal Tool-Integrated Agents with Pareto Ranking Policy Optimization
About
Recent advances in tool-integrated language agents have significantly improved their ability to solve complex reasoning tasks. However, existing alignment methods predominantly focus on maximizing task accuracy, while overlooking auxiliary objectives such as tool-use efficiency, which are essential for practical deployment. To address this gap, we introduce ParetoPO, a two-stage multi-objective optimization framework for aligning tool-using large language models (LLMs) under competing objectives. In the first stage, ParetoPO leverages hypervolume-guided dynamic scalarization to adapt reward weights based on global Pareto frontier progress. In the second stage, it replaces scalarized learning signals with Pareto-ranking-based advantage computation, promoting nondominated trajectories through dominance-aware credit assignment. This design enables fine-grained, action-level optimization across multiple conflicting objectives. Experimental results on mathematic reasoning and multi-hop QA tasks show that ParetoPO consistently discovers policies with superior accuracy-efficiency trade-offs compared to static and heuristic baselines.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-hop QA | NQ (test) | EM50.1 | 14 | |
| Multi-hop QA | HotpotQA (test) | Exact Match (EM)45.2 | 14 | |
| Mathematical Reasoning | MATH 500 | EM (Accuracy)84.6 | 12 | |
| Mathematical Reasoning | AIME25 | EM Accuracy33.3 | 12 | |
| Mathematical Reasoning | Olympiad | EM Accuracy52.2 | 12 | |
| Mathematical Reasoning | AMC23 | EM Accuracy77.5 | 12 | |
| Mathematical Reasoning | AIME24 | EM Accuracy36.7 | 12 |