Search papers, labs, and topics across Lattice.
This paper introduces SIEVE, a novel structure-aware data selection method designed to enhance imitation learning in Vision-Language-Action (VLA) models by focusing on reusable visuo-motor primitives and transition interfaces. By maximizing structural exposure and selecting central medoid trajectories, SIEVE effectively reduces the amount of training data needed while improving policy performance. Experimental results demonstrate that SIEVE can outperform traditional data selection methods, achieving better outcomes with only half the data and training steps compared to full-data training.
SIEVE reveals that leveraging reusable structures in demonstration data can lead to more efficient and effective imitation learning, outperforming full-data training with significantly less input.
Vision-Language-Action (VLA) models are typically trained by imitation learning on large-scale robot demonstration datasets, but more data does not necessarily yield better policies due to redundancy, noise, and uneven coverage. Existing data selection methods often assess demonstrations at either the trajectory or state-action level, missing the reusable structures that compose long-horizon behaviors. In this paper, we propose SIEVE, a structure-aware data selection method for VLA imitation learning. SIEVE views demonstrations as compositions of reusable primitives and transition interfaces. It first discovers visuo-motor primitives from segmented trajectories, then allocates selection budgets to composition patterns by maximizing reuse-aware structural exposure under diminishing returns. Finally, it selects medoid trajectories within each composition-pattern bucket to retain central, stable, and imitation-friendly demonstrations. Experiments across multiple datasets, benchmarks, and VLA models show that SIEVE consistently outperforms competitive data selection baselines. Notably, SIEVE can surpass full-data training while using only 50% of demonstrations and 50% of training steps, suggesting that reusable structure, captured through primitives and transitions, is an important signal for efficient VLA imitation learning.
