척추측만증 수술 후 방사선 사진 합성 기술 ViT3Flow 개발
ViT3Flow: A Test-Time Training Transformer MeanFlow for Postoperative Radiograph Synthesis in Scoliosis
arXiv척추측만증 수술 계획을 위해 수술 전 방사선 사진을 바탕으로 수술 후의 척추 형태를 예측하는 새로운 방법 ViT$^{3}$Flow가 개발되었습니다.
- 이 기술은 수술 교정을 유한 구간 생성 운송으로 모델링하고, 환자별 해부학적 특징에 적응하는 토큰 믹서와 구조 추출 과정을 결합했습니다.
- 기존 방식 대비 시각적 품질과 해부학적 충실도 모두에서 최고 성능을 달성했으며, 단일 네트워크 평가만으로 높은 신뢰성을 제공합니다.
- 본 연구는 최초의 전후 방사선 사진 데이터셋 ScoliSurg를 활용했기에, 실제 임상 환경에서의 추가적인 검증이 필요할 수 있습니다.
수술 전 흉부 방사선 사진을 바탕으로 수술 후의 척추 형태를 예측하는 것은, 수술 교정 과정에서 큰 공간적 변화가 발생하지만 해부학적 구조는 충실하게 유지되어야 하므로 어려운 과제입니다. 본 연구는 이 문제를 수술 후 척추측만증 방사선 사진 합성 문제로 정의하고, 이러한 작업을 위한 최초의 쌍을 이루는 데이터셋인 ScoliSurg를 구축했습니다. 이 데이터셋은 구조화된 형태학적 정보를 포함하는 632쌍의 전후(preoperative--postoperative) 전체 척추 방사선 사진으로 구성되어 있습니다.
제안된 ViT$^{3}$Flow는 단일 NFE 조건부 MeanFlow 프레임워크를 통해 효율적인 수술 후 방사선 사진 합성을 수행합니다. 이 모델은 수술 교정을 유한 구간 생성 운송(finite-interval generative transport)으로 모델링하며, 기존의 셀프 어텐션 메커니즘을 테스트 시점 훈련 믹서로 대체하여 각 사례별 해부학적 구조와 변형 패턴에 맞는 적응을 수행합니다. 또한, Spinal Morphology Extraction Agent가 전후 방사선 사진에서 지배 곡선 영역과 방향 분포를 추출하고, 이 정보는 Diagnosis-Routed Interval Cross-Attention (DRICA)을 안내하여 공간적으로 대응하는 해부학적 증거를 통합하도록 설계되었습니다.
ScoliSurg 데이터셋에 대한 광범위한 실험 결과, ViT$^{3}$Flow가 지각 이미지 품질, 해부학적 충실도, 그리고 임상적으로 관련된 기하학적 정확성 측면에서 비교된 방법들 중 가장 우수한 성능을 달성했습니다. 이 기술은 단일 네트워크 평가만으로 높은 신뢰성을 제공하며, 척추측만증 수술 계획에 효율적이고 해부학적으로 충실한 수술 후 방사선 사진 합성의 잠재력을 입증했습니다.
용어 풀이
- 토큰
- AI 모델이 글을 처리하는 단위. 단어보다 작은 조각이며 사용량과 요금을 셀 때 기준이 돼요.