MEKAWIRE
上級arXiv

Breaking the Average: Overcoming Convergent Defaults in Robotic Manipulation via Diagnostic Chain-of-Thought and Structured Guidance

Mohammed Daba, Waleed Abdulrahman Saleh AlAkwa
日本語版は未提供のため、英語原文で表示しています。
要旨より

Recent Vision-Language-Action (VLA) models, such as π_(0.5), have demonstrated impressive generalization capabilities by scaling behavior cloning across massive datasets. However, these models frequently suffer from distributional convergence, a phenomenon where the policy regresses to "safe" but suboptimal mean behaviors due to mode averaging in the training data.

要旨より。当社による要約は作成中です。

論文全文

Breaking the Average: Overcoming Convergent Defaults in Robotic Manipulation via Diagnostic Chain-of-Thought and Structured Guidance

Mohammed Daba, Waleed Abdulrahman Saleh AlAkwa

CC BY 4.0 のもとで公開された論文です。出典を明記して転載しています。原文は arXiv:2610.17983(PDF)。

Abstract

Recent Vision-Language-Action (VLA) models, such as \pi_{0.5}, have demonstrated impressive generalization capabilities by scaling behavior cloning across massive datasets. However, these models frequently suffer from distributional convergence, a phenomenon where the policy regresses to “safe” but suboptimal mean behaviors due to mode averaging in the training data. This often results in generic grasp strategies, Convergent Defaults, that fail in constrained or long-tail scenarios. In this work, we propose a novel framework to identify and correct these behaviors without parameter updates. We introduce a three-stage methodology: (1) Probing for Defaults, identifying task-specific mode collapse; (2) Diagnostic Root Cause Analysis, utilizing the VLA’s latent reasoning capabilities to verbalize the logic behind suboptimal actions; and (3) Structured Guidance Injection, where concrete, distinct alternatives are encoded into reusable “Lens Prompts” that shift the sampling distribution. We validate this approach using \pi_{0.5} in ManiSkill3 simulation environments. We hypothesize that our method outperforms vanilla \pi_{0.5} and standard Chain-of-Thought prompting by effectively breaking convergent defaults in complex manipulation tasks.

Index Terms:

Vision-Language-Action Models, Robotic Manipulation, In-Context Learning

I Introduction

The scaling of Foundation Models in robotics has led to the emergence of general-purpose Vision-Language-Action (VLA) models [1, 2]. Models such as \pi_{0.5} utilize flow matching and massive datasets to learn versatile manipulation policies. However, a critical limitation remains: distributional convergence.

In generative AI, this manifests as generic outputs where models produce statistically probable but safe responses to minimize loss [3]. In robotics, we observe an analogous phenomenon: Mode-Averaged Action Priors. When faced with ambiguity, VLA models often default to the kinematic mean of their training demonstrations. For example, a robot may approach an object from a generic top-down angle regardless of occlusion or fail to account for workspace constraints because the ”average” training example assumes a clutter-free table. These “Convergent Defaults” are not hallucinations in the traditional sense; they are safe, high-likelihood predictions that fail in specific, context-heavy scenarios. We visually illustrate this phenomenon and our proposed solution in Fig. 1.

Existing solutions largely focus on online self-correction [4] or reactive replanning [5]. While effective for error recovery, these methods are reactive. They do not address the underlying prior that biased the robot toward the suboptimal action in the first place.

Fig. 1: Breaking the Convergent Default. VLA models often regress to the mean of their training data, causing failures in constrained environments. Our method identifies these modes and injects structured guidance (Lens) to shift the action distribution toward expert behaviors.

In this paper, we propose a framework to proactively reshape the VLA’s action distribution. We formalize a methodology to: 1) Isolate Convergent Defaults by identifying where the model chooses a “safe mean” over a “context-optimal” action; 2) Diagnose via Reasoning by forcing the VLA to perform a “Debug Mode” analysis to verbalize latent assumptions; and 3) Inject Structured Guidance by constructing reusable prompt modules, termed Lens Prompts, that explicitly prohibit the default and provide concrete alternatives.

We apply this to \pi_{0.5} in the ManiSkill3 [6] simulation environment. Our contribution is a formalized pipeline for extracting expert-level performance from generalist VLAs without fine-tuning, shifting the paradigm from prompt engineering to distribution shaping.

Fig. 2: The Diagnostic Lens Framework. We treat the VLA as a reasoning engine to diagnose its own action priors. By extracting the root cause of a default behavior, we construct a textual ”Lens” that reshapes the flow matching distribution during inference.

II Related Work

II-A Vision-Language-Action (VLA) Models

Recent works like RT-2 [2] and \pi_{0} [1] integrate visual perception with language reasoning. \pi_{0.5}, our baseline, utilizes flow matching to handle multi-modal distributions. Despite this, flow models can still suffer from mode collapse in the conditional generation step when prompts are underspecified [7]. Our work addresses the gap where the model technically can perform the task (if given the right steering), but the probability mass is centered on the wrong (generic) mode.

II-B Test-Time Adaptation & Reasoning

Methods like “Inner Monologue” [8] and “Self-Correcting VLAs” [4] use LLMs to critique robot actions. However, these often rely on binary success/failure feedback. Our work differentiates itself by focusing on the root cause analysis of the prior distribution. We argue that generic feedback (“You failed, try again”) is less effective than Structured Guidance, a pre-computed set of positive constraints derived from analyzing the model’s specific convergent defaults.

III Methodology

Our goal is to modify the policy \pi(a_{t}|s_{t},l) (where l is the language instruction) to a shaped policy \pi(a_{t}|s_{t},l,\mathcal{G}_{lens}) without fine-tuning weights. We achieve this through the Diagnostic Lens Framework, a three-step cycle illustrated in Fig. 2.

III-A Phase 1: Probing for Convergent Defaults

We formalize a Convergent Default \mathcal{D} as a high-density region in the action space where the model exhibits low variance (high confidence) despite suboptimal task performance. This occurs when the training distribution P_{train}(a|s) is dominated by generic trajectories.

To identify \mathcal{D}, we probe the \pi_{0.5} model with a zero-shot instruction l_{base}. We define the default behavior a_{default} as the mode of the sampled actions. In constrained scenarios, a_{default} often corresponds to the ”kinematic average” of the training data (e.g., a straight-line approach) rather than a context-sensitive trajectory. Identification involves observing systematic failures where the physical constraints of the specific scene (e.g., obstacles, tight tolerances) contradict the statistical prior of the model.

III-B Phase 2: Diagnostic Root Cause Analysis

Merely observing the failure is insufficient; we must identify the latent misconception driving the default. We exploit the VLA’s capacity for verbalization to perform Diagnostic Root Cause Analysis. By freezing the visual context and prompting the model with a diagnostic query \mathcal{P}_{diag}, we extract the reasoning trace r_{latent}.

This process assumes that the VLA possesses the correct physical knowledge in its latent space but fails to activate it during standard action generation due to the strength of the convergent default. The diagnostic prompt forces the model to shift attention from action generation to state explanation. The output reveals whether the failure stems from a perceptual hallucination (e.g., ”I did not see the shelf”) or a heuristic bias (e.g., ”I prioritized the top grasp because it is generally more stable”). This verbalized bias constitutes the Root Cause.

Fig. 3: Illustrative Example: Constrained Mug Grasping. This figure demonstrates the practical application of the methodology. The ”Convergent Default” (top-down grasp) is identified, diagnosed as a statistical bias, and corrected via a specific Lens Prompt containing concrete alternatives.

III-C Phase 3: Structuring Guidance (The “Lens”)

The final phase involves synthesizing the root cause into a structured intervention, denoted as the Lens (\mathcal{G}_{lens}). The Lens is not merely a hint; it is a rigid context modifier designed to prune the default mode from the sampling distribution.

The Lens is constructed with three mandatory components:

  1. Negative Constraint: Explicitly forbidding the identified convergent default.
  2. Positive Alternative: Providing a concrete, geometrically distinct behavior to replace the default.
  3. Reasoning Anchor: grounding the alternative in the physical logic derived from Phase 2.

Mathematically, this modifies the conditioning variable of the flow matching model, shifting the probability mass away from \mathcal{D} and toward the long-tail region containing the optimal solution. An example of this construction is detailed in Fig. 3.

IV Experimental Setup

IV-A Simulation Environment

We conduct experiments in ManiSkill3 [6], a SAPIEN-based simulator known for high-fidelity physics and realistic rendering. This environment is crucial for testing VLA perception, as it supports diverse object geometries and physical interactions.

IV-B Selected Benchmark Tasks

We select tasks specifically designed to punish “average” behaviors (Convergent Defaults):

  • Constrained Grasping: Picking up objects from inside a shelf. The default behavior (top-down) causes collision with the shelf ceiling.
  • Precision Insertion: Peg-in-hole with tight tolerances. The default behavior often lacks the necessary precision and re-orientation.
  • Articulated Cabinet Opening: Opening a hinged cabinet door. The default behavior often attempts a linear pull (regression to the mean) rather than the required arced trajectory, causing the gripper to slip or the mechanism to jam.

IV-C Baselines

We compare our method against the following:

  1. Vanilla \pi_{0.5}: Zero-shot execution with standard instructions.
  2. Generic Chain-of-Thought (CoT): Prompting with “Think step by step” without domain-specific structured guidance.
  3. Reflexion (Online): A standard feedback loop where the model retries based on binary failure flags [9].

IV-D Metrics

  • Success Rate (SR): Percentage of successful task completions.
  • Reasoning Alignment: A qualitative score measuring if the robot’s generated text explanation matches its physical actions.

V Hypothesis and Expected Outcomes

We hypothesize that Vanilla \pi_{0.5} will exhibit high failure rates in Constrained Grasping and Articulated tasks due to mode-averaging (convergent defaults). We expect Generic CoT to improve planning logic but potentially hallucinate affordances due to a lack of grounded constraints.

We anticipate that the Diagnostic Lens Framework will achieve the highest success rate. By explicitly shifting the probability mass toward the “long-tail” solutions (Concrete Alternatives) identified during the diagnostic phase, the model should effectively break out of the mode collapse. We expect to see a significant reduction in collisions for the constrained grasping task and improved trajectory adherence for articulated objects.

VI Conclusion

This work bridges the gap between generative distribution shaping and robotic control. By identifying and restructuring the “Convergent Defaults” of \pi_{0.5}, we enable the model to access expert-level behaviors hidden within its latent space. This approach offers a pathway to improve generalist robot performance in constrained environments without the high computational cost of parameter fine-tuning.

References

  1. [1] K. Black, N. Brown, D. Driess, A. Esmail, M. Equi, C. Finn, N. Fusai, L. Groom, K. Hausman, B. Ichter, et al. \pi0: A vision-language-action flow model for general robot control. corr, abs/2410.24164, 2024. doi: 10.48550. arXiv preprint ARXIV.2410.24164.
  2. [2] B. Zitkovich, T. Yu, S. Xu, P. Xu, T. Xiao, F. Xia, J. Wu, P. Wohlhart, S. Welker, A. Wahid, et al. (2023) Rt-2: vision-language-action models transfer web knowledge to robotic control. In Conference on Robot Learning, pp. 2165–2183.
  3. [3] J. Wei, X. Wang, D. Schuurmans, M. Bosma, F. Xia, E. Chi, Q. V. Le, D. Zhou, et al. (2022) Chain-of-thought prompting elicits reasoning in large language models. Advances in neural information processing systems 35, pp. 24824–24837.
  4. [4] C. Li, J. Liu, G. Wang, X. Li, S. Chen, L. Heng, C. Xiong, J. Ge, R. Zhang, K. Zhou, et al. (2024) A self-correcting vision-language-action model for fast and slow system manipulation. arXiv preprint arXiv:2405.17418.
  5. [5] P. Sharma, B. Sundaralingam, V. Blukis, C. Paxton, T. Hermans, A. Torralba, J. Andreas, and D. Fox (2022) Correcting robot plans with natural language feedback. arXiv preprint arXiv:2204.05186.
  6. [6] J. Gu, F. Xiang, X. Li, Z. Ling, X. Liu, T. Mu, Y. Tang, S. Tao, X. Wei, Y. Yao, X. Yuan, P. Xie, Z. Huang, R. Chen, and H. Su (2023) ManiSkill2: a unified benchmark for generalizable manipulation skills. In International Conference on Learning Representations,
  7. [7] P. Florence, C. Lynch, A. Zeng, O. A. Ramirez, A. Wahid, L. Downs, A. Wong, J. Lee, I. Mordatch, and J. Tompson (2022) Implicit behavioral cloning. In Conference on robot learning, pp. 158–168.
  8. [8] W. Huang, F. Xia, T. Xiao, H. Chan, J. Liang, P. Florence, A. Zeng, J. Tompson, I. Mordatch, Y. Chebotar, et al. (2022) Inner monologue: embodied reasoning through planning with language models. arXiv preprint arXiv:2207.05608.
  9. [9] N. Shinn, F. Cassano, A. Gopinath, K. Narasimhan, and S. Yao (2023) Reflexion: language agents with verbal reinforcement learning. Advances in Neural Information Processing Systems 36, pp. 8634–8652.