Gaussian splatting for sparse-view X-ray phase-contrast reconstruction in MHz tomoscopy

기술 요약: MHz 토모스코피를 위한 희소 뷰 X선 위상 대비 재구성용 가우시안 스래팅(Gaussian Splatting)

문제 정의
MHz 토모스코피는 단일 자유 전자 레이저(XFEL) 펄스 내에서 여러 각도로부터의 동시 X선 투영을 포착함으로써 빠른 동적 현상(예: 충격파, 공동 현상, 생물학적 유체 역학)을 이미징할 수 있게 합니다. 그러나 이 방식은 심각한 데이터 제한 문제에 직면합니다. 현재의 XFEL 구현에서는 일반적인 컴퓨터 단층 촬영(CT)에 필요한 수백 개의 뷰와 비교하여 획득 가능한 각도 뷰의 수가 극도로 제한됩니다(통상 4개). 이는 3D 재구성 문제를 매우 과소 결정된(underdetermined) 상태로 만듭니다.

이 문제는 연질 생물학적 시료의 물리적 특성으로 인해 더욱 복잡해집니다. 경 X선 에너지 영역에서는 굴절률 감소량($\delta$)이 흡수 지수($\beta$)보다 몇 자릿수 더 높게 나타납니다. 표준 감쇠 기반 재구성 알고리즘은 이러한 위상 지배적 영역에서 대비를 생성하는 데 실패합니다. ONIX와 같은 기존의 희소 뷰 신경망 접근 방식은 오직 흡수 영역에서만 작동하며 $\delta$를 복구할 수 없는 반면, 전통적인 복셀 기반 반복 방법은 조밀한 각도 샘플링을 요구합니다. 따라서 압축적이고 빠르며 미분 가능하고, 위상 대비 물리 법칙에 충실한 장면 표현 방식이 절실히 필요합니다.

방법론
저자들은 실사적인 신규 뷰 합성(novel-view synthesis)을 위해 개발된 **3D 가우시안 스래팅(3DGS)**을 X선 위상 대비 이미징에 맞게 변형하였습니다. 핵심적인 수정 사항은 다음과 같습니다:

1. 장면 표현: 구형 조화 함수(spherical-harmonic) 색상 계수와 불투명도를 갖는 표준 3DGS 프리미티브(primitive)를 물리적 파라미터인 복소 굴절률 $n = 1 - \delta + i\beta$의 흡수 지수($\beta$)와 굴절률 감소량($\delta$)으로 대체하였습니다. 이 값들은 비음수(non-negative)로 제한됩니다.
2. 기하학 및 투영: 원근 투영(perspective projection)을 MHz 토모스코피와 일치하는 평행 빔 기하 구조로 교체하였습니다. 표준 RGB 알파 블렌딩 합성법은 각 광선(ray)을 따라 $\beta$와 $\delta$를 적분하는 2채널 누적 모델로 대체되었습니다.
3. 미분 가능한 순방향 모델: 핵심적인 혁신은 미분 가능한 프레넬 전파(Fresnel propagation) 모델의 통합입니다. 누적된 투과 맵은 검출기에서 기록된 근접장 회절 홀로그램을 재현하기 위해 이 모델을 통과하며, 이때 검출기 블러(blur)를 시뮬레이션하기 위해 가우시안 점 퍼짐 함수(PSF)가 포함됩니다. 이를 통해 전체 순방향 모델은 모든 가우시안 파라미터에 대해 미분 가능해지며, 명시적인 위상 복구(phase retrieval) 과정 없이 강도(intensity) 기반 측정값으로부터 기하 구조와 재료 특성을 그래디언트 기반으로 공동 최적화할 수 있게 합니다.
4. 훈련 파이프라인: 대규모의 주석이 달린 실험적 XFEL 데이터셋이 부족하기 때문에, 저자들은 물리 기반 시뮬레이션 파이프라인을 개발했습니다. 분자 역학 시뮬레이션(ESPResSo/PyOIF)을 사용하여 거친 입자 형태의 적혈구(RBC) 기하 구조를 생성하고, 이를 복셀화한 뒤 20 keV에서의 X선 광학 상수를 할당했습니다. 합성된 위상 대비 홀로그램은 GPU 가속 레이캐스팅(raycasting)과 프레넬 전파를 사용하여 생성되었습니다.
5. 최적화 전략: 훈련에는 픽셀 단위 L1 오차, 구조적 유사도(SSIM), 푸리에 링 상관관계(FRC) 항, 그리고 총 변동(Total Variation) 정규화를 결합한 복합 손실 함수가 사용되었습니다. 다중 뷰 그래디언트 누적 전략을 채택하여, 매 반복마다 여러 개의 동시 뷰로부터 손실을 평가함으로써 교차 각도 간의 기하학적 일관성을 강제하고 일치하지 않는 가우시안을 제거(prune)했습니다.

주요 결과
본 프레임워크는 12개의 훈련 투영과 4개의 테스트 뷰를 사용하여 시뮬레이션된 적혈구 홀로그램에 대해 테스트되었습니다.

• 성능: 최적의 구성(초기 가우시안 8,000개, 학습률 $2 \times 10^{-3}$, 4개 동시 뷰)은 SSIM 0.870과 FRC 메트릭(FRCM) 0.096을 달축했습니다. 재구성은 단일 NVIDIA A100 GPU에서 약 15분 만에 완료되었습니다.
• 메트릭 분석: 연구는 3DGS의 표준 벤치마크인 PSNR이 홀로그래피 데이터에 대해서는 오해의 소지가 있는 지표임을 강조했습니다. 244,000개의 가우시안을 사용한 구성은 가장 높은 PSNR(19.7 dB)을 기록했으나 가장 낮은 SSIM(0.449)을 보였습니다. 이 구성은 배경 강도는 일치시키지만 위상 신호를 인코딩하는 프레넬 무늬 구조를 재현하지 못하는 퇴보적인 "흡수 전용(absorption-only)" 솔루션으로 수렴했습니다. 반면, SSIM과 FRCM은 고주파 위상 정보를 보존하는 구성을 정확하게 식별해 냈습니다.
• 최적화 역학: 초기 가우시안 수를 8k에서 244k로 늘리는 것이 기본 학습률 하에서 오히려 구조적 품질을 저하시킨다는 역설적인 결과가 나타났습니다. 8k 초기화 상태에서 타겟팅된 학습률 튜닝을 적용한 것이 30배 많은 프리미티브 수를 가진 경우보다 우수한 성능을 보였으며, 이는 병목 현상이 표현 용량이 아닌 최적화 역학에 있음을 시사합니다.
• 재료 복구: 학습된 재료 필드는 질적으로 올바르게 나타났습니다. 즉, 위상 투영 맵($\delta$)이 지배적이고 흡수 맵($\beta$)은 0에 가깝게 유지되어 연질 조직의 물리적 특성과 일치했습니다. 다만, 가우시안당 $\delta/\beta$ 비율은 이론적인 헤모글로빈 값보다 낮았는데, 이는 강도 기반 측정으로부터 위상을 복구하는 과정의 본질적인 불량 설정(ill-posedness)을 반영합니다.

의의 및 주장
본 논문은 3D 가우시안 스래팅이 희소 뷰 X선 위상 대비 재구성을 위한 효과적이고 압축적이며 미분 가능한 표현 방식임을 입증합니다. 저자들은 입자 기반의 장면 묘사가 MHz 토모스코피의 데이터 볼륨 제약을 본질적으로 해결한다고 주장합니다. 또한, 미분 가능한 물리 기반 순방향 모델을 통해 빔타임(beamtime) 제약 시간 내에 최적화가 완료될 수 있는 근접 온라인(near-online) 재구성을 가능하게 합니다.

본 연구는 홀로그래피 또는 간섭계 데이터의 경우, 표준 지표인 PSNR이 기만적일 수 있으며 위상 정보의 복구를 평가하기 위해서는 SSIM, FRCM과 같은 구조적 메트릭이 필수적임을 보여줍니다. 저자들은 차기 우선 과제로 다중 분자 역학 타임스텝을 통한 4D 재구성 확장과, 4-동시 뷰 실험 구성으로부터 얻은 실제 XFEL 투영 데이터를 이용한 방법론 검증을 제시했습니다.