atlas

Research map / Vision-language and multimodal / Robot action and navigation

KISA: A Unified Keyframe Identifier and Skill Annotator for Long-Horizon Robotics Demonstrations

ICML 2024 · Vision-language and multimodal › Robot action and navigation · approach: Memory and subgoal-based manipulation

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes KISA, a temporal-enhanced pretrained vision-language representation trained with history-aware contrastive learning and monotonic distance loss to identify keyframes and annotate skills in long-horizon robot demonstrations.

Problem

Identifying keyframes and annotating semantic skills in unlabeled long-horizon robot manipulation demonstrations is inaccurate with frame-level visual distances or heuristics, which lack language grounding.

Approach

A lightweight Transformer temporal module over frozen-then-tuned pretrained robot representations (R3M/VIP/LIV) aggregates history frames, aligned to skill language embeddings via contrastive learning with hard negatives (incorrect skills, mismatched history, reversed video) plus a monotonic score-distance loss within sub-tasks.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use KISA instead of UVD for keyframe identificationLIV-based keyframe identification · ManiskillF1 Score: 40.2 ± 1.1% -> 99.7 ± 0.2%Helped
Use KISA instead of UVD for keyframe identificationLIV-based keyframe identification · CALVINF1 Score: 36.9 ± 0.5% -> 85.2 ± 0.9%Helped
Use KISA instead of UVD for keyframe identificationLIV-based keyframe identification · FrankaKitchenF1 Score: 64.8 ± 2.4% -> 98.7 ± 0.6%Helped
Annotate demonstrations with KISA instead of using LISA's discovered skillsLISA policy learning · CALVIN (1 instruction)Success rate: 55.4% ± 2.7 -> 82.5% ± 1.6Helped
Annotate demonstrations with KISA instead of using LISA's discovered skillsLISA policy learning · CALVIN (5 instructions)Success rate: 9.3% ± 0.8 -> 17.6% ± 1.0Helped
Add monotonic alignment (ablation)KISA-LIV · Maniskill2Top-1 Accuracy: 90.2 ± 0.3 % -> 99.2 ± 0.1%Helped
Add monotonic alignment (ablation)KISA-LIV · CALVINTop-1 Accuracy: 82.1 ± 0.4 % -> 94.7 ± 1.1%Helped
Add monotonic alignment (ablation)KISA-LIV · FrankaKitchenTop-1 Accuracy: 89.1 ± 0.7% -> 96.1 ± 0.2%Helped
Use history-aware contrastive learning instead of vanilla contrastive learning (ablation)KISA-LIV · Maniskill2Top-1 Accuracy: 59.1 ± 1.9 % -> 99.2 ± 0.1%Helped
Use history-aware contrastive learning instead of vanilla contrastive learning (ablation)KISA-LIV · CALVINTop-1 Accuracy: 58.1 ± 1.3 % -> 94.7 ± 1.1%Helped
Use history-aware contrastive learning instead of vanilla contrastive learning (ablation)KISA-LIV · FrankaKitchenTop-1 Accuracy: 73.7 ± 0.4% -> 96.1 ± 0.2%Helped
Add temporal enhancement (ablation)KISA-LIV · Maniskill2Top-1 Accuracy: 22.0 ± 1.6 % -> 99.2 ± 0.1%Helped

Surprising findings

Limitations

Experimental setup

Datasets: ManiSkill2 CALVIN Franka Kitchen Ego-style long-horizon kitchen demonstrations (Xu et al., 2023)

Benchmarks: ManiSkill2 CALVIN Franka Kitchen real kitchen demonstrations

Metrics: keyframe number error MAE F1 score Top-1 accuracy

Baselines: KTS VideoRLCS UVD R3M VIP LIV LCBC LISA

Builds on: LIV CLIP UVD LISA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy