기술 요약: 등변 세계 모델을 위한 공형 궤도 유효 신뢰 지평 (Conformal Orbit-Valid Trust Horizons for Equivariant World Models)

1. 문제 정의

학습된 세계 모델(World models)은 본질적으로 그 롤아웃 오차(rollout error)에 의해 제한되며, 시간이 지남에 따라 성능이 저하된다. 모델 기반 계획(model-based planning)과 자기 지도 학습(self-supervision)은 이러한 모델을 앞으로 전개(rolling forward)하는 과정에 의존하지만, 모델의 예측이 신뢰할 수 있는 특정 단계 수인 "신뢰 지평(trust horizon)"을 결정하는 원칙적인 방법은 존재하지 않는다. 기존 방식들은 단일 분포에 특화된 고정된 하이퍼파라미터(예: 고정된 $H=25$)에 의존하는 경우가 많으며, 이는 정식적인 보증(formal guarantees)을 결여하고 있다.

나아가, 알려진 군 대칭성(group symmetries, 등변 모델)을 가진 세계 모델의 경우, 다음과 같은 중요한 질문이 남는다. "특정 지역에서 도출된 신뢰 인증서가 모든 변환된 상태에 대해 별도의 캘리브레이션(calibration)을 요구하지 않고도 전체 군 궤도(group orbit) 전체에서 유효할 수 있는가?" 기존 방법들은 정식적인 안전 보증이 부족하거나, 캘리브레이션 비용을 줄이기 위해 등변성(equivariance)의 기하학적 구조를 활용하는 데 실패한다.

2. 방법론

2.1 등변 세계 모델 설정

저자들은 다음과 같이 구성된 잠재 세계 모델(latent world model)을 정의한다:

• 인코더 (Encoder, $E$): 관측값 $x$를 잠재 상태 $z$로 매핑한다.
• 예측기 (Predictor, $f$): 행동 $a$에 조건화되어 잠재 상태를 전진시킨다 ($\hat{z}_{t+1} = f(z_t, a_t)$).
• 대칭성 (Symmetry): 모델은 $E(g \cdot x) = \rho(g)E(x)$ 및 $f(\rho(g)z, \sigma(g)a) = \rho(g)f(z,a)$를 만족하는 등변 조건을 충족한다 (여기서 $g \in \mathcal{G}$).
• 구현: 인코더는 CN-steerable 네트워크(2D) 또는 Vector-Neuron/SE(3) 네트워크(3D)로 구현된다. 등변성은 기계 정밀도 또는 학습된 허용 오차에 대한 잔차(residual)를 확인하는 단위 테스트를 통해 검증된다.

2.2 가공되지 않은 지평(Raw Horizon) 및 공형 캘리브레이션

본 방법론의 핵심은 신뢰 지평 인증서를 구축하는 것이다:

1. 가공되지 않은 곡선 (Raw Curve): 1단계 잠재 잔차 $\hat{\delta}$와 유한 시간 확장 추정치 $\hat{\lambda}_1$ (Benettin/QR 스킴을 통해 유도된 예측기 자코비안으로부터 도출)이 가공되지 않은 오차 전파 곡선을 형성한다:
   $$\widehat{Err}(H) = \hat{\delta} \sum_{t=0}^{H-1} e^{\hat{\lambda}_1 t}$$
   가공되지 않은 지평 $H_{raw}$는 이 오차가 허용 오차 $\epsilon$ 아래에 머무는 최대 단계 수이다.
2. 분할 공형 캘리브레이션 (Split-Conformal Calibration): 통계적 타당성을 보장하기 위해, 가공되지 않은 곡선은 곱셈 인자 $\gamma_\alpha$에 의해 스케일링된다. 인증된 지평 $H_{conf}$는 $\widehat{Err}_{conf}(H) = \gamma_\alpha \widehat{Err}(H) \le \epsilon$을 만족하도록 정의된다.
   • 인자 $\gamma_\alpha$는 새로운 교환 가능한 감사(exchangeable audit) 세트에 대해 $P(H_{conf} \le H_{meas}) \ge 1-\alpha$를 보장하도록 분할 공형 예측(split-conformal prediction)을 통해 결정된다.
   • 결정적으로, 저자들의 재현 가능한 감사 세트에서 가공되지 않은 곡선은 이미 보수적이었으므로 $\gamma_\alpha = 1.0$이었다.

2.3 궤도 유효 전송 (Orbit-Valid Transport)

본 논문은 통계적 캘리브레이션과 등변성의 기하학적 특성을 결합한 구조적 결과를 소개한다:

• 정리 A (궤도 전송): 모델이 정확하게 등변하다면, 롤아웃 오차, 확장률, 그리고 결과적인 지평 곡선은 군 궤도를 따라 일정하다.
• 따름정리 C (궤도 유효 인증서): 만약 인증서가 기본 영역(fundamental region, 즉 "쐐기(wedge)")에서 캘리브레이션되었고 정확한 등변성이 성립한다면, 해당 인증서는 동일한 신뢰 수준 $1-\alpha$로 전체 궤도에 대해 유효하다.
• 시사점: 등변성은 궤도 전체의 커버리지 이벤트를 (합집합 상한(union bounds)과 방대한 데이터가 필요한) 모든 군 변환에 대한 합집합 문제로부터 단일 캘리브레이션 이벤트로 변환한다. 이는 궤도별 캘리브레이션 데이터를 필요로 하지 않게 한다.

2.4 근사적 등변성 처리

학습된 네트워크는 근사적으로만 등변하기 때문에, 저자들은 인코더와 예측기의 결함에 대한 리프시츠 경계(Lipschitz bounds)를 사용하여 전송 잔차($R_t$)를 고려한다. 경험적으로 이러한 잔차는 매우 작음(중앙값 1.1%)을 발견하였으며, 이를 통해 인증서가 최소한의 팽창만으로도 유효성을 유지할 수 있음을 확인했다.

3. 주요 기여

1. 공형 신뢰 지평 인증서: 가공되지 않은 잠재 오차 전파 곡선을 분할 공형 캘리브레이션을 사용하여 일방향적이고 보수적인 인증서로 변환하는 방법론을 제시하였다. 이 방법은 안정적인 감사에서 0/50의 반-보수적(anti-conservative) 위반을 기록했으며, 위반율에 대해 정확한 이항 상한 95%는 5.8%였다.
2. 궤도 유효성 정리: 정확한 등변성이 존재할 때, 단일 섹터에서 캘리브레이션된 인증서가 추가적인 캘리브레이션 비용 없이 전체 군 궤도로 전송될 수 있음을 증명하였다 (따름정리 C).
3. 날카로움(Sharpness) 및 잔차 고려: 인증서는 비공허(non-vacuous)하며, 인증된 대비 측정된 지평 비율($H_{conf}/H_{meas}$)의 중앙값은 0.67이다. 구현 수준의 전송 잔차는 작았다 (중앙값 1.1%, 최대 4.1%).
4. 기질 의존적 캘리브레이션 비용 (Substrate-Dependent Calibration Cost): 본 논문은 캘리브레이션 비용을 줄이는 등변성의 이점이 데이터 기질(data substrate)에 따라 조건부임을 밝힌다.
   • 2D 대칭 기질: 비등변 베이스라인은 이미 대칭적인 환경 덕분에 근사적으로 궤도 강건성을 가졌으므로, 등변성이 캘리브레이션 비용 측면에서 차별화된 이점을 제공하지 못했다.
   • 3D Yaw 기질: 비등변 베이스라인은 상당한 여유(slack), 위반, 또는 시드 불안정성 없이 궤도 안전성을 달성하는 데 실패했다. 반면, 등변 모델은 단일 캘리브레이션 섹터로부터 안전하고 비공허한 인증서를 얻어냈다.

4. 결과 및 경험적 발견

• 안전성: 50개의 안정적인 감사 셀(audit cells) 전반에 걸쳐, $\gamma_\alpha=1.0$인 공형 인증서는 반-보수적 위반을 전혀 보이지 않았다.
• 날카로움: 172개의 셀 레벨 체크에서, 98%의 체크가 측정된 지평의 최소 25%를 유지했고, 69%는 최소 50%를 유지했다. 이 인증서는 짧은 지평($H \in [1,8]$)에 대한 날카로운 실행 가능성 게이트 역할을 한다.
• 캘리브레이션 비용:
  • 2D (PushT)의 경우, 등변, 일반, 증강 모델 모두 하나의 섹터로부터 궤도 유효성을 가졌다.
  • 3D (ManiSkill yaw)의 경우, 등변 모델은 단일 섹터 안전 인증서를 얻었으나, 건강한 비등변 베이스라인은 위반, 여유 승수($\gamma_\alpha \approx 1.9$), 또는 날카로움 붕괴라는 비용을 치러야 했다.
• 계획 적용 (부정적 결과): 저자들은 인증된 지평을 3D CEM-MPC 플래너의 서브골 간격(subgoal spacing)을 가이드하는 데 사용하려고 시도했으나, 성공하지 못했다. 계획 기질은 시드 전반에 걸쳐 높은 분산을 보였으며, 인증된 신뢰 지평과 최적의 계획 지평 사이에는 감지 가능한 연관성이 없었다. 저자들은 이를 "상태 충분성 병목 현상(state-sufficiency bottleneck)"으로 설명하며, 잠재 표현이 중력 파괴 환경에서의 역학 예측에 필요한 속도 정보를 결여하고 있다고 분석했다.

5. 의의 및 주장

본 논문은 **"등변성은 캘리브레이션을 대체하는 것이 아니라, 캘리브레이션에 기하학을 부여하는 것"**이라고 주장한다.

• 구조적 vs 경험적: 궤도 전송 정리는 정확한 등변성이 있을 때 성립하는 구조적 보증이다. 그러나 경험적 이점(감사 비용 감소)은 데이터 분포가 비등변 베이스라인에 대해 이미 근사적인 궤도 강건성을 유도하지 않을 때만 나타난다.
• 보증의 성격: 이 인증서는 **보수적인 분포 감사(distributional audit)**이지, 전역적인 도달 가능성(reachability) 보증이 아니다. 이는 모델이 가능한 모든 최악의 시나리오에서 안전함을 보증하는 것이 아니라, 측정된 감사 분포에 대해 신뢰할 수 있음을 인증하는 것이다.
• 계획에 대한 겸손함: 저자들은 "인증된 서브골 간격" 적용이 확인되지 않았음을 명시적으로 밝힌다. 3D 플래너에서의 부정적 결과는 경계 탐색 결과로서 보고되었으며, 이는 등변 모델의 구조-분산 경제학(높은 계획 성능 분산)이 다운스트림 제어 계층의 이론적 이점을 상쇄할 수 있음을 강조한다.

이 연구는 단순한 스칼라나 구간이 아닌 *지평 함수(horizon function)*를 캘리브레이션하고, 이를 군 궤도로 전송함으로써 로컬 감사를 궤도 유효 인증서로 변환한다는 점에서 기존 연구와 차별화된다. 저자들은 등변성이 예측 정확도나 계획 성능을 보편적으로 향 향상한다고 주장하는 대신, 인증 프로세스 자체의 기하학적 효율성에 집중한다.