Latent Action Learning Requires Supervision in the Presence of Distractors

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Nikulin, Alexander, Zisman, Ilya, Tarasov, Denis, Lyubaykin, Nikita, Polubarov, Andrei, Kiselev, Igor, Kurenkov, Vladislav
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908405414756352
author Nikulin, Alexander
Zisman, Ilya
Tarasov, Denis
Lyubaykin, Nikita
Polubarov, Andrei
Kiselev, Igor
Kurenkov, Vladislav
author_facet Nikulin, Alexander
Zisman, Ilya
Tarasov, Denis
Lyubaykin, Nikita
Polubarov, Andrei
Kiselev, Igor
Kurenkov, Vladislav
contents Recently, latent action learning, pioneered by Latent Action Policies (LAPO), have shown remarkable pre-training efficiency on observation-only data, offering potential for leveraging vast amounts of video available on the web for embodied AI. However, prior work has focused on distractor-free data, where changes between observations are primarily explained by ground-truth actions. Unfortunately, real-world videos contain action-correlated distractors that may hinder latent action learning. Using Distracting Control Suite (DCS) we empirically investigate the effect of distractors on latent action learning and demonstrate that LAPO struggle in such scenario. We propose LAOM, a simple LAPO modification that improves the quality of latent actions by 8x, as measured by linear probing. Importantly, we show that providing supervision with ground-truth actions, as few as 2.5% of the full dataset, during latent action learning improves downstream performance by 4.2x on average. Our findings suggest that integrating supervision during Latent Action Models (LAM) training is critical in the presence of distractors, challenging the conventional pipeline of first learning LAM and only then decoding from latent to ground-truth actions.
format Preprint
id arxiv_https___arxiv_org_abs_2502_00379
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Latent Action Learning Requires Supervision in the Presence of Distractors
Nikulin, Alexander
Zisman, Ilya
Tarasov, Denis
Lyubaykin, Nikita
Polubarov, Andrei
Kiselev, Igor
Kurenkov, Vladislav
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Recently, latent action learning, pioneered by Latent Action Policies (LAPO), have shown remarkable pre-training efficiency on observation-only data, offering potential for leveraging vast amounts of video available on the web for embodied AI. However, prior work has focused on distractor-free data, where changes between observations are primarily explained by ground-truth actions. Unfortunately, real-world videos contain action-correlated distractors that may hinder latent action learning. Using Distracting Control Suite (DCS) we empirically investigate the effect of distractors on latent action learning and demonstrate that LAPO struggle in such scenario. We propose LAOM, a simple LAPO modification that improves the quality of latent actions by 8x, as measured by linear probing. Importantly, we show that providing supervision with ground-truth actions, as few as 2.5% of the full dataset, during latent action learning improves downstream performance by 4.2x on average. Our findings suggest that integrating supervision during Latent Action Models (LAM) training is critical in the presence of distractors, challenging the conventional pipeline of first learning LAM and only then decoding from latent to ground-truth actions.
title Latent Action Learning Requires Supervision in the Presence of Distractors
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2502.00379