Scalar-pathway fidelity improves physical accuracy in short-range equivariant interatomic potentials

Resumo Técnico: Fidelidade da Via Escalar em Potenciais Interatômicos Equivariantes de Curto Alcance

Problema
Potenciais interatômicos equivariantes de curto alcance (MLIPs), como MACE, Allegro e NequIP, avançaram o campo ao representar a geometria atômica local através de características tensoriais $O(3)$-equivariantes. Essas arquiteturas capturam com sucesso correlações geométricas de alta ordem de corpo. No entanto, um gargalo crítico permanece: as previsões finais relevantes para implantação (energia, forças e tensão) são derivadas pela projeção dessas ricas representações tensoriais em canais escalares invariantes ($\ell=0$).

Os autores identificam que esta "via escalar" é frequentemente uma fonte de imprecisão física devido a duas limitações sistêmicas:

1. Agregação Uniforme: As mensagens escalares são tipicamente agregadas via somas invariantes à permutação e sem pesos. Isso carece de um controle de amplitude adaptativo condicionado à coordenação local, densidade ou distorção geométrica, levando a uma sub-resolução em ambientes heterogêneos (ex: superfícies, defeitos, regiões tensionadas).
2. Viés Espectral: As não-linearidades escalares (ex: SiLU) exibem um forte viés espectral de baixa frequência. Isso suprime a estrutura de alta frequência na superfície de energia potencial (PES), resultando em erros físicos como repulsão de curto alcance suavizada, modos de fônons ópticos deslocados para o vermelho (red-shifted) e deriva de energia durante a integração microcanônica de passo de tempo finito.

Abordagens atuais frequentemente tratam o readout escalar como um detalhe de implementação fixo, em vez de uma dimensão de design controlável. O artigo postula que os erros físicos residuais em MLIPs de curto alcance podem surgir significamente deste estágio de compressão escalar, e não apenas do backbone de tensores equivariantes.

Metodologia
Para abordar essas limitações, os autores introduzem dois módulos leves, preservadores de simetria, que atuam exclusivamente nos canais escalares $\ell=0$, deixando o backbone de produto tensorial equivariante inalterado. Esses módulos servem como sondas controladas da interface escalar:

1. Agrupamento de Vizinhança Consciente da Física (PAN - Physics-Aware Neighborhood Pooling):

   • Mecanismo: Substitui a agregação escalar uniforme por modulação de amplitude usando portões (gates) sigmoides.
   • Entrada: Os portões são derivados de descritores de pares invariantes, incluindo números atômicos, um descritor de coordenação suave ($\rho_i$) e uma variância de ângulo de cosseno ponderada pelo corte ($\eta_i$).
   • Função: Isso permite que a agregação escalar seja sensível à distorção geométrica local, perda de coordenação e variações de densidade sem quebrar a invariância à permutação.

2. Misturadores Espectrais Guiados pela Física (PGS - Physics-Guided Spectral Mixers):

   • Mecanismo: Aumenta as características escalares com bases espectrais motivadas pela física para melhorar a resolução espectral de curto alcance.
   • Implementação:
     • Nível de Aresta (Edge Level): Injeta uma expansão de Fourier–Bessel em distâncias de par.
     • Nível de Readout: Injeta uma expansão de Exponencial-do-Semicírculo (ES) suavizada em latentes invariantes por átomo.
   • Função: Estas bases são projetadas para melhor representar a curvatura acentuada de curto alcance e modos de alta frequência que as não-linearidades padrão suavizam.

Design Experimental
Os autores utilizam primordialmente o MACE como um arcabouço mecanístico de alta ordem de corpo para testar essas intervenções. O protocolo experimental garante que:

• O backbone equivariante, profundidade de interação, corte e orçamentos de treinamento permaneçam idênticos entre os modelos de base e os modificados.
• Comparações são feitas entre cinco sementes de treinamento independentes.
• Os módulos são testados em sistemas diversos: metal Ag, Si covalente, um subconjunto iônico de curto alcance LiF/Li–F, e os benchmarks moleculares MD17/rMD17.
• Testes de Portabilidade: Os módulos PAN+PGS são inseridos no Allegro e NequIP em locais arquitetonicamente análogos para verificar se os benefícios se transferem entre diferentes backbones equivariantes de curto alcance.

Principais Resultados

1. Melhorias de Precisão:

   • Em Ag, Si e LiF/Li–F, o modelo combinado MACE+PAN+PGS reduziu o Erro Médio Absoluto (MAE) de forças em 22–27% e o MAE de energia em 19–22% comparado ao MACE base.
   • Em sistemas com rótulos de tensão, os erros de tensão diminuíram em 27–28%.
   • Nos benchmarks moleculares (rMD17), os erros de força foram reduzidos em uma média de 13,9%, com os maiores ganhos observados em moléculas flexíveis e anharmônicas.
   • Controles de capacidade correspondente (alargamento/aprofundamento do backbone sem PAN/PGS) renderam apenas 4–7% de melhoria, demonstrando que os ganhos não se devem meramente ao aumento da contagem de parâmetros.

2. Portabilidade:

   • A inserção de PAN+PGS no Allegro e NequIP reduziu os erros de força em 17–19%, confirmando que a correção é portátil entre distintos backbones equivariantes de curto alcance, embora a magnitude da melhoria dependa da arquitetura (maior no MACE).

3. Observáveis Físicos:
   A correção da via escalar atenuou quatro modos distintos de falha física previstos pelos autores:

   • Torções Anarmônicas: Reduziu a divergência de Kullback–Leibler nas distribuições torcionais da aspirina em ~3x, corrigindo o "vazamento de barreira" (barrier leakage).
   • Modos Vibracionais: Reduziu o deslocamento para o vermelho dos fônons ópticos de alta frequência no Si, melhorando o erro de posição do pico da densidade de estados vibracionais (VDOS) por um fator de ~4.
   • Deriva de Integração: Reduziu a deriva de energia secular em simulações NVE de 500 ps de Ag bulk por um fator de ~7.
   • Estrutura Líquida: Atenuou picos de sub-camada espúrios na função de distribuição radial (RDF) do Ag líquido, corrigindo a aproximação atômica não física.

4. Custo Computacional:
   As melhorias foram alcançadas com aproximadamente 5% de FLOPs adicionais de inferência, posicionando o método em uma região de baixo erro e baixo custo da fronteira custo–precisão.

Significância e Alegações
O artigo afirma identificar a fidelidade da via escalar como uma dimensão de design ortogonal e prática para potenciais interatômicos equivariantes de curto alcance. Os autores argumentam que:

• A expressividade tensorial e a fidelidade escalar não são substituíveis; um backbone equivariante expressivo ainda pode perder detalhes físicos se as operações escalares subsequentes forem insuficientemente adaptativas ou limitadas espectralmente.
• A via escalar não é meramente um detalhe de implementação, mas uma interface crítica onde a informação geométrica é convertida em um cenário de energia.
• PAN e PGS abordam formas distintas e específicas de perda de informação (sub-resolução geométrica e viés espectral) sem exigir termos físicos explícitos (como eletrostática de longo alcance) ou substituir o backbone equivariante.
• O método é complementar às extensões existentes (ex: para interações de longo alcance ou reatividade), em vez de uma alternativa, visando especificamente o erro residual de curto alcance que permanece após a construção da representação local.

Os autores concluem que futuros MLIPs de curto alcance devem tratar a agregação escalar e a mistura espectral como componentes de design de primeira classe para garantir que a precisão física da representação equivariante seja preservada quando ela se torna uma superfície de energia implantável.