조직병리학에서 암 비율 추정을 위한 다중 인스턴스 학습(MIL) 회귀 적용 방법

이 요약은 AI가 원문을 분석해 생성했습니다. 정확한 내용은 원문 기준으로 확인하세요.

핵심 요약

DINOv2 특징 추출기와 MIL 프레임워크를 결합하여 조직병리 이미지 내 암의 면적 비율을 정량적으로 예측하는 회귀 모델 구현 방안을 논의한다.

사용자가 DINOv2를 특징 추출기로 사용하고 MIL로 특징을 통합하는 조직병리학 프로젝트를 진행 중이며, 기존의 분류 방식 대신 암의 면적 비율을 예측하는 회귀 방식으로 MIL을 확장하는 방법을 문의했다.

MIL은 조직병리학에서 단순 분류를 넘어 정량적 분석인 회귀 작업으로 확장될 수 있는 유연한 구조를 가졌다. DINOv2와 같은 최신 비전 모델과의 결합은 의료 데이터의 약한 지도학습 문제를 해결하는 표준적인 기술 경로로 자리 잡고 있다.

사용자의 접근 방식이 타당하다는 의견이 주를 이루며, 구체적인 아키텍처 변경 방법과 관련 연구 방향에 대한 관심이 나타났다.

01찬성다수

MIL은 본질적으로 특징 응집(Aggregation) 방식이므로 마지막 레이어만 수정하면 회귀 작업에 매우 적합하다.

Attention-based MIL의 마지막 분류 레이어를 Linear 레이어로 교체하고 MSE(Mean Squared Error) 손실 함수를 사용하여 학습시킨다.
DINOv2 가중치를 고정(Frozen)한 상태에서 MIL 어그리게이터만 먼저 학습시켜 베이스라인 성능을 확보한다.

DINOv2를 자기지도학습 기반의 특징 추출기로 활용하여 이미지 타일에서 고차원 벡터를 생성한 뒤 MIL로 통합하는 구조를 구축했다. 기존의 이진 분류(암 유무)를 넘어 암이 차지하는 면적 비율을 예측하는 회귀 문제로의 확장을 목표로 한다.

어텐션 기반 MIL(Attention-based MIL) 아키텍처에서 마지막 레이어를 회귀 헤드로 교체하는 구현 방식이 검토됐다. 각 타일의 중요도를 계산하는 어텐션 메커니즘을 유지하면서 최종 출력값을 연속적인 수치로 변환하는 과정이 핵심이다.

조직병리학 분야의 약한 지도학습(Weakly Supervised Learning) 환경에서 MIL의 유연성이 강조됐다. 슬라이드 전체의 레이블만 있는 상황에서도 어텐션 가중치를 통해 어떤 타일이 회귀 결과에 기여했는지 해석할 수 있는 가능성이 제시됐다.

DINOv2추천

자기지도학습 기반 이미지 특징 추출

Attention-based MIL추천

타일 특징 응집 및 중요도 가중치 계산