Multi Task 기반 확률 모델링으로 배달시간플랫폼을 구축한 DoorDash

Multi Task 기반 확률 모델링으로 배달시간플랫폼을 구축한 DoorDash
Photo by Marques Thomas / Unsplash

Overview

Multi Task Model

  • Multi Task(MT) Model 사용시 장점
    • 신규 서비스 배달시간 예측시, Foundation Layer에 추가적으로 레이어릅 붙이는 식으로 대응이 가능
    • 플랫폼 내부에서 고객의 여정에 따라 여러번 배달시간을 업데이트해줘야 하는데, 이 부분에 대해서 빠르게 대응 가능, 기존에 별도 모델 사용할 경우,모델간 시간예측 결과 일관성을 보장하기 어려웠음
    • 최대한 많은 Fature를 사용함으로써 예상하지 못할 불확실성을 파악하여 예측할 수 있음
    • 배달 대비 포장의 데이터는 적은 편인데, 여기서 야기되는 데이터 불균형을 해결하여 예측정확도를 높일 수 있음(Knowledge Transfer?)

Deep Learning

  • Numerical Feature 외에 Categorical Feature를 Embedding을 이용하여 활용할 수 있게 됨
  • Feature간 Interaction을 반영할 수 있게 되었음
  • DL의 특성인 유연성을 활용하여 다양한 목적으로 모델을 확장 및 응용이 가능해짐

Probabilistic Model

  • 확률기반 모델링을 가져감으로써 불확실성에 대해서 함께 고려함으로써 기존의 점추정 형태를 넘어 가능한 값 별 확률을 파악하고자 함
  • 아래 그림을 봐도 둘다 20분을 예측하지만 20분에 대한 Density가 다르고 분포 역시 Variance도 다르다.

Evaluation

  • Calibration
    • 모델의 예측 분포가 실제 결과와 밀접하게 일치하도록 보정하였음
  • Accuracy
    • 확률기반 모델이기 때문에,Continuous Ranked Probability Score (CRPS)을 활용$$\begin{equation*} \mathrm{CPRS}(F(|X|), x) = \int (F(y|X) - 1{y \geq x})^2 dy \end{equation*}$$
    • 위 식에서 볼 수 있듯이 실제 값과 예측값간의 차이를 좁히는 것이 핵심, 아래 그림에서 볼 수 있듯이 실제값의 분포와 예측값 간의 분포를 줄이는 것이 목표가 됨, 이를 통해서 모델에 점수를 매길 수 있게 되었음 → 목표로 활용 가능

실험결과

  • 다음 두가지 지표로 측정하였음
    • Accuracy: 예측 정확도(예측시간대에 배달이 제때 오는지)
    • Consistency: 가드레일 지표로 고객에게 노출한 배달시간의 어느정도 일관성을 갖고 있고, 크게 흔들리지 않은지를 확인

Reference

Read more

DataFrame은 Pandera로, 모델은 Pydantic으로 데이터를 검증한다.

DataFrame은 Pandera로, 모델은 Pydantic으로 데이터를 검증한다.

Pandera: 데이터프레임 검증에 최적화된 도구 주요 장점 * Pandas와 통합: Pandas 데이터프레임에 대해 스키마 기반 검증을 수행합니다. * 유연한 검증 조건: 열 데이터 타입, 값 범위, Null 여부 등 다양한 검증 조건을 정의할 수 있습니다. * 명확한 오류 메시지: 스키마 불일치에 대한 명확한 오류 메시지를 제공합니다. 단점 * 대용량 데이터 검증에서는 속도가 느릴 수

Tobit Regression은 Censored Data에 적합한 Regression이다.

Tobit Regression은 Censored Data에 적합한 Regression이다.

Tobit Regression * Tobit 회귀(Tobit Regression)는 종속 변수가 특정 값에서 절단(Censored)된 상황에서 데이터를 분석하기 위해 사용되는 통계 기법입니다. * James Tobin이 처음 제안한 이 모델은 경제학과 사회과학 분야에서 자주 사용되며, 일반 선형 회귀로는 설명할 수 없는 상황에서 효과적으로 적용할 수 있습니다. Tobit Regression 수식 1. 관측된 종속 변수