Towards building a World Model to simulate perturbation responses in cellular biology

Motivation

• Known: 기존 세포 섭동 예측 모델들(scGen, CPA, GEARS, CellOT, scGPT 등)은 고가변성 유전자(highly variable genes)만 사용하거나, 제한된 특성 표현에 의존하거나, 맥락별 일반화 능력이 부족하다는 한계를 가진다.
• Gap: 기존 방법들은 (1) 불완전한 잠재 표현: HVGs 제한으로 인한 정보 손실, (2) 생물학적 재구성 왜곡: 강력한 디코더 부재로 인한 할루시네이션, (3) 동적 이전성 부족: 맥락별 일반화 불가로 인한 영점 사격 예측 실패 등 세 가지 구조적 결함이 있다.
• Why: 섭동 응답 예측은 치료법 발견에 핵심적이지만, 실험 스크리닝은 생물학적 공간의 조합론적 광대함으로 인해 제약받는다. 계산 시뮬레이션은 확장 가능한 대안이지만 기존 모델들은 일반화 능력이 부족하여 미관찰 세포 맥락에서 동역학을 전달할 수 없다.
• Approach: AlphaCell은 세 개의 상호 연동하는 구성요소로 설계된다: (1) Base Model Encoder가 전체 단백질 코딩 전사체를 처리하여 Virtual Cell Space로 투영, (2) Base Model Decoder가 잠재 상태를 전사체 프로필로 복원, (3) Flow Model이 Optimal Transport Conditional Flow Matching으로 연속 벡터장 모델링.

Achievement

Figure 4: Performance comparison of AlphaCell against state-of-the-art baselines (including STATE, scGPT,

• Latent Manifold Rectification: 전체 유전체를 사용하여 미분 가능한 Virtual Cell Space 구성, 기술적 노이즈 필터링과 고유 세포 위상 보존
• Biological Reality Reconstruction: 대규모 지식-풍부 디코더로 고충실도 유전체 전역 발현 프로필 재구성
• Universal State Transition: Optimal Transport Conditional Flow Matching으로 결정론적 벡터장 학습, 섭동을 이산 점프에서 연속 흐름으로 추상화
• 성능: 기준선 모델들을 능가하는 조성 일반화 및 영점 사격 예측 성능 달성

How

Figure 2: Architecture and training paradigm of the AlphaCell Base Model for Virtual Cell Space construction.

• 전체 단백질 코딩 전사체를 입력으로 사용하여 정보 완전성 확보
• 거대한 디코더로 생물학적 현실로의 고충실도 변환 보장
• Optimal Transport Conditional Flow Matching으로 세포 상태의 비선형 진화를 수학적으로 모델링
• 맥락 독립적 동역학 법칙 추상화로 다양한 세포 상황 간 전이성 확보

Originality

• World Model 패러다임: 단편화된 분석 도구에서 통합 World Model로의 패러다임 전환 제시
• 전체 유전체 처리: 제한된 유전자 집합 대신 전체 단백질 코딩 전사체 활용
• Optimal Transport Conditional Flow Matching 적용: 세포 상태 전이를 결정론적 연속 흐름으로 모델링하는 물리 기반 접근
• 조성 일반화와 영점 사격: 학습 분포를 벗어난 미관찰 맥락에 대한 동역학 전달 가능성 입증

Limitation & Further Study

• 데이터 규모: 훈련에 사용된 단일 세포 데이터셋의 크기와 다양성에 대한 상세한 설명 부재
• 계산 복잡성: 전체 유전체 처리 및 거대한 디코더로 인한 계산 비용 및 확장성 논의 부족
• 생물학적 검증: 시뮬레이션된 동역학의 실험적 검증 부재, 예측이 실제 세포 행동과 일치하는지 확인 필요
• 한계: 복잡한 다중 섭동, 시간에 따른 동적 반응, 세포간 이질성에 대한 모델의 성능 미평가

Evaluation

총평: AlphaCell은 세포 생물학 계산 패러다임에 중요한 진전을 제시하는 잘 설계된 World Model으로, 기존 방법의 근본적 결함을 체계적으로 해결하고 혁신적 기술 구성요소로 인한 우수한 일반화 성능을 보여준다. 다만 실험적 검증과 계산 복잡성에 대한 더 깊은 논의가 필요하다.