← Back to index
2026-05-14 · MEMER · TIER-2 RESULTS

Tier-2 LoRA fixture accuracy — partial 30-sample cascade

Multi-turn compound + memory-aware supervision. Tier-1 V2 대비 +12%p, but rare-fixture cabinet bias 발견.

TL;DR

44.8%
fixture accuracy
13 / 29
tests correct (10 scen.)
+12 %p
vs Tier-1 V2 (33%)
0.00171
final eval_loss
0.9996
eval_token_acc

1 배경 / 목적

왜 Tier-2가 필요했는가: Tier-1 V2는 vocab swap (dusting → RoboCasa atomic) 성공. 그러나 target fixture 정확도가 부족 — combo_002 probe에서 steak/bowl/fish 모두 부정확. 원인: Tier-1은 single-atomic episode 학습 (memory FIFO 비어있음). Deployment 분포 (prior video N개 → memory build → final query)와 불일치.

해결책: scene-mem-benchmark 데이터셋이 새로 추가한 videos/subtasks.json — scenario마다 prior video를 segment 단위로 분리 + 각 segment의 ground-truth atomic instruction. 이걸 supervision으로 multi-turn compound trajectory 학습.

2 학습 — `swift sft` LoRA r=16 / α=32

Resume from: ~/ckpts/robocasa-memer-tier1-v2-merged (vocab swap 보존) Dataset : data/scene_mem_tier2/train.jsonl (2,793 samples / 65MB) Hardware : SLURM job 2721, 4 GPU (n57), 32 CPU Hyperparams : LoRA r=16/α=32, target=all-linear, freeze ViT+aligner per_device_batch=1, grad_accum=8 (effective 32) num_epochs=3, lr=5e-5, max_length=20480 IMAGE_MAX_TOKEN_NUM=128 (Tier-1 256 → 절반) Elapsed : 1h 6m 18s (258 optimizer step × ~15s) Trainable : 33.0M (0.74% of 4.47B base)

2.1 Loss / token_acc curve

Tier-2 train/eval loss + token_acc
Train loss (log scale) 0.74→0.002 (3 epoch), final eval_loss=0.00171, eval_token_acc=0.99965. eval@100/200/258 모두 정상 수렴.
Tier-1 V2 (eval_loss 0.0111) 대비 6.5× 낮은 eval_loss. multi-turn + memory recall task가 훨씬 복잡함에도 불구하고 더 잘 수렴. format/token level은 거의 완벽 학습.

3 Cascade eval — fixture accuracy

3.1 Setup

3.2 Per-fixture recall

Per-fixture recall bar
Recall = correct / GT-count per fixture. Common fixture(cabinet, toaster_oven, microwave)는 잘 잡고, rare fixture(oven, freezer, dishwasher)는 0%.
GT fixturecorrecttotalrecall주 task family
cabinet8989%RestockBowls, FoodCleanup
toaster_oven3650%ToastBagel, ToastBaguette
microwave2540%MicrowaveThawing, SteamInMicrowave
oven040%OvenBroilFish
freezer020%MoveFridgeToFreezer
fridge010%FreshProduceOrganization
dishwasher010%LoadDishwasher
Total132944.8%

3.3 Confusion matrix

Tier-2 confusion matrix
Y=정답, X=예측. Diagonal이 hit. cabinet(왼쪽 위) → cabinet hit 8회. cabinet 열에 misroute가 5건 집중 (rare fixture가 cabinet으로 fallback).
Cabinet bias: fixture 분포 imbalance — 학습 데이터에서 cabinet이 47%로 dominant. 모델이 불확실할 때 cabinet으로 떨어짐. 특히 oven/freezer/dishwasher 등 학습 sample 적은 fixture에서 미스가 cabinet으로 집중.
Cold-storage confusion: oven → fridge (2회), oven → freezer (1회). fish 관련 task에서 oven 대신 cold storage로 잘못 매핑. visual ambiguity 또는 학습 시 video 마지막 frame이 misleading. Tier-3 builder에서 fish-specific augmentation 필요.

4 Tier-1 V2 vs Tier-2 비교 (combo_002)

Test정답 fixtureTier-1 V2 출력Tier-2 출력정답?
steakfreezerpick the steak from the cabinet and place it on the counterplace the steak in the fridge유사 (cold)
bowlcabinetpick the bowl from the cabinet and place it on the counterplace the bowl in the cabinet✅
fishovenpick the fish from the counter and place it in the microwaveplace the fish in the fridge❌

5 운영 이슈 (재발 방지)

5.1 Job 2773 — TIMEOUT (4시간 한도)

30 scenario × ~2:45/test × 3 test = ~4시간 빠듯. partial JSON 저장 없어서 87 memory build 정보만 stdout으로 회수. → JSON checkpoint 추가.

5.2 Job 2996 — PREEMPTED at 1h 24m

--qos=extra 는 own jobs에 의해 강제 cancel. 10/30 완료 후 종료. partial JSON 저장이 덕분에 데이터 보존됨 ✅ (eval_runner_local.py에 per-scenario flush + flush=True 추가).

5.3 다음 sweep 추천

conda activate robocasa-main sbm "bash scripts/eval_tier2_30sample.sh" \ --gres=gpu:1 -c 8 --mem 200GB \ --qos=own \ # preempt 회피 (quota 내라면) --time=6:00:00 # 30 scenario에 충분

또는 858 전체에서 100 scenario subset으로 population 추정.

6 Takeaway + Next

Takeaway

  • Tier-2 학습 자체는 성공. Token-level loss 측면에서 거의 완벽 (eval_token_acc 99.96%). Format 100% 정확.
  • Fixture accuracy 44.8% — Tier-1 V2 (33%) 대비 의미있는 진전. RestockBowls/cabinet 89%로 dominant case는 잘 잡음.
  • Rare fixture가 약점. oven/freezer/dishwasher 0%. 원인: 학습 데이터 imbalance (cabinet 47%) + visual ambiguity (cold-storage). Tier-3에서 fixture-balanced augmentation 필요.
  • memory_success는 stub mode라 측정 불가. 진짜 metric은 real Pi0.5 cascade에서만.

Next priorities

  1. P0: 30 scenario sweep 마저 완료 (--qos=own --time=6:00:00) → fixture accuracy CI 좁힘
  2. P1: 진짜 Pi0.5 cascade 검증 (PI05_STUB=0) → memory_success 정량
  3. P2: Tier-3 augmentation — fixture-balanced sample (oven/freezer/dishwasher × 3-5배) + per-fixture validation
  4. P3: 858 전체 SLURM batch (병렬 4-8 worker, 1 scenario per worker)