RICK 2.0: GPU와 HeFFTe 기반 차세대 라디오 천문 이미지 파이프라인

초록

RICK 2.0은 HeFFTe 분산 FFT와 MPI‑I/O, 비균등 도메인 분할을 도입해 GPU와 CPU 모두에서 높은 확장성을 달성한다. MeerKAT와 LOFAR 실험 데이터를 통해 정확성과 성능을 검증했으며, 기존 버전에서 96%를 차지하던 통신 오버헤드를 크게 감소시켜 SKA‑Mid·Low 시대에 적합한 이미지 재구성 솔루션을 제공한다.

상세 분석

본 논문은 차세대 대용량 라디오 인터페라터 데이터를 처리하기 위한 RICK 2.0의 설계와 구현을 상세히 제시한다. 핵심 기술은 HeFFTe 라이브러리를 이용한 분산 Fast Fourier Transform이다. HeFFTe는 MPI 기반의 다중 노드 환경에서 GPU와 CPU 양쪽 모두에 최적화된 FFT를 제공함으로써, 기존 cuFFT·cuFFTMp에 비해 이식성과 성능을 동시에 확보한다. 특히 w‑stacking 그리더와 결합했을 때, w‑term 보정이 필요한 넓은 시야(FoV) 관측에서도 정확한 이미지 복원을 가능하게 한다.

통신 병목을 해소하기 위해 저자들은 두 단계의 혁신을 적용했다. 첫째, MPI‑I/O를 활용해 측정 세트(Measurment Set)를 병렬 읽고 결과 이미지를 병렬 쓰는 방식을 도입함으로써 파일 시스템 의존성을 낮추고 I/O 스케일을 향상시켰다. 둘째, 전통적인 1‑D 슬랩 기반 도메인 분할을 비균등 가우시안 분포로 재구성하였다. 이는 u‑v 평면에서 중앙 영역에 데이터가 집중되는 특성을 반영해, 각 MPI 랭크가 처리해야 할 그리드 포인트 수를 가중치에 따라 동적으로 할당한다. 결과적으로 모든 랭크가 비슷한 작업량을 갖게 되어 그리딩 단계와 FFT 단계에서의 부하 불균형이 크게 감소한다.

GPU 가속 측면에서는 CUDA와 HIP 양쪽을 지원하도록 설계했으며, NCCL을 이용한 GPU‑GPU 직접 통신을 유지한다. HeFFTe 기반 FFT는 GPU 메모리 내에서 직접 수행되며, 기존 구현에서 전체 그리드를 Allreduce로 교환하던 방식을 폐지하고, 필요한 슬랩만을 교환하도록 변경했다. 실험 결과, GPU 클러스터에서 2배 이상의 속도 향상을 보였으며, 특히 대규모 픽셀 해상도(>8192²)와 다중 주파수 플레인(>64) 상황에서 통신 비중이 5% 이하로 감소했다.

정확도 검증은 MeerKAT와 LOFAR 실제 관측 데이터를 사용했으며, 기존 RICK 1.x와 WSClean, DDFacet 등과 비교했을 때 잔차 이미지 품질이 동등하거나 약간 우수함을 확인했다. 에너지 효율 측면에서도 GPU 사용 시 CPU 대비 30% 이상 절감되는 결과가 보고되었다.

전반적으로 RICK 2.0은 고성능 컴퓨팅(HPC) 환경에서 라디오 천문 이미지 재구성 파이프라인이 직면한 메모리·통신·이식성 문제를 종합적으로 해결한 사례로, 향후 SKA‑Mid·Low와 같은 페타바이트 규모 데이터 스트림을 실시간에 가깝게 처리할 수 있는 기반을 제공한다.