Search papers, labs, and topics across Lattice.
This paper introduces UCA-Flow, a unified condition-action modeling framework that enhances one-step action generation for robot manipulation by integrating observation and action representations into a shared token space. By employing a Unified Condition-Action Transformer, the model dynamically reconstructs condition representations during action refinement, leading to significant improvements in both accuracy and inference speed. The results show a 9.3 percentage point increase in success rate compared to the best baseline, alongside substantial speedups over existing methods.
UCA-Flow achieves a remarkable 9.3% increase in action success rates while being up to 45.6 times faster than previous approaches.
Robot manipulation requires policies that are both accurate and efficient, as robot control must respond to changing observations under tight latency constraints. Recent diffusion and flow policies are promising, but they often treat conditions as auxiliary signals rather than jointly evolving them with action trajectories. We find that this limitation can be effectively mitigated by a \textbf{simple yet effective unified condition-action modeling design} that represents conditions and actions in a shared token space, allowing a compact model to achieve high performance while improving both inference speed and accuracy. Therefore, we propose UCA-Flow, a unified condition-action modeling framework for accurate one-step action generation. Our method unifies observation conditions, timestep conditions, interval conditions, and action tokens into a single sequence, and processes them with a Unified Condition-Action Transformer for joint condition-action representation learning. As a result, condition representations are dynamically reconstructed according to the current generation stage, highlighting information most relevant for action refinement. Furthermore, we introduce an improved dual-pass supervision scheme over $u$ and $v$ for stronger optimization of unified condition-action modeling. UCA-Flow improves the average success rate by 9.3 percentage points over the strongest baseline, while achieving $45.6\times$ and $33.4\times$ speedups over DP3 and Simple DP3, and remaining $4.3\times$ and $2.3\times$ faster than one-step FlowPolicy and MP1, respectively.
