엔비디아 과학 이미지 분석 가속화 도구 cuPhoton 공개

주장엔비디아(NVIDIA)가 센서 데이터 읽기부터 분류까지 모든 이미지 처리 과정을 그래픽 처리 장치(GPU) 메모리 내에서 수행하는 오픈소스 쿠다 엑스(CUDA-X) 툴킷인 cuPhoton을 공개했습니다. 이 기술은 기존 중앙 처리 장치(CPU) 기반 파이프라인의 한계를 극복합니다. 과학적 분석 시간을 수개월에서 수분 단위로 단축합니다.

팩트cuPhoton은 FITS 로딩을 수행하는 엑스데이터리더(xDataReader), 이미지 정렬을 담당하는 엑스렙(xRep), 점상분포함수(PSF) 매칭 및 차분을 처리하는 엑스포이스(xPois) 모듈을 제공합니다. 또한 쌍극자 피팅을 수행하는 엑스핏(xFit), 후보 검토를 담당하는 엑스스캔(xScan), 시간 영역 엑스선 탐지기 분석을 위한 엑스레이(xRay) 모듈을 포함합니다.

팩트대표적인 워크로드에서 cuPhoton은 일반적인 중앙 처리 장치 대비 이미지 로딩 속도를 최대 1만4900배, 신호 처리 속도를 최대 1만4550배 가속했습니다. 대규모 과학 데이터 분석에서 발생하는 계산 병목 현상을 해소합니다.

교차검증제시된 최대 가속 수치는 개별 연산에 대한 성능을 측정한 결과입니다. 전체 파이프라인의 최종 속도 향상 폭은 데이터셋의 크기, 하드웨어 구성, 구현 방식에 따라 달라집니다. 실제 적용 시 워크로드 특성에 따른 성능 차이를 검토해야 합니다.

팩트미국 국립과학재단(NSF)과 에너지부(DOE)가 운영하는 베라 C. 루빈 천문대(Vera C. Rubin Observatory)의 엘에스에스티캠(LSSTCam)은 39초마다 3.2기가픽셀의 노출 데이터를 생성합니다. cuPhoton은 매일 발생하는 20테라바이트 규모의 이미지와 1000만 개의 후보 객체를 60초에서 120초 이내의 경고 창 시간 내에 처리합니다.

주장현대 과학 시설은 페타바이트 단위의 다차원 데이터를 생성합니다. 중앙 처리 장치 중심의 구현은 결과 도출에 수개월이 소요되어 연구 속도를 저해합니다. cuPhoton은 데이터가 생성되는 즉시 분석할 수 있는 그래픽 처리 장치 네이티브 경로를 제공하여 연구자들이 대화형으로 결과를 확인하도록 지원합니다.

팩트cuPhoton은 엔비디아 그레이스 블랙웰(Grace Blackwell) 및 베라 루빈(Vera Rubin) 시스템과 같은 다중 그래픽 처리 장치 및 다중 노드 환경에서 확장됩니다. 전체 이미지 쌍과 후보 좌표를 그래픽 처리 장치 워커 간에 분산 처리하면서도 배열 데이터를 장치 내에 유지합니다. 데이터 이동 비용을 최소화합니다.

팩트cuPhoton 0.1.3 버전은 리눅스 환경에서 파이썬 3.12부터 3.14 버전 및 쿠다 13을 지원합니다. 설치를 위해서는 씨플러스플러스17 컴파일러, 쿠다 및 씨유파일(cuFile) 개발 헤더, 씨피티에스아이오(CFITSIO) 개발 설치가 필요합니다. 공식 깃허브 저장소를 통해 내려받을 수 있습니다.

교차검증cuPhoton은 씨유파이(CuPy), 파이토치(PyTorch), 눔바 쿠다(Numba-CUDA), 퀴크아이오(KvikIO), 엔비디아 엔브이컴프(nvCOMP) 등 다양한 라이브러리와 연동되어 작동합니다. 사용자는 제공된 가이드를 통해 가상 환경을 설정하고 합성 데이터를 사용하여 워크플로우를 사전에 테스트할 수 있습니다.

출처엔비디아 기술 블로그(https://developer.nvidia.com/blog/faster-scientific-image-analysis-with-nvidia-cuphoton/)의 보도를 교차 검증했습니다.