Efficient Large-Scale STEM-EELS Simulations With Torched-TACAW

Resumo Técnico: Simulações Eficientes de STEM-EELS em Grande Escala com torched-TACAW

Definição do Problema
A simulação de espectros de perda de energia eletrônica (EELS) de ultra-baixa perda, decorrentes de excitações vibracionais e de magnons em Microscopia Eletrônica de Varredura por Transmissão (STEM), enfrenta barreiras computacionais significativas quando aplicada a sistemas de materiais realistas e de grande escala. Embora o método da Autocorrelação Temporal de funções de onda Auxiliares (TACAW) ofereça um caminho para simular esses espectros aproveitando cálculos de dinâmica molecular (MD) e multislice (MS), a aplicação prática é dificultada por duas restrições primárias:

1. Tamanho do Sistema e Custo Computacional: Modelos realistas contendo defeitos, interfaces ou contornos de grão frequentemente exigem supercélulas com dezenas a centenas de milhares de átomos. Potenciais interatômicos de alta precisão padrão (como os potenciais aprendidos por máquina, como o ORB) são computacionalmente caros, muitas vezes excedendo a capacidade de memória das atuais GPUs de alto desempenho (limitando sistemas a ~25.000 átomos em uma GPU de 64 GB) e exigindo tempos de execução proibitivos em CPUs.
2. Vazão de Dados: Um cálculo TACAW completo envolve a geração de milhares de instantâneos (snapshots) de estrutura a partir de trajetórias de MD e a realização de propagações multislice para cada posição do feixe. Para um escaneamento típico, isso gera funções de onda auxiliares na ordem de terabytes (por exemplo, 29 TB para um escaneamento de 1 nm²), criando um gargalo de fluxo de dados que excede as capacidades de armazenamento e I/O de estações de trabalho padrão e complica o pós-processamento.

Metodologia
Os autores apresentam um fluxo de trabalho prático e uma nova implementação de software de código aberto, o torched-TACAW, projetado para superar esses gargalos mantendo uma precisão próxima à ab initio. A metodologia integra três componentes principais:

1. Potenciais Interatômicos Aprendidos por Máquina (MLIPs): O fluxo de trabalho utiliza modelos fundamentais, especificamente o potencial ORB, que oferece um equilíbrio entre precisidade comparável a cálculos ab initio e um custo computacional significativamente menor. Para gerenciar as restrições de memória, os autores empregam o particionamento de supercélula. Em vez de simular uma única supercélula massiva, a estrutura alongada (tipicamente em formato de "macarrão" ao longo da direção do feixe) é dividida em sub-supercélulas menores e independentes. Estas são simuladas separadamente via MD e depois empilhadas para reconstruir a trajetória efetiva do sistema maior.
2. Arquitetura de Software torched-TACAW: Implementado em Python utilizando a biblioteca PyTorch e o motor multislice pyms, o software foi projetado para aceleração por GPU. As principais características arquitetônicas incluem:
   • Processamento On-the-fly: Para evitar o armazenamento de terabytes de funções de onda intermediárias, o código processa dados em lotes computacionais. Ele retém apenas a Região de Interesse (ROI) no espaço recíproco e na energia, descartando o restante imediatamente após a transformação de Fourier.
   • Paralelização: O fluxo de trabalho é "embaraçosamente paralelo", permitindo o processamento independente de posições de feixe e blocos de trajetória através de múltiplas GPUs sem comunicação direta GPU-GPU, sendo sincronizado apenas via bloqueio de arquivo (file locking).
   • Gestão de Dados: Os resultados são armazenados eficientmente usando o formato Zarr, que suporta armazenamento em blocos (chunks) e compactação, adequado para grandes conjuntos de dados.
3. Refinamentos Numéricos: O artigo aborda artefatos numéricos inerentes às transformadas de Fourier de tempo finito:
   • Janelamento (Windowing): Os autores avaliam funções de janela (Tukey, Hann, Retangular) para suprimir o vazamento espectral. Eles demonstram que a janela de Hann fornece o compromisso ideal, eliminando eficazmente artefatos (como listras verticais nos pontos Gamma) sem excessiva dispersão espectral.
   • Estabilidade de MD: O fluxo de trabalho inclui protocolos para evitar artefatos não físicos, como o desvio do centro de massa (ao fixar o COM) e o "salto de átomos" através de fronteiras periódicas (ao garantir camadas de vácuo suficientes).

Principais Resultados
Utilizando TiO₂ rutilo como um sistema modelo, os autores validaram a metodologia:

• Precisão do Particionamento: Comparar uma simulação de supercélula completa de 24.576 átomos contra uma trajetória reconstruída de 8 sub-supercélulas independentes de 3.072 átomos revelou que o particionamento introduz apenas erros menores, do tipo ruído. As diferenças foram negligenciáveis para a maioria dos modos e particularmente insignificantes para o EELS de resolução atômica fora do eixo.
• Eficácia do Janelamento: As simulações confirmaram que a janela de Hann elimina os artefatos de listras verticais vistos com janelas de Tukey e a superestimação da cauda de alta energia vista com janelas retangulares, mantendo a resolução espectral.
• Demonstração de Larga Escala: Os autores simularam com sucesso uma amostra de TiO₂ de 192.000 átomos (24 nm de espessura, 9,3 nm × 9,3 nm de dimensões laterais). O cálculo, realizado em 16 núcleos de GPU, levou aproximadamente 3 horas e 40 minutos, produzindo um conjunto de dados de 48 GB que foi reduzido para 760 kB após a integração do detector.
• Observáveis Físicos: As imagens STEM filtradas em energia resultantes reproduziram contrastes físicos esperados: os sinais de baixa energia foram dominados pelas vibrações dos átomos de Titânio, enquanto os intervalos de maior energia e ângulos de espalhamento maiores mostraram um contraste pronunciado da coluna de Oxigênio, consistente com achados experimentais e teóricos anteriores.

Significância e Alegações
O artigo afirma que o torched-TACAW fornece uma plataforma escalável, eficiente e acessível para simular STEM-EELS de baixa perda em materiais complexos contendo até centenas de milhares de átomos. Ao combinar MLIPs fundamentais, particionamento de supercélula e processamento de dados otimizado on-the-fly, o método permite simulações de qualidade "near ab initio" que eram anteriormente intratáveis computacionalmente.

Os autores posicionam este trabalho como uma solução prática para interpretar experimentos de STEM-EELS monocromáticos vibracionais e de baixa perda em sistemas com nanoestruturas não triviais (defeitos, interfaces, contornos de grão). Eles declaram explicitamente que o método evita a necessidade de cálculos explícitos de modos de fônon, que são computacionalmente proibitivos para grandes sistemas. O artigo conclui modestamente, observando que o software está sob desenvolvimento ativo, com trabalhos futuros planejados para incorporar dinâmica molecular de polímero de anel termostatizada (TRPMD) para efeitos quânticos em baixas temperaturas e efeitos relacionados ao magnetismo para estudos de acoplamento spin-rede.