The WEST code for large-scale excited-state materials simulations

Resumo Técnico do Artigo do Código WEST

Enunciado do Problema
Simulações de primeiros princípios baseadas na teoria do funcional da densidade de Kohn-Sham (KS-DFT) são indispensáveis para propriedades do estado fundamental, mas frequentemente falham em descrever com precisão fenômenos de estados excitados críticos para a microeletrônica, óptica, catálise e tecnologias quânticas. Embora métodos como a TDDFT (DFT dependente do tempo) e a teoria de perturbação de muitos corpos (MBPT), especificamente a aproximação GW e a equação de Bethe-Salpeter (BSE), forneçam estruturas rigorosas para estados excitados, sua aplicação a ambientes de materiais realistas permanece computacionalmente proibitiva. Implementações convencionais dependem do cálculo explícito de um grande número de estados eletrônicos vazios (virtuais) e de grandes matrizes dielétricas, levando a um escalonamento computacional desfavorável com o tamanho do sistema. Além disso, a maioria dos códigos existentes carece de implementações eficientes de forças de estado excitado analíticas e acoplamentos não-adiabáticos (NACs), que são essenciais para investigar relaxações estruturais, processos de emissão e dinâmica eletrônica-nuclear acoplada. À medida que a ciência dos materiais avança para a computação de exaescala e requer conjuntos de dados de alta fidelidade para aprendizado de máquina, há uma necessidade urgente de software escalável e de código aberto capaz de lidar com sistemas de centenas a milhares de átomos sem sacrificar a precisão.

Metodologia e Arquitetura de Software
O código WEST aborda esses desafios por meio de um framework algorítmico unificado que evita o cálculo explícito de estados eletrônicos virtuais. Ele integra GW de frequência total, teoria de embutimento de defeito quântico (QDET), BSE e TDDFT dentro de um formalismo de pseudopotencial de ondas planas. As principais inovações metodológicas incluem:

• Evitação de Estados Vazios: O código emprega a técnica de Potencial de Eigen deção de Dieletrônico Projetivo (PDEP) para diagonalizar iterativamente a função de resposta densidade-densidade irredutível simetrizada. Isso permite o cálculo do blindagem dielétrica e da autoenergia sem somar sobre estados vazios ou inverter grandes matrizes dielétricas.
• Teoria de Perturbação da Matriz de Densidade (DMPT): Para BSE e TDDFT, o WEST reformula o problema usando DMPT. Esta abordagem elimina a necessidade de construir grandes Hamiltonianos em uma base elétron-buraco, reduzindo significativamente a complexidade computacional.
• Técnicas de Baixo Posto e Localização: O código utiliza representações de baixo posto da blindagem dielétrica e do intercâmbio exato. Ao transformar funções de onda ocupadas em funções de Wannier localizadas via algoritmo de Diagonalização Aproximada Conjunta de Matrizes (JADE), a avaliação de integrais de Coulomb é restrita a pares sobrepostos, reduzendo o número de integrais em uma ordem de magnitude.
• Embutimento Quântico (QDET): Para sistemas com estados multiconfiguracionais (ex: defeitos pontuais), o QDET particiona o sistema em um espaço ativo (tratado com métodos de função de onda correlacionada de alto nível como FCI ou SCI) e um ambiente (tratado no nível GW). Isso permite descrições precisas de estados localizados a uma fração do custo de um tratamento de muitos corpos completo.
• Forças Analíticas e NACs: O WEST implementa forças de estado excitado analíticas e vetores de acoplamento não-adiabático usando um formalismo de Lagrange estendido e o método Z-vector. Estes são acelerados pela combinação do método de Intercâmbio Comprimido Adaptativamente (ACE), localização de Wannier e técnicas de subespaço de Krylov inexatas.
• Paralelização e Hardware: O código apresenta uma estratégia de paralelização hierárquica que distribui autovetores, canais de spin, bandas e coeficientes de onda plana. Ele suporta aceleração por GPU via diretivas OpenACC, aproveitando bibliotecas otimizadas por fornecedores (cuFFT, cuBLAS) e comunicações MPI não-bloqueantes para alcançar um escalonamento forte quase ideal em milhares de GPUs.
• Interoperabilidade: O WEST é projetado com modularidade, interfaceando-se com pacotes Python (PySCF, Qiskit, PyRET, PyPL, PyFRSOC, PyEPFD) para permitir fluxos de trabalho automatizados para computação quântica, acoplamento vibrônico e cálculos de acoplamento spin-órbita.

Principais Contribuições e Resultados
O artigo apresenta o WEST como uma plataforma escalável para simulações preditivas de estados excitados, validada através de aplicações representativas em diversas classes de materiais:

1. Defeitos de Spin em Estado Sólido (NV⁻ em Diamante): O código caracterizou com sucesso o ciclo óptico completo do centro de nitrogênio-vacância. Computou energias de excitação vertical e adiabática, linhas de zero-fônon e taxas de decaimento radiativo/não-radiativo (incluindo interconversão de sistemas e conversão interna) usando uma combinação de G0W0, QDET, BSE e TDDFT. Os resultados demonstraram consistência entre os métodos e concordância com tempos de vida e espectros experimentais, incluindo a previsão de primeiros princípios do tempo de vida do estado ¹A₁.
2. Escalabilidade para Grandes Sistemas: O WEST permitiu o estudo de efeitos de tamanho finito nas energias de excitação vertical para o centro de vacância de silício em diamante, revelando dependências lineares no tamanho da supercela para estados de triplet. Também simulou defeitos de spin próximos a deslocações mesoscópicas no diamante (1.727 átomos), demonstrando que as deslocações podem servir como andaimes para montar arrays de qubits de spin com propriedades de coerência comparáveis ou melhoradas.
3. Perovskitas de Metal-Haleto: O código investigou excitons auto-aprisionados (STEs) em Cs₄SnBr₆ e Cs₂AgInCl₆. Combinando efeitos excitônicos explícitos com relaxações de geometria de estado excitado, o WEST reproduziu com precisão os espectros de emissão de banda larga e desvios de Stokes, destacando a necessidade de relaxar a superfície de energia potencial do estado excitado.
4. Água e Gelo: O WEST foi usado para computar espectros de absorção óptica para água líquida e gelo Ih, mostrando concordância notável com o experimento. Também investigou como defeitos (vacâncias, iônicos e defeitos de Bjerrum) modificam a fotofísica do gelo, revelando estados de gap que deslocam para o vermelho as energias de absorção e emissão.
5. Plataformas Emergentes: O código foi aplicado a defeitos de spin em óxidos (MgO, CaO), nitretos (AlN), materiais 2D (hBN) e qubits moleculares, demonstrando a versatilidade do funcional SE-RSH e do QDET em descrever ambientes dielétricos heterogêneos e estados fortemente correlacionados.

Significância e Alegações
O artigo afirma que o WEST estabelece um novo padrão para simulações de materiais de estado excitado em grande escala ao remover os gargalos computacionais associados a estados vazios e grandes matrizes dielétricas. Sua principal significância reside na capacidade de realizar simulações de primeiros princípios precisas de sistemas contendo mais de mil átomos, uma escala anteriormente inacessível para métodos como GW e BSE. Ao fornecer um framework unificado que computa energias de quase-partícula, energias de excitação, espectros ópticos, forças de estado excitado e acoplamentos não-adiabáticos, o WEST facilita o estudo de materiais complexos e heterogêneos em ambientes experimentalmente relevantes.

Os autores enfatizam que a natureza de código aberto do código, seu design modular e sua interoperabilidade com ecossistemas de computação quântica e aprendizado de máquina o posicionam como uma ferramenta crítica para a descoberta de materiais de alto rendimento e para a geração de conjuntos de dados de alta fidelidade. A aplicação bem-sucedida a diversos sistemas — desde defeitos de spin em diamante até STEs em perovskitas e defeitos no gelo — demonstra a precisão e versatilidade do código, pavimentando o caminho para simulações preditivas de processos de estado excitado em materiais complexos em escalas sem precedentes.