Long-Range Electrostatics and Born Effective Charges from Invariant Moment Tensor Potentials with Latent Ewald Summation (MTP-LES)

Resumo Técnico: Eletrostática de Longo Alcance e Cargas Efetivas de Born a partir de Potenciais de Tensor de Momento Invariantes com Soma de Ewald Latente (MTP–LES)

Definição do Problema

Potenciais interatômicos de aprendizado de máquina (MLIPs) alcançaram alta precisão com baixo custo computacional, mas são fundamentalmente limitados por sua dependência de descritores locais de curto alcance (tipicamente cortes de 4–6 Å). Essa suposição de localidade falha em capturar interações eletrostáticas de longo alcance e respostas de polarização essenciais para modelar líquidos polares, óxidos ferroelétricos e sistemas moleculares finitos. Embora frameworks recentes como a Soma de Ewald Latente (LES) tenham demonstrado que cargas latentes por átomo podem ser inferidas apenas a partir do treinamento de energia e força — recuperando propriedades como cargas efetivas de Born (BECs) e momentos de dipolo sem supervisão explícita de carga — as implementações existentes têm sido restritas a arquiteturas baseadas em grafos ou equivariantes (ex: CACE-LR, MACE-LES, NequIP-LES). Essas arquiteturas codificam explicitamente informações direcionais, levantando a questão de se tais características são estritamente necessárias para recuperar respostas eletrostáticas latentes fisicamente significativas, ou se a chave reside em como os graus de liberdade de carga são restringidos e estabilizados.

Metodologia

Os autores introduzem o MTP–LES, um framework que acopla o formalismo da Soma de Ewald Latente (LES) aos Potenciais de Tensor de Momento (MTP), que utilizam descritores rotacionalmente invariantes. Diferente das implementações de LES anteriores que dependem de passagem de mensagens equivariante, o MTP–LES utiliza contrações escalares de tensores de momento. Para permitir o aprendizado estável de cargas latentes dentro deste espaço de descritores invariantes, os autores implementam três modificações arquiteturais específicas:

1. Ramos Independentes: O modelo utiliza ramos de redes neurais separados para a energia de curto alcance ($E_{SR}$) e para a previsão de carga latente ($q_i$). Isso evita que os graus de liberdade de carga sejam "escravizados" à representação de energia local.
2. Viés Dependente de Espécie e Projeção de Neutralidade: Um viés de carga ($b_{s_i}$) dependente da espécie e treinável inicializa o canal de carga com uma ordenação quimicamente sensível. Crucialmente, o modelo impõe a neutralidade de carga explícita para cada configuração ao projetar a média de carga para zero ($\sum_i q_i = 0$). Isso remove cargas residuais não físicas da célula que surgem em modelos não restritos.
3. Perda de Dipolo Opcional: Para sistemas moleculares finitos (não periódicos), onde a contribuição de Ewald de longo alcance está ausente e o treinamento de energia/força sozinho pode não restringir suficientemente a escala absoluta da carga, um termo de perda auxiliar opcional é adicionado para igualar o momento de dipolo molecular total contra valores de referência (calculados via cargas MBIS).

A energia total é definida como $E_{total} = E_{SR} + E_{LR}$, onde $E_{LR}$ é calculado via soma de Ewald para sistemas periódicos ou soma de Coulomb pareada amortecida para sistemas não periódicos. Forças e BECs são obtidos via diferenciação automática da energia total e da polarização, respectivamente.

Resultados Principais

O framework MTP–LES foi avaliado em quatro sistemas de materiais diversos:

• PbTiO$_3$ Ferroelétrico: Em um benchmark rigoroso usando dados SCAN-DFT, o MTP–LES alcançou uma correlação de BEC diagonal de $R = 0,994$, igualando-se ao baseline CACE-LR estabelecido ($R = 0,993$) em dados idênticos. O modelo recuperou com sucesso as respostas de carga anômalas de Ti e O (ex: $Z^*_{Ti} \approx +6$) sem qualquer supervisão de carga ou BEC. Embora o RMSE absoluto para as magnitudes de BEC tenha sido maior que o do CACE-LR, as correlações de tensor e a ordenação anômala de espécies foram preservadas.
• Água Líquida: O modelo reproduziu correlações estruturais (função de distribuição radial oxigênio-oxigênio) e o espectro de absorção infravermelha (modos libracionais, de flexão e de estiramento) consistentes com o experimento. A adição do módulo LES melhorou a precisão das forças em aproximadamente 21% comparado a um baseline apenas de MTP. Os BECs recuperados caíram dentro da faixa de outros benchmarks aumentados por LES publicados (MACE-LES, NequIP-LES, etc.).
• Dipeptídeos Polares: Para sistemas finitos no vácuo, a inclusão de uma perda de correspondência de dipolo estabilizou a escala de carga latente. O modelo alcançou alta precisão na predição de dipolos moleculares ($R^2 = 0,996$) e BECs ($R^2 = 0,93$), superando a ablação sem supervisão de dipolo.
• Hafnia (HfO$_2$): Aplicado a um conjunto de dados recém-curado de HfO$_2$ monoclínico, o modelo preservou a fidelidade estrutural (reproduzindo a coordenação de sete vezes Hf–O) e recuperou cargas de Born anômalas de Hf ($Z^* \approx +5$) sem baselines de LES prévios para este sistema.
• Eficiência Computacional: Testes de tempo em sistemas de água líquida (até 15.360 átomos) mostraram que a correção LES introduz apenas um overhead de tempo de execução de 3,4% em relação ao baseline apenas de MTP, preservando o escalonamento quase linear com o tamanho do sistema.

Significância e Alegações

O artigo afirma que a recuperação precisa de carga latente em MLIPs não é determinada unicamente pelo uso de arquiteturas baseadas em grafos ou equivariantes. Em vez disso, os fatores críticos são como os graus de liberdade de carga são expostos, restringidos e estabilizados. Ao desacoplar os ramos de carga e energia, impor a neutralidade de carga e inicializar com um prior quimicamente sensível, o MTP–LES demonstra que potenciais baseados em descritores, rotacionalmente invariantes, podem modelar efetivamente eletrostática de longo alcance e respostas de polarização.

Este trabalho estende a aplicabilidade do framework LES para uma arquitetura baseada em descritores computacionalmente eficiente, oferecendo uma rota para incorporar eletrostática de longo alcance em MLIPs sem o custo computacional da propagação de características equivariantes ou a necessidade de rótulos de carga explícitos durante o treinamento. Os autores posicionam o MTP–LES como uma alternativa viável para simular materiais polares, iônicos e ferroelétricos onde as interações de longo alcance são fisicamente significativas.