cs.AIAug 20, 2026

Learning What to Activate: Combinatorial Capability Allocation for Long-Horizon Multimodal Agents

Authors: Wenhao Yuan, Chenchen Lin, Jian Chen, Jinfeng Xu, Shuo Yang, Edith Cheuk-Han Ngai

Organizations: Department of Electrical and Computer Engineering, The University of Hong Kong

Abstract

Long-horizon multimodal agents rely on specialized capabilities for perception, retrieval, reasoning, verification, and execution. Existing designs typically activate a fixed capability set or invoke a predefined workflow, incurring substantial computational overhead while failing to accommodate stage-dependent capability demands. In this paper, we study the combinatorial capability allocation problem for long-horizon multimodal agent systems, where the system selects a cost-sensitive subset of specialized capabilities at each interaction stage, which is nontrivial since capability values depend on the selected subset, while previous allocations alter the states encountered by subsequent decisions. We introduce CoCA, an on-policy learning framework that recovers a deployable capability-subset policy from sparse conditional comparisons. On states visited by the student policy, the stronger teacher compares the marginal net values of candidate capabilities, conditioned on the currently selected subset. Then, we adopt a conditional utility model to transform such comparisons into an autoregressive capability-subset policy, avoiding explicit enumeration. We further introduce dual-level on-policy distillation to address distribution mismatch both across environment states and within the partial subsets encountered during set construction. Finally, trajectory-level reinforcement learning refines the distilled policy toward task success, activation cost, and allocation stability. At inference time, allocation is performed solely by the lightweight student policy without teacher queries or online updates. Experiments on long-horizon multimodal environments and controlled capability-demand shifts demonstrate the superiority of our method over the state-of-the-art baseline methods.

Figures & tables

Appendix figures & tables3 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Beyond Simply Environment Scaling: Designing Effective Environment Distributions for Multimodal Agent Learning

    Aug 4, 2026Kejian Zhu, Zhuoran Jin, Dongqi Huang +4Curriculum LearningMultimodal Agents

  2. History Matters: Meta-policy Delegation with Heterogeneous Multi-agent Reinforcement Learning

    Aug 4, 2026Ziqing Lu, Avinash Reddy Mudireddy, Sarra Alqahtani +1Multi-Agent SystemsCooperative MARL