DRATS: Adaptive Task Sampling Boosts Worst-Task Performance in Multi-Task RL

Cohere · youtube · 2026-10-23

In a Cohere Labs talk, UW-Madison PhD Nicholas Edward Corrado presented DRATS, which tackles imbalanced data allocation in multi-task RL. Formalizing MTRL as a feasibility problem yields a minimax objective on worst-case return gap; the algorithm adaptively prioritizes tasks furthest from solved. On MetaWorld-MT10/MT50 it improves data efficiency and worst-task performance vs existing task samplers. Builds on AutoMixAlign (ACL 2025), a DPO-based adaptive data mixing method developed with Amazon's Rufus team.

Original post →

More from Research

Research channel →