mlsanityData to 7 Oct 2026

Why Sample What You Can Enumerate? Exact Policy Optimization for Genomic Tool Selection

arXiv 2609.10221 ↗First posted 9 Sep 2026cs.AIcited by 0

On the map: LLM reasoning → Reinforcement learning for reasoning

Abstract (arXiv)

Reinforcement learning over a frozen reasoner has become a common recipe for teaching a policy which external tools to invoke. We show that this recipe becomes structurally mismatched in specialist scientific settings where the complete tool-subset space is enumerable. There, a small set of recurring computational capabilities covers the domain, so the space of tool subsets is combinatorial yet small enough to enumerate, and GRPO still estimates an action expectation from a handful of sampled rollouts. Worse, the approximation degrades as training succeeds: as the policy concentrates on preferred subsets it resamples them, sampled rewards collide, and the group-normalized advantage vanishes. On genomic reasoning the fraction of questions yielding no reward signal rises from 0.2% under a uniform reference policy to 20.8% after GRPO training. As a remedy, we introduce FGPO (Full-Group Policy Optimization), which (1) scores every tool subset and optimizes the exact action expectation, so each update sees the complete action space, and (2) precomputes the reward of each question--subset pair into an exhaustive table, removing frozen-reasoner calls from the training loop entirely. Across five frozen reasoners and three genomic benchmarks, FGPO outperforms GRPO in all 15 settings by 6.75 points on average and up to 14.20, while a standard on-demand GRPO schedule would require 2.4 times as many frozen-reasoner reward evaluations and, on GenomeQA, FGPO cuts invoked tools per question from 2.36 to 1.40.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 26 of its 42 references that are in the map. First posted on arXiv 9 Sep 2026.

Builds on

26 of 42 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  2. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar

Other references · 24

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  4. 2025Qwen3 Technical Reportcited by 8,195
  5. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  6. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  7. 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIscited by 1,880
  8. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  9. 2023Gorilla: Large Language Model Connected with Massive APIscited by 1,302
  10. 2023HuggingGPT: Solving AI Tasks with ChatGPT and its Friends in Hugging Facecited by 1,173
  11. 2023API-Bank: A Comprehensive Benchmark for Tool-Augmented LLMscited by 574
  12. 2024InternLM2 Technical Reportcited by 391
12 more
  1. 2023Chameleon: Plug-and-Play Compositional Reasoning with Large Language Modelscited by 352
  2. 2025ToolRL: Reward is All Tool Learning Needscited by 342
  3. 2023HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolutioncited by 187
  4. 2023GeneGPT: Augmenting Large Language Models with Domain Tools for Improved Access to Biomedical Informationcited by 115
  5. 2024Caduceus: Bi-Directional Equivariant Long-Range DNA Sequence Modelingcited by 108
  6. 2025VisualToolAgent (VisTA): A Reinforcement Learning Framework for Visual Tool Selectioncited by 34
  7. 2018Expected Policy Gradients for Reinforcement Learningcited by 28
  8. 2017Expected Policy Gradientscited by 14
  9. 2017Mean Actor Criticcited by 8
  10. 2026How Many Tools Should an LLM Agent See? A Chance-Corrected Answercited by 7
  11. 2026AuTAgent: A Reinforcement Learning Framework for Tool-Augmented Audio Reasoningcited by 6
  12. 2026GenomeQA: Benchmarking General Large Language Models for Genome Sequence Understandingcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025ESPO: Entropy Importance Sampling Policy Optimization
  2. 2025Sample More to Think Less: Group Filtered Policy Optimization for Concise Reasoning
  3. 2025Tool-Augmented Policy Optimization: Synergizing Reasoning and Adaptive Tool Use with Reinforcement Learning
  4. 2026Active-GRPO: Adaptive Imitation and Self-Improving Reasoning for Molecular Optimization
  5. 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce
  6. 2026Learning Generative Selection for Best-of-N
  7. 2026One Policy Is Enough: Single-Agent Reinforcement Learning Outperforms Tree Search for Chemistry Tool Learning
  8. 2025MiGrATe: Mixed-Policy GRPO for Adaptation at Test-Time