Conceptos Básicos
본 논문에서는 개별적인 의사 결정에 중요한 조건부 보정의 중요성을 강조하며, 이를 개선하기 위해 conformal prediction 기반의 새로운 방법론인 CSD-iPOT를 제안합니다.
본 연구는 기존 생존 분석 모델의 한계점인 조건부 보정 문제를 해결하고자 합니다. 특히, 특정 특징을 가진 그룹에 대한 예측 정확도를 향상시키는 데 중점을 둡니다.
본 논문에서는 conformal prediction 기반의 새로운 방법론인 **CSD-iPOT (Conformalized Survival Distribution using Individual Survival Probability at Observed Time)**를 제안합니다. CSD-iPOT는 개별 생존 함수의 관측된 시간에서의 생존 확률을 conformity score로 활용하여 보정된 생존 분포를 생성합니다.
CSD-iPOT는 크게 세 가지 단계로 구성됩니다.
연속적인 ISD 예측: 먼저, 훈련 데이터셋을 훈련 세트와 conformal 세트로 분할합니다. 훈련 세트를 사용하여 임의의 생존 알고리즘 또는 분위수 회귀 알고리즘을 통해 모델 M을 훈련시키고, 이를 기반으로 conformal 세트에 대한 ISD 예측값을 생성합니다.
conformity score 계산: 각 개체에 대해 관측된 시간에서의 예측된 생존 확률을 나타내는 iPOT (predicted Individual survival Probability at Observed Time) 값을 계산합니다. censored 개체의 경우, 관측된 시간이 실제 사건 시간의 하한선이므로, iPOT 값은 ˆSM(ei | xi)의 상한선으로 사용됩니다. 이를 바탕으로 균등 분포에서 여러 개의 잠재적인 conformity score를 추출합니다.
예측 분포 보정: conformal 세트에서 계산된 conformity score의 경험적 분포를 기반으로 테스트 개체에 대한 예측된 ISD를 재보정합니다. 이를 위해 각 백분위수 수준에 대해 conformal 세트에서 해당 백분위수에 대응하는 시간을 찾고, 이를 기반으로 테스트 개체의 ISD 곡선을 조정합니다.