[논문 리뷰] The FHI-aims Code: All-electron, ab initio materials simulations towards the exascale

Volker Blüm, Mariana Rossi외 2명|arXiv (Cornell University)|2022. 08. 25.

Machine Learning in Materials Science인용 수 10

한 줄 요약

이 논문은 수치 원자 중심 궤도함수(NAOs)를 기반으로 한 고정밀 전자구조 계산 코드인 FHI-aims를 제시한다. 이 코드는 엑사스케일 컴퓨팅을 위해 설계되었으며, 핵전자를 완전히 고려한 전자전용(ab initio) 계산을 가능하게 하며, 다스케일 및 인공지능 기반 모델링 워크플로우를 지원한다. 이로 인해 다양한 물질 시스템에서 높은 병렬 처리 효율성과 정확도를 달성할 수 있다.

ABSTRACT

FHI-aims is a quantum mechanics software package based on numeric atom-centered orbitals (NAOs) with broad capabilities for all-electron electronic-structure calculations and ab initio molecular dynamics. It also connects to workflows for multi-scale and artificial intelligence modeling.

연구 동기 및 목표

• 엑사스케일 HPC 환경에서 사용 가능한 확장 가능한 전자전용 ab initio 전자구조 코드를 개발하는 것.
• 수치 원자 중심 궤도함수(NAOs)를 사용하여 핵전자와 비가역 전자를 완전히 고려한 고정밀 물질 시뮬레이션을 가능하게 하는 것.
• 물질 설계를 위한 다스케일 및 인공지능 기반 모델링 워크플로우에 FHI-aims를 통합하는 것.
• 다양한 물질 및 화학 환경에서 높은 병렬 처리 효율성과 수치적 안정성을 확보하는 것.

제안 방법

• FHI-aims는 체계적인 기저함수 완비성과 전자전용 계산에서의 고정밀도를 제공하는 수치 원자 중심 궤도함수(NAOs)를 기저 함수로 사용한다.
• 국소화된 기저함수를 사용하는 선형 스케일링 접근법을 적용하여 대규모 병렬 아키텍처에서 효율적인 계산을 가능하게 한다.
• 핵전자와 비가역 전자를 완전히 고려한 전자전용 케논-샤무스 밀도함수이론(DFT)을 구현한다.
• 적응형 시간단계와 에너지 보존 기능을 갖춘 ab initio 분자역학(AIMD) 시뮬레이션을 지원한다.
• 표준화된 인터페이스를 통해 외부 워크플로우와 통합하여 다스케일 모델링 및 머신러닝 응용을 지원한다.
• 최적화된 메모리 액세스 패턴, 로드 밸런싱 및 통신 회피 알고리즘을 통해 엑사스케일 준비성을 달성한다.

실험 결과

연구 질문

• RQ1 어떻게 엑사스케일 HPC 환경에서 전자전용 ab initio 시뮬레이션을 확장 가능하고 효율적으로 만들 수 있는가?
• RQ2 수치 원자 중심 궤도함수를 사용한 대규모 물질 시뮬레이션에서 FHI-aims의 성능과 정확도는 어떠한가?
• RQ3 FHI-aims는 다스케일 및 인공지능 기반 물질 모델링 파ip라인에 어떻게 원활하게 통합될 수 있는가?
• RQ4 다양한 물질 및 화학 환경에서 높은 병렬 처리 효율성과 수치적 안정성은 어느 정도 달성될 수 있는가?
• RQ5 FHI-aims는 엑사스케일 하드웨어 자원으로 확장되면서도 높은 정확도를 유지할 수 있는가?

주요 결과

• FHI-aims는 대규모 HPC 시스템에서 높은 병렬 처리 효율성을 확보하여 수천 개의 CPU 코어까지 강력한 스케일링 성능을 보였다.
• 체계적인 기저함수 수렴을 통해 전자전용 정확도를 제공하여 전자적, 구조적, 열역학적 성질에 대한 신뢰할 수 있는 예측이 가능하다.
• 수치적 결과는 전이 금속 산화물 및 복잡한 반도체를 포함한 다양한 물질에서 실험 데이터 및 기준 방법과 뛰어난 일치를 보였다.
• FHI-aims가 다스케일 및 인공지능 워크플로우에 통합되어 물질 발견과 성질 예측을 가속화할 수 있었다.
• 코드의 설계는 에너지 보존과 장기간의 수치적 안정성을 확보한 장기적인 ab initio 분자역학 시뮬레이션을 지원한다.
• FHI-aims는 엑사스케일 시스템에서 생산용으로 사용 가능하며, 최적화된 메모리 액세스 및 통신 패턴을 통해 현대 HPC 아키텍처 간의 성능 이식성을 보장한다.