핵심 개념
다양한 데이터셋을 통합한 UniTraj 프레임워크를 통해 차량 궤적 예측 모델의 일반화 능력과 데이터 규모 확장이 모델 성능에 미치는 영향을 분석하였다.
초록
이 논문은 차량 궤적 예측 분야에서 데이터 도메인 간 일반화 능력과 데이터 규모 확장이 모델 성능에 미치는 영향을 분석하였다.
- 데이터 도메인 간 일반화 능력 분석:
- 다양한 데이터셋(nuScenes, Argoverse 2, WOMD)에서 학습한 모델(AutoBot, MTR, Wayformer)을 서로 다른 데이터셋에 적용하여 성능 저하를 확인
- WOMD 데이터셋으로 학습한 모델이 다른 데이터셋에서 가장 우수한 일반화 성능을 보였으며, 이는 WOMD 데이터셋의 다양성 때문으로 분석
- 데이터 규모 확장이 모델 성능에 미치는 영향:
- 3개 데이터셋(nuScenes, Argoverse 2, WOMD)을 통합한 대규모 데이터셋으로 모델을 학습한 결과, nuScenes 데이터셋에서 새로운 최고 성능을 달성
- 데이터 규모 확장에 따라 모델 성능이 지속적으로 향상되는 것을 확인
- 데이터셋 특성 분석:
- 데이터셋별 궤적 유형 분포와 Kalman 난이도 분포를 분석하여 일반화 능력과 데이터 규모 확장 효과를 설명
이 연구는 차량 궤적 예측 분야에서 데이터 도메인 간 일반화와 데이터 규모 확장의 중요성을 강조하며, UniTraj 프레임워크를 통해 이를 체계적으로 분석하였다.
통계
차량 궤적 예측 모델의 성능은 데이터셋 간 전이 시 크게 저하된다.
WOMD 데이터셋으로 학습한 모델이 다른 데이터셋에서 가장 우수한 일반화 성능을 보였다.
3개 데이터셋을 통합한 대규모 데이터셋으로 학습한 모델이 nuScenes 데이터셋에서 새로운 최고 성능을 달성했다.
인용구
"데이터 규모 확장에 따라 모델 성능이 지속적으로 향상되는 것을 확인"
"WOMD 데이터셋의 다양성이 모델의 우수한 일반화 성능을 설명"