NeFut Logo NeFut
Admin Login

[CS.AI] Turn-level Multiscale Density Ratio Estimation for LLM Agents

Published at: 2026-09-16 22:00 Last updated: 2026-09-18 00:46
#AI #Machine Learning #LLM

With the rapid progress of large language models (LLMs), agent systems powered by LLMs show great promise for tackling complex tasks that require multi‑step reasoning or tool interaction. To fully exploit LLMs within a well‑designed agent paradigm, post‑training across diverse agent scenarios is often needed to boost performance.\

Among various post‑training techniques, alignment methods such as PPO, DPO, DIL and GRPO have become popular because they can penalize negative samples while keeping training complexity manageable. However, most of these methods focus on simple single‑turn tasks, leaving room for improvement on complex multi‑turn problems.\

We introduce Turn‑level Multiscale Density Ratio Estimation (tlm‑DRE). The key idea is to assign distinct weights to each turn and perform asymmetric token‑level training based on the gap between positive and negative sample spaces across multiple turns. Specifically:\

We evaluate tlm‑DRE on a suite of public agent benchmarks, covering tool use, dialogue planning, code generation and more. The results show that tlm‑DRE matches or slightly outperforms traditional alignment methods overall, and it excels in multi‑turn reasoning scenarios, maintaining high success rates both in‑domain and out‑of‑domain.\

The advantages of the approach are:\

  1. Turn‑level weighting captures the varying importance of information across long sequences;\
  2. Asymmetric token‑level training amplifies the signal gap between positive and negative samples;\
  3. Training overhead is comparable to existing alignment methods, making it easy to plug into current LLM training pipelines.\

Review: tlm‑DRE provides a fine‑grained alignment mechanism for multi‑turn tasks, addressing the gap left by single‑turn alignment techniques. Its promising results merit further exploration across a broader range of LLM‑driven agent systems.

Original Source: https://arxiv.org/abs/2609.16760

[h] Back to Home