Daily AI Paper

Daily AI Paper

2026-10-11

Archived
2026-10-11

A Balanced Data Diet: Addressing the Exploration Bottleneck in Mega-Scale RL for Robot Control

Octi Zhang, Mateo Guaman Castro, Patrick Yin, Ignacio Dagnino, Abhishek Gupta, Rosario Scalise, et al.

arXiv:2610.12465v1

General-purpose robots need to handle a huge range of behaviors. A legged robot may have to cross irregular terrain, while an arm may need to grasp, align, insert, or thread parts together with very little margin for error. This paper is about how to train those kinds of robots with reinforcement learning in simulation, and specifically how to make large-scale simulation actually pay off when the task is hard. The basic promise of sim-to-real reinforcement learning is attractive: train in a physics simulator, then transfer the learned policy to the real robot. Over the last several years, this has produced impressive results in locomotion and manipulation. But there is a catch. In practice, these systems often depend on a lot of task-specific engineering. You need carefully shaped rewards, hand-designed curricula, demonstrations, or other priors that make exploration easier. That means every new task becomes a new engineering project. The paper argues that this is exactly the opposite of what we want if the goal is a general-purpose robot training recipe. A more recent line of work has tried to reduce that burden by improving the starting states the robot sees during training. Instead of always beginning from a standard reset, the simulator can reset the robot into useful intermediate states, so the policy does not have to discover everything from scratch. This is especially helpful for contact-rich tasks like assembly, where the probability of stumbling into a useful state by random exploration is tiny. The authors build on that idea, and on prior work like OmniReset, which showed that diverse, programmatically generated resets can make difficult tasks learnable with massive parallel simulation. But they identify a second bottleneck. Even if you have many useful reset states, you still have to decide how to allocate training across them. If you sample them uniformly, a lot of experience gets wasted. Some resets are already too easy, so the policy gets little new learning signal. Others are far too hard, so the policy cannot yet make progress. As you scale up to more and more parallel environments, this problem gets worse, because a growing fraction of your batch can be spent on configurations that are not currently informative. In other words, the bottleneck is not just exploration in the usual sense, but exploration allocation. That is the central idea behind Success Guided Sampling, or SGS. The method is simple: keep track of how successful the policy currently is on different task configurations, and preferentially sample the configurations where the policy is in the productive middle ground. Those are the cases that are neither solved nor hopeless, but right near the frontier of what the policy can do. The intuition is very similar to a curriculum, but importantly it is not a manually ordered curriculum. The policy’s own success rate determines where training should focus. To make this concrete, the paper sets up goal-conditioned reinforcement learning. Each episode starts from a task configuration that includes the initial state, the goal, and the environment setup. In locomotion, that means things like terrain type, terrain geometry, and the robot’s starting and target poses. In manipulation, it means the assembly task, object and robot initial states, and the receptacle pose. The policy receives a shared reward within each domain: one reward function for all locomotion tasks and one for all manipulation tasks. That is important, because it means they are not hand-tuning a separate reward for every terrain or every assembly problem. SGS itself works over a fixed set of sampled configurations. For each configuration, it maintains a rolling estimate of recent success, based on the last several episode outcomes. Then it scores each configuration with a smooth beta-shaped weighting function that favors a target success rate. You can think of that target as the sweet spot of learning: not too easy, not too hard. Configurations near that target are sampled more often, while configurations with very high or very low success still remain in the mix with nonzero probability. That nonzero probability matters, because it keeps the success estimates fresh and prevents the sampler from abandoning difficult configurations entirely. The paper uses this same basic mechanism in two domains. The first is difficult-terrain quadruped locomotion, where the robot must learn whole-body control across a range of terrains. The second is dexterous assembly, including tasks such as threaded assembly, insertion, connector assembly, and gear meshing. These are contact-rich tasks where precise alignment and sustained interaction matter, and they are exactly the kind of problems where naive exploration tends to struggle. The experimental setup is notable for scale. The authors train with PPO and massive parallel simulation, using up to about one million parallel environments. For manipulation they use a fixed set of 32,768 task configurations, and for locomotion 104,000. They use a rolling window of recent successes, and they choose different target success levels and kernel settings for locomotion and manipulation. The paper also emphasizes that SGS is not especially sensitive to most of these hyperparameters, which is encouraging because it suggests the method is not brittle. A key design choice is that SGS is paired with shared rewards and no demonstrations. That makes the setup much cleaner than many prior robot RL pipelines. Instead of relying on per-task reward engineering or demonstration bootstrapping, the system tries to let the reset distribution and the adaptive sampler do the heavy lifting. The authors are very explicit that the contribution is not a brand-new theory of curriculum learning. Rather, it is a practical and surprisingly effective way to combine success-based sampling with diverse resets and large-scale RL. So what do the results show? The headline result is that SGS makes scale actually useful on tasks where uniform sampling runs into diminishing returns. Across their benchmarks, SGS outperforms uniform sampling and Prioritized Level Replay on the hardest locomotion and assembly problems. In particular, it helps on challenging multi-terrain whole-body locomotion and on nut-and-bolt assembly, where prior methods fail to solve the tasks. On simpler tasks, SGS performs about comparably to uniform sampling, which is also what you would hope to see: the method is not meant to hurt easy problems, but to concentrate learning where the frontier is moving. The broader takeaway is that more parallel environments are only helpful if those environments are producing informative experience. If you simply add more workers and keep sampling configurations uniformly, you can end up with more wasted data rather than more learning. SGS changes that balance by directing the batch toward the configurations that are currently most useful for improvement. That is why the paper frames the problem as an exploration bottleneck in mega-scale RL: the issue is not just whether the robot can explore at all, but whether the training system is spending its compute on the right part of the task space. The paper also includes a real-robot transfer result. The learned manipulation policies are distilled into RGB-based policies and then transferred zero-shot to several challenging assembly tasks on hardware. Zero-shot transfer is important because it means the learned behavior carries over without additional real-world fine-tuning. The paper presents this as evidence that the training recipe is not just succeeding in simulation, but learning skills that are robust enough to survive the sim-to-real gap for precise assembly tasks. There are, of course, limitations and open questions. SGS depends on having a meaningful set of task configurations to sample from in the first place, so it is not a replacement for good environment design. It also uses success rate as the main signal for where to sample next, which is simple and effective, but may not capture every aspect of learning progress. The authors do compare against a few related methods, including uniform sampling, PLR, and in locomotion an SFL-style baseline, but they are careful not to claim that SGS is a fundamentally new curriculum principle. It is a practical instantiation of a broader idea. Another thing to watch is that the strongest gains appear on difficult tasks at large scale. That suggests the method’s value grows when the configuration space is broad and the environment count is high. On simpler tasks, the benefit is smaller. So SGS seems best understood as a scaling tool: it helps large-scale simulation avoid wasting its own capacity. In the end, the paper’s message is quite clear. If you want general-purpose robot learning, it is not enough to make resets diverse. You also have to sample those resets intelligently. Success Guided Sampling is a lightweight way to do that, and in this work it turns massive parallel simulation from a blunt instrument into something much more targeted. That lets the authors solve harder locomotion and assembly problems than prior approaches, while also showing real-world transfer for manipulation. For anyone interested in robot RL, the key lesson is that data quality is not only about what states you include, but about when and how often you train on each one.

Previous daily papers

2026-10-15Latent Compaction with Test-Time Training2026-10-14Tev: Scaling Long-Horizon Agentic Environments with Dense Progress Rewards2026-10-14The Last Remote Sensing Model Built by Humans? Casting Any Earth Observation Task as Recursive Self-Improvement2026-10-14Hybrid Reasoning Agents2026-10-13AI Models Can Provably Hide Arbitrary Capabilities2026-10-13RATED: Reliability-Aware AI Agent Tool Retrieval via Execution Dynamics2026-10-13Designing Protein Structures from Natural Language2026-10-12RSI-Anything: Unlocking Recursive Self-Improvement with World Models2026-10-11FAITH: Feasibility-Aware Safety-Filtered RL for High-Dimensional Systems2026-10-11VioLA: Learning Generalist Humanoid Control Policies from Human Data2026-10-10FastBench: Can Streaming VLMs Perceive High-Dynamic Real-World Streams?2026-10-10Ecology of AI Agents: Collaboration Creates a Population Threshold for Takeoff2026-10-08Long-WAM: Scaling the Context of World-Action Models2026-10-08SciExam for ENSO: Can AI Agents Build Climate Models?2026-10-08RoboJEPA: Scaling Robotic Latent World Models2026-10-08RECAST: Learning to Compute the Right Context through Adaptive Evidence Routing2026-10-07WorldSonus: Bringing Sound to Worlds2026-10-07IdeaAnchor: Teaching LLMs to Turn Literature into Research Ideas2026-10-07Agent in a Bottle: Can LLM Agents Turn Their Capabilities Into Cheap, Scalable Artifacts?2026-10-06CLIFT: Conformal Self-Verification for Web Agent Training and Test-Time Scaling2026-10-06PlotGround: Grounding Plot Digitization in Real Scientific Figures and Their Source Data2026-10-06Base Models Can Reason By Taking a Cue From Training Data2026-10-05LESSER: Post-Training Data Selection with Output-Layer Gradients2026-10-05Language Models that Play Chess and Explain Their Moves2026-10-05Less Decoder is More Encoder: Geometric Representation Learning from Novel View Synthesis2026-10-04One Basis to Animate Them All: Gaussian Blendshape Distillation for Real-Time Avatars2026-10-04The Missing Primitive: Diagnosing and Repairing Mathematical Reasoning in Large Language Models2026-10-04DMAD: Distribution Matching as Adversarial Distillation for Fast Visual Generation2026-10-03VISTA: A Visual Harness for Reasoning in an Interactive World2026-10-01Looped Diffusion Transformer2026-10-01Scaling Laws for Looped Mixture of Experts2026-10-01EvoDuet: Bilevel Co-Evolution of Web Searching and Task Solving for Scientific Discovery2026-09-30Beyond the Timeline: Augmenting Long-Video Memory with Grounded Entity Biographies2026-09-30Thinking Before Thinking: Scaling Agentic Inference Through Meta-Reasoning2026-09-30Pretraining Latent Information Feedback Transformers with Teacher Supervision2026-09-29Unifying Distributional Training for One-Step Visual Generation2026-09-29Improving Test-Time Scaling with Adaptive Looped Transformers2026-09-29Telescopic Language Models2026-09-28Statistical attribute alignment for black-box generative AI via output post-processing2026-09-28User Model Extraction via Belief Self-Distillation2026-09-28Learning to Stop without Learning to Stop: Self-Supervised Confidence Training Improves Reasoning Efficiency2026-09-27TrackEverything: Long Horizon Dense Tracking via De-Duplicating 3D Scene Representations2026-09-27LLM Agents Can Easily Tamper With Their Own Traces2026-09-27Instrumental Monitor Evasion Emerges Under Ordinary Task Pressure2026-09-26PoEM: Predicting RL Outcomes from Existing Policies2026-09-24Frozen Flows Forget: Diagnosing and Restoring Lost Motion in a Latent-flow World Model2026-09-24StudentBench: AI and human tutoring yield equivalent GRE learning gains2026-09-24Agent-Editing World Model: Rethinking World Modeling for LLM Agents2026-09-23Grow the Harness, Not the Context: From Strategy-Free Scaffolds to Reusable Specialist Agents2026-09-23FleXray: Universal Clinical X-ray Segmentation2026-09-23SWE-Serve: Benchmarking Agentic Engineering For Production Inference Serving2026-09-22GameHorizon Suite: Multi-Horizon Data and Evaluation in Gameplay2026-09-22Emergent Collusion in Long-Horizon LLM Agent Interaction2026-09-22WorldCrafter: Consistent Video World Model with Implicit 3D-aware Memory2026-09-21Value-Sensitive Delegation in Everyday AI Agent Use: Evidence from OpenClaw2026-09-21Designer-RSI: Evolving Procedural Memory from User Traffic for Agentic Graphic Design2026-09-21CodeMidas: Scaling Agentic Coding RL Environments from Code Itself2026-09-20RetireOPD: Self-Retiring On-Policy Distillation for Agentic Reinforcement Learning2026-09-20Quantifying Overclaiming Propensity in Frontier LLM Agents