엔비디아(www.nvidia.co.kr, CEO 젠슨 황)는 킹스칼리지런던(King’s College London)의 연구원들이 엔비디아의 캠브리지-1(Cambridge-1) 슈퍼컴퓨터 및 MONAI로 오픈소스 합성 뇌 이미지의 보고를 생성해 의료분야에서 인공지능(AI)을 가속화했다고 8일 밝혔다.

킹스칼리지런던의 연구원 겸 런던 AI 센터(London AI Center)의 CTO인 호르헤 카르도소(Jorge Cardoso)는 의료 연구원에게 무료로 제공되는 10만 개의 합성 뇌 이미지를 만들고 있다. 이는 치매, 노화 또는 모든 종류의 뇌 질환에 대한 이해를 가속화할 수 있는 보고이다.

카르도소는 교사, CTO, 기업가, MONAI 오픈소스 컨소시엄의 창립 멤버이자 의료 영상용 AI 연구원이다. 의료 영상용 AI 연구원의 역할에서 카르도소와 그의 팀은 AI를 사용해 인간 두뇌의 사실적인 고해상도 3D 이미지를 만드는 방법을 발견했다.

무료로 이용할 수 있는 뇌 이미지의 세계 최대 저장소라는 점을 감안하면 큰 기부다. UK 바이오뱅크(UK Biobank)는 현재 5만명 이상의 참가자로부터 얻은 여러 개의 뇌 이미지를 유지 관리하고 있으며 예상 비용은 1억 5000만 달러다.

이미지는 소비자 및 비즈니스 애플리케이션의 컴퓨터 비전에서 이미 널리 사용되는 합성 데이터의 의료분야에서 새롭게 부상하고 있는 지점을 나타낸다. 역설적이게도 이러한 필드는 수백만 개의 실제 이미지가 포함된 개방형 데이터 세트에도 접근할 수 있다.

대조적으로 의료 이미지는 환자의 개인 정보를 보호해야 하는 필요성을 감안할 때 일반적으로 대형 병원과 연관된 연구원만 사용할 수 있어 상대적으로 부족하다. 그럼에도 불구하고 의료 이미지는 병원이 제공하는 인구 통계를 반영하는 경향이 있으며 반드시 더 많은 인구가 필요한 것은 아니다.

새로운 AI 접근 방식의 다행스러운 부분은 주문에 따라 이미지를 만들 수 있다는 것이다. 여성의 두뇌, 남성의 두뇌, 늙은 두뇌, 젊은 두뇌, 질병이 있거나 없는 두뇌 등 다양하며, 필요한 것을 연결하면 생성된다.

이 작업은 슈퍼 소프트웨어를 실행하는 슈퍼컴퓨터가 필요했다. 의료분야의 획기적인 AI 연구에 전념하는 슈퍼컴퓨터인 엔비디아 캠브리지-1가 엔진이었다. 의료 영상을 위한 AI 프레임워크인 MONAI가 소프트웨어 연료를 제공했다. 이는 연구원들이 수백 가지의 실험을 실행하고 최고의 AI 모델을 선택하며, 이미지를 생성하기 위한 추론을 실행할 수 있도록 하는 합성 데이터용 AI 공장을 함께 만들었다.

엔비디아 DGX 슈퍼POD(SuperPOD)인 캠브리지-1에는 640개의 엔비디아 A100 텐서 코어 GPU(NVIDIA A100 Tensor Core GPU)가 포함돼 있으며, 각 GPU에는 1,600만 3D 픽셀로 구성된 팀의 방대한 이미지 중 하나 또는 두 개를 처리하기에 충분한 메모리가 있다.

MONAI의 빌딩 블록에는 도메인별 데이터 로더, 메트릭, GPU 가속 변환 및 최적화된 워크플로 엔진이 포함된다. 카르도소는 소프트웨어의 스마트 캐싱 및 다중 노드 확장은 작업을 최대 10배까지 가속화할 수 있다고 언급했다. 또한 그는 cuDNN과 "훨씬 더 빠르게 작업하는 데 도움이 된 전체 엔비디아 AI 소프트웨어 스택"을 높이 평가했다.

카르도소는 10만 개의 뇌 이미지를 호스팅하기 위해 국가 저장소인 헬스 데이터 영국 보건 데이터 연구소(Health Data Research UK)와 협력하고 있다. AI 모델도 활용 가능해 연구자는 필요한 모든 이미지를 만들 수 있다.

또 이 팀은 MRI, CAT 또는 PET 스캔과 같은 모든 의료 영상 모드에서 모델이 인체 해부학의 모든 부분에 대한 3D 이미지를 만드는 방법을 탐구하고 있다.