SkillDiscovery-STEVE1-loss-only

Open-World Skill Discovery from Unsegmented Demonstrations

• Paper https://huggingface.co/papers/2503.10684
• Code https://github.com/CraftJarvis/SkillDiscovery
• Project Page https://craftjarvis.github.io/SkillDiscovery

This is the improved version of STEVE-1 trained on the segmented dataset (loss only) of the SkillDiscovery project.

Abstract

Learning skills in open-world environments is essential for developing agents capable of handling a variety of tasks by combining basic skills. Online demonstration videos are typically long but unsegmented, making them difficult to segment and label with skill identifiers. Unlike existing methods that rely on random splitting or human labeling, we have developed a self-supervised learning-based approach to segment these long videos into a series of semantic-aware and skill-consistent segments. Drawing inspiration from human cognitive event segmentation theory, we introduce Skill Boundary Detection (SBD), an annotation-free temporal video segmentation algorithm. SBD detects skill boundaries in a video by leveraging prediction errors from a pretrained unconditional action-prediction model. This approach is based on the assumption that a significant increase in prediction error indicates a shift in the skill being executed. We evaluated our method in Minecraft, a rich open-world simulator with extensive gameplay videos available online. The SBD-generated segments yielded relative performance improvements of 63.7% and 52.1% for conditioned policies on short-term atomic tasks, and 11.3% and 20.8% for their corresponding hierarchical agents on long-horizon tasks, compared to random segmented baselines. Our method can leverage the diverse YouTube videos to train instruction-following agents.

Citation

@article{deng2025openworld,
  title={Open-World Skill Discovery from Unsegmented Demonstrations},
  author={Jingwen Deng and Zihao Wang and Shaofei Cai and Anji Liu and Yitao Liang},
  journal={arXiv preprint arXiv:2503.10684},
  year={2025}
}