핵심 개념
본 연구는 최첨단 세포 AI 기반 모델을 사용하여 신장 병리학에서의 세포핵 분할 작업의 성능을 평가하고, 인간 참여 데이터 증강 전략을 통해 모델 성능을 향상시키는 방법을 제시합니다.
초록
인간 참여 데이터 증강을 통한 신장 병리학에서의 세포 AI 기반 모델 평가: 얼마나 뛰어난가?
본 연구는 최첨단 세포 AI 기반 모델을 사용하여 신장 병리학에서의 세포핵 분할 작업의 성능을 평가하고, 인간 참여 데이터 증강 전략을 통해 모델 성능을 향상시키는 방법을 제시합니다.
데이터셋 구축
다양한 공개 및 비공개 데이터셋에서 인간 및 설치류 조직에서 얻은 2,542개의 전체 슬라이드 이미지(WSI)로 구성된 대규모 신장 데이터셋 구축
H&E, PASM, PAS 염색을 사용하여 염색된 샘플 포함
오염되었거나 품질이 낮은 이미지 패치를 제거한 후 각 WSI에서 4개의 512x512 픽셀 이미지 패치를 무작위로 추출하여 총 8,818개의 이미지 패치로 구성된 평가 데이터셋 생성
데이터 증강 및 모델 미세 조정
Cellpose, StarDist, CellViT 세 가지 기반 모델을 사용하여 이미지 패치에서 세포핵 인스턴스 분할 수행
병리학자의 평가를 기반으로 예측 마스크를 "좋음", "보통", "나쁨"으로 분류
"좋음"으로 평가된 이미지 패치는 의사 레이블로 사용하고, "나쁨"으로 평가된 이미지 패치는 병리학자가 수동으로 수정하여 모델 미세 조정에 활용
"좋음" 이미지 패치만 사용, "나쁨" 이미지 패치만 사용, 두 가지 모두 사용 등 세 가지 데이터 증강 전략을 통해 모델 성능 향상 평가
성능 평가 지표
Recall, Precision, F1 점수를 사용하여 인스턴스 분할 성능 평가
Intersection over Union (IoU) 임계값을 0.5로 설정하여 True Positive (TP)로 간주