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

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.

Junyi Li, Xiaowei Qian, Yingyi Zhang, Wenlin Zhang, Guojing Li, Sheng Zhang, Xiao Han, Yichao Wang, Xiangyu Zhao• 2026

Related benchmarks

TaskDatasetResultRank
Multi-hop QANQ (test)
EM50.1
14
Multi-hop QAHotpotQA (test)
Exact Match (EM)45.2
14
Mathematical ReasoningMATH 500
EM (Accuracy)84.6
12
Mathematical ReasoningAIME25
EM Accuracy33.3
12
Mathematical ReasoningOlympiad
EM Accuracy52.2
12
Mathematical ReasoningAMC23
EM Accuracy77.5
12
Mathematical ReasoningAIME24
EM Accuracy36.7
12
Showing 7 of 7 rows

Other info

Follow for update