Traj-CoA: Patient Trajectory Modeling via Chain-of-Agents for Lung Cancer Risk Prediction
Created by
Haebom
저자
Sihang Zeng, Yujuan Fu, Sitong Zhou, Zixuan Yu, Lucas Jing Liu, Jun Wen, Matthew Thompson, Ruth Etzioni, Meliha Yetisgen
Traj-CoA: 환자 궤적 모델링을 위한 Multi-Agent 시스템
개요
본 논문은 환자 궤적 모델링을 위해 대규모 언어 모델(LLM)의 일반화된 접근 방식을 제시하지만, 전자 건강 기록(EHR) 데이터의 길고 잡음이 많은 특성으로 인해 발생하는 문제점을 해결하고자 합니다. 이를 위해 Traj-CoA라는 multi-agent 시스템을 도입합니다. Traj-CoA는 일련의 작업자 에이전트를 사용하여 EHR 데이터를 순차적으로 처리하고, 중요한 이벤트를 EHRMem이라는 공유 장기 기억 모듈로 추출하여 노이즈를 줄이고 포괄적인 타임라인을 유지합니다. 최종 관리자 에이전트는 작업자 에이전트의 요약과 EHRMem에 추출된 타임라인을 종합하여 예측을 수행합니다. 5년간의 EHR 데이터를 기반으로 한 1년의 폐암 위험 예측 작업에서 Traj-CoA는 4가지 범주의 baseline을 능가합니다. 분석 결과, Traj-CoA는 임상적으로 일치하는 시간적 추론 능력을 보여주며, 복잡한 환자 궤적 모델링을 위한 유망하고 견고하며 일반화 가능한 접근 방식을 제시합니다.