The BiP-PRISM algorithm for fast and scalable core-loss STEM-EELS simulations

기술 요약: 빠르고 확장 가능한 코어 손실(Core-loss) STEM-EELS 시뮬레이션을 위한 BiP-PRISM 알고리즘

문제 제기
원자 해상도의 주사 투과 전자 현미경-전자 에너지 손실 분광법(STEM-EELS)에 대한 정량적 해석에는 코어 손실 전이 전후의 전자 프로브에 대한 동역학적 시뮬레이션이 필요하다. 기존의 전이 포텐셜 멀티슬라이스(transition-potential multislice) 알고리즘은 모든 프로브 위치와 비탄성 이벤트마다 탄성 전파를 반복해야 하므로, 규모가 큰 이질적인 시야(field of view)를 대상으로 할 때 계산 비용이 지나치게 높으며 규모에 따른 확장성이 떨어진다. PRISM(Plane-wave Reciprocal-space Interpolated Scattering Matrix) 알고리즘은 스캔 위치 전반에 걸쳐 프로브 형성 산란 행렬($S_1$)을 재사용함으로써 이러한 시뮬레이션을 가속화했으나, 특히 검출기 통합형 원소 맵(detector-integrated elemental maps)에 필요한 대규모 산란 행렬의 저장 및 출사 측 전파(exit-side propagation)와 관련하여 여전히 상당한 메모리 및 계산 병목 현상에 직면해 있다.

방법론
저자들은 프로브 형성 산란 행렬($S_1$)과 검출기 전파 산란 행렬($S_2$)을 모두 분할(partitioning)하는 방식으로 PRISM 프레임워크를 확장한 BiP-PRISM (Beam Partitioned PRISM) 알고리즘을 소개한다.

1. 빔 분할 (Beam Partitioning): 모든 조리개 빔을 전파하는 대신, 알고리즘은 (조리개의 육각형 링 하위 샘플링을 통해 선택된) "부모 빔(parent beams)"의 희소 집합 위로 산란 행렬을 투영한다. 전체 조리개는 **자연 이웃 보간법(natural-neighbor interpolation)**을 통해 이온화된 원자 근처에서 국부적으로 재구성된다.
2. 이중 행렬 분할 (Dual-Matrix Partitioning):
   • $S_1$ (프로브 형성): 부모 빔 상에서 계산되며 이온화 평면에서 재구성된다.
   • $S_2$ (검출기 측): 출사 전파를 나타내는 이 행렬은 출사 표면에서 이온화 평면까지의 어드조인트(adjoint) 멀티슬라이스 역전파를 통해 구축된다. 이 또한 희소한 부모 검출기 빔 집합 위로 분할된다.
3. 국부성 정리 (Locality Theorem): 본 논문은 전체 시뮬레이션 오차가 전적으로 **원자 상의 재구성 오차(on-atom reconstruction error)**에 의해 결정된다는 국부성 결과를 입증한다. 전이 포텐셜($H_{n0}$)은 이온화된 원자 주변에 매우 날카롭게 국부화되어 있으므로, 이 작은 창 외부에서의 프로브 또는 검출기 재구성 오차는 최종 강도(intensity)에 영향을 미치지 않는다. 이를 통해 행렬을 전체 그리드가 아닌 각 원자 주변의 작은 "크롭 윈도우(crop window)"에서만 재구성할 수 있다.
4. 크기 보존 재구성 (Magnitude-Preserving Reconstruction): 복소 자연 이웃 보간의 결맞음 해제(decoherence)로 인한 진폭 손실 문제를 해결하기 위해, 저자들은 크기 보존 재구성 기술을 도입했다. 이는 완전한 부모 한계(full-parent limit) 내에서 근사 오차를 도입하지 않으면서도, 정량적 매핑에 필수적인 정확한 절대 강도 스케일을 복원한다.
5. 초점 역전파 (Focal Back-Propagation): 프로브 형성 행렬($S_1$)의 경우, 위상 확산을 최소화하여 재구성 오차를 줄이기 위해 부모 열(column)들을 보간 전 산란 중심 평면으로 역전파한다.

주요 기여

• 이중 분할 (Double Partitioning): EELS 맥락에서 프로브 형성($S_1$) 및 검출기($S_2$) 행렬 모두에 빔 분할을 적용한 첫 사례이다.
• 메모리 감소: $S_2$를 분할함으로써 알고리즘은 종종 메모리 소비의 주범이 되는 전체 검출기 행렬을 저장할 필요를 없앤다. 이를 통해 기존에는 메모리 제한으로 불가능했던 시뮬레이션을 소비자급 GPU에서 수행할 수 있게 한다.
• 스캔당 출사 전파 제거: 검출기 통합형 원소 맵의 경우, 알고리즘은 스캔 위치에 대한 단일 행렬-벡터 곱으로 대체함으로써 스캔당 출사 파동 전파 루프를 완전히 제거한다.
• 통합 출력 모드: 이 프레임워크는 원소 맵, 운동량 분해 qEELS, 그리고 전체 4D 큐브의 처리를 통합한다. 원소 맵과 qEELS는 이중 분할($S_1 + S_2$)의 이점을 얻는 반면, 전체 4D 큐브는 전체 검출기 기저가 필요하므로 $S_1$ 분할만을 사용한다.
• 유효 영역 특성화: 저자들은 알고리즘의 유효 영역을 정의하며, 오차는 주로 시편 두께(채널링으로 인해)에 의해 결정되며 디포커스(defocus) 및 가장자리 깊이 변화에 대해 견고함을 보여준다.

결과
저자들은 Bi-PRISM을 기존의 멀티슬라이스(그라운드 트루스) 및 정확한 듀얼 행렬 PRISM(정확한 참조 모델)과 벤치마킹하였다:

• 메모리 효율성: 23,000개의 원자로 구성된 FePt 나노입자 시뮬레이션에서, 이중 분할 알고리즘은 피크 GPU 메모리를 12.7 GB(정확한 듀얼 행렬 방식)에서 2.6 GB로(약 5배 감소) 줄였다. 19,082 그리드의 전체 해상도에서 정확한 방식은 48 GB 이상의 GPU RAM을 초과하여 메모리 부족(OOM)이 발생했으나, BiP-PRISM은 12.6 GB에서 성공적으로 실행되었다.
• 정확도: 9,362 그리드 해상도에서 이중 분할 맵은 정확한 참조 모델과 0.999의 피어슨 상관계수 및 **9.3%**의 상대적 L2 오차를 달성했다. 크기 보존 재구성이 절대 강도를 복원하는 데 필수적임을 보여주었으며, 이를 통해 오차를 (보정 없을 시) 최대 2배 수준에서 약 10% 이내로 줄였다.
• 확장성: 알고리즘은 시편 두께에 따라 선형적으로 확장되며 스캔 위치의 수에 따라 거의 선형적으로 확장됨을 보여주었으며, 이는 기존 멀티슬라이스의 이차 함수적(quadratic) 스케일링과 대조된다.
• 다중 모드 시연: 이 방법은 LaAlO$_3$/SrTiO$_3$ 계면의 동시 5개 엣지(Sr-L, La-M, Ti-L, Al-K, O-K) 원자 해상도 맵을 성공적으로 시뮬레이션하였으며, 피크 메모리 사용량은 단 2.4 GB였다.

의의 및 주장
본 논문은 BiP-PRISM이 기본적으로 메모리 가용성을 위한 가속기라고 주장하며, 이를 통해 이전에는 메모리 제약으로 불가능했던 소비자급 하드웨어에서의 전체 해상도, 대규모 시야 STEM-EELS 시뮬레이션을 가능하게 한다. 이 방법은 행렬 구축 단계에서 상당한 속도 향상을 제공하지만, 전체적인 속도 향상은 비통합 출력(4D 큐브와 같은)에 필요한 출사 전파에 의해 암달의 법칙(Amdahl-limited)의 제한을 받는다.

저자들은 부모 빔의 수가 온-아톰 재구성 오차 지표를 충족하도록 조정될 때 알고리즘이 높은 정량적 정확도(패턴 충실도 및 절대 강도 모두)를 유지한다고 강조한다. 이 방법은 퍼스트 원칙(first-principles) 기반의 전이 포텐셜 시뮬레이션과 운동량 분해 EELS를 일반적인 GPU에서 수행할 수 있게 함으로써, 이론적 시뮬레이션 능력과 실험 데이터 규모 사이의 간극을 메우는 실용적인 도구로 제시된다. 논문은 이 근사법이 제어 가능하며, 부모 빔의 수가 전체 빔 수에 접근함에 따라 정확한 PRISM으로 수렴한다고 결론짓는다.