Instant Policy
Jak działa
1) Reprezentacja grafowa: z segmentowanych chmur punktów budowany jest heterogeniczny graf, który jednocześnie wyraża kontekst (demonstracje), bieżącą obserwację i przyszłe akcje, wprowadzając właściwe obciążenia indukcyjne i umożliwiając ustrukturyzowane wnioskowanie o relacjach robot–otoczenie. 2) ICIL jako generowanie grafu przez dyfuzję: uczony jest rozkład nad grafami za pomocą procesu dyfuzji z forward/backward łańcuchem Markowa — podczas treningu do reprezentacji dodawany jest szum, a model uczy się go usuwać. 3) Inferencja: startując od akcji próbkowanych z rozkładu normalnego, model iteracyjnie aktualizuje (odszumia) wyłącznie części grafu reprezentujące akcje robota, implicytnie modelując warunkowy rozkład akcji (np. 8 kolejnych akcji w przyszłość). 4) Dane: trening na pseudo-demonstracjach — dowolnych trajektoriach generowanych w symulacji — jako praktycznie nieskończonej puli danych.
Rozwiązany problem
Klasyczne imitation learning / behavioural cloning koduje każde zadanie w wagach sieci i wymaga zebrania wielu demonstracji oraz kosztownego treningu dla każdego nowego zadania. Instant Policy eliminuje douczanie na zadanie: nowe umiejętności są przyswajane natychmiast z 1–2 demonstracji podanych jako kontekst, co radykalnie skraca czas wdrożenia nowego zachowania robota.
Komponenty
Graf budowany z segmentowanych chmur punktów, który jednocześnie reprezentuje kontekst (demonstracje), bieżącą obserwację i przyszłe akcje, wprowadzając obciążenia indukcyjne dla ustrukturyzowanego wnioskowania.
Model dyfuzji uczący rozkład nad grafami (forward/backward łańcuch Markowa). Przy inferencji odszumia iteracyjnie wyłącznie węzły akcji, implicytnie modelując warunkowy rozkład akcji robota.
Dowolne trajektorie generowane w symulacji, używane jako praktycznie nieskończona pula danych treningowych zamiast ręcznie zbieranych demonstracji na zadanie.
Implementacja
Ewolucja
Pierwsza wersja pracy „Instant Policy: In-Context Imitation Learning via Graph Diffusion" (19.11.2024).
Przyjęcie na ICLR 2025 jako prezentacja ustna (Oral); zdobycie nagrody Best Paper na warsztacie Robot Learning (WRL@ICLR 2025).
Hiperparametry (konfigurowalne osie)
Liczba demonstracji podawanych jako kontekst in-context. W pracy wystarczają 1–2.
Liczba kolejnych akcji przewidywanych w jednym kroku generacji grafu.
Liczba iteracji odszumiania w procesie dyfuzji przy generowaniu akcji.