TripleSurv 시간 적응형 삼중 좌표 손실을 이용한 생존 분석
본 논문은 기존 순위 손실과 최대우도 손실의 한계를 보완하기 위해, 생존 시간 차이를 반영한 시간‑적응형 쌍(pair) 순위 손실(T‑APR‑loss)을 도입하고, 이를 likelihood 손실 및 캘리브레이션 손실과 결합한 삼중 좌표 손실(TripleSurv)을 제안한다. 실험 결과, 제안 방법은 다양한 실제 및 합성 데이터셋에서 C‑index, Brier Score, TD‑AUC 등 주요 지표에서 기존 최첨단 모델들을 능가하며, 높은 검열 …
저자: Liwen Zhang, Lianzhen Zhong, Fan Yang
본 논문은 생존 분석에서 흔히 직면하는 두 가지 핵심 문제—순위 손실이 실제 생존 시간의 양적 차이를 무시한다는 점과 최대우도(MLE) 기반 손실이 검열 데이터와 이상치에 민감하다는 점—을 동시에 해결하고자 한다. 이를 위해 저자들은 “시간‑적응형 삼중 좌표 손실(TripleSurv)”이라는 새로운 손실 프레임워크를 제안한다.
첫 번째 단계는 기존 순위 손실을 확장한 시간‑적응형 쌍(pair) 순위 손실, 즉 T‑APR‑loss를 도입하는 것이다. 전통적인 순위 손실은 위험 점수의 순서만 맞추면 되지만, 실제 임상 상황에서는 두 환자 사이의 생존 시간 차이가 중요한 정보를 제공한다. T‑APR‑loss는 위험 점수 차이 \((risk_i - risk_j)\) 에 실제 시간 차이 \((t_j - t_i)\) 를 가중치 \(\rho\) 로 곱해 보정함으로써, 모델이 “얼마나 차이가 나는가”까지 학습하도록 만든다. 이때 위험 점수는 “1‑평균 사망 시간” 형태로 정의되어, 확률 질량 함수 \(p_k\)와 연계된 구간 평균을 이용한다.
두 번째 단계는 T‑APR‑loss를 기존 MLE 기반 likelihood 손실과 캘리브레이션 손실과 선형 결합하는 것이다. 손실 함수는
\
원본 논문
고화질 논문을 불러오는 중입니다...
댓글 및 학술 토론
Loading comments...
의견 남기기