Interpolative Separable Density-Fitting for Transcorrelated Hamiltonians

Resumo Técnico: Densidade de Ajuste por Interpolação Separável para Hamiltonianos Transcorrelacionados

Definição do Problema
O tratamento preciso da correlação eletrônica na teoria de estrutura eletrônica ab initio é dificultado pela convergência lenta das energias correlacionadas em relação ao tamanho da base de funções de um elétron. As expansões padrão de produtos orbitais convergem como $O(X^{-3})$ (onde $X$ é o número cardinal da base) porque orbitais suaves antissimétricos não conseguem reproduzir a cunha (cusp) elétron-elétron. Embora métodos explicitamente correlacionados (R12/F12) acelerem essa convergência, eles exigem integrais especializadas e são frequentemente adaptados a implementações específicas.

O método Transcorrelacionado (TC) oferece uma alternativa ao dobrar um correlator Jastrow no Hamiltoniano via transformação de similaridade ($\bar{H} = F^{-1}\hat{H}F$). Isso incorpora a cunha diretamente na interação efetiva, acelerando a convergência da base. No entanto, o Hamiltoniano TC introduz um operador de três corpos genuíno e um termo de dois corpos efetivo. Embora o esquema de dois corpos efetivos (xTC) contraia o operador de três corpos em uma correção de dois corpos usando uma matriz de densidade de referência, a avaliação das integrais resultantes permanece um gargalo.

• Esquemas analíticos são limitados a correlatores específicos e tornam-se intratáveis para formas multicêntricas flexíveis (ex: Boys–Handy), necessárias para correlações simultâneas elétron-elétron, elétron-núcleo e de três corpos.
• A integração por grade numérica é geral, mas proibitivamente cara, escalando como $O(N_g N_{orb}^4)$ para a montagem das correções de dois corpos efetivas, onde $N_g$ é o número de pontos de grade. Esse custo impede a aplicação de métodos TC a sistemas grandes ou bases de funções grandes (ex: cc-pV5Z).

Metodologia
Os autores propõem um framework combinando a aproximação de Ajuste de Densidade por Interpolação Separável (ISDF) com o tratamento transcorrelacionado efetivo (xTC) e diferenciação automática.

1. Fatoração ISDF: O método comprime produtos de orbitais e gradientes orbitais em uma grade no espaço real em uma representação de baixo posto (low-rank) ancorada em um pequeno conjunto de pontos de interpolação (pivôs).

   • Seleção de Pivôs: Em vez de QR ou k-means, os autores utilizam uma decomposição de Cholesky com pivoteamento iterativo na densidade de sobreposição orbital ($\rho_{pq} = \phi_p \phi_q$) e na densidade de gradiente orbital. Esta seleção é independente da forma específica do Jastrow.
   • Fusão: Conjuntos de pivôs escalares e de gradiente são fundidos para garantir um único dimensão auxiliar para todas as contrações subsequentes.
   • Representação de Baixo Posto: Mostra-se que as integrais de dois corpos transcorrelacionadas ($K$) e a correção de dois corpos efetiva ($\Delta U$) herdam essa estrutura de baixo posto, reduzindo custos de armazenamento e integração em ordens de magnitude.

2. Otimização de Jastrow: O método emprega um correlator Boys–Handy (BH) multicêntrico flexível. Os parâmetros são otimizados via Monte Carlo Variacional (VMC) usando um objetivo de variância de referência (minimizando a variância da energia local da função de onda Slater–Jastrow) em vez da minimização da energia.

   • Diferenciação Automática: Os gradientes do fator Jastrow necessários para os núcleos integrais são obtidos via diferenciação automática de modo direto (forward-mode), eliminando a necessidade de rotinas de integração codificadas à mão e específicas para cada correlator.

3. Implementação e Escalonamento:

   • Pipeline Multi-GPU: Um pipeline produtor/consumidor com armazenamento out-of-core lida com os massivos tensores intermediários (ex: o $X_{qs\mu}$ intermediário para $\Delta U$) que excedem a memória da GPU.
   • Projetor de Baixo Posto: Uma inovação algorítmica chave aplica um projetor de baixo posto à contração da matriz de densidade no núcleo de troca, reduzindo o escalonamento de termos específicos de $O(N_g N_\mu^2)$ para $O(N_g N_\mu N_{orb})$.
   • Integração com Solver: As integrais não-hermitianas resultantes são passadas para um solver CCSD restrito adaptado para Hamiltonianos transcorrelacionados.

Principais Contribuições

• Fatoração ISDF-xTC: Uma fatoração ISDF das integrais de dois corpos transcorrelacionadas $K$ e $\Delta U$ para correlatores avaliados em grade, utilizando seleção por Cholesky com pivoteamento independente da forma do Jastrow.
• Fluxo de Trabalho Diferenciável: Um fluxo de trabalho compatível com GPU combinando otimização VMC de variância de referência com diferenciação automática, removendo a necessidade de derivações analíticas de integrais.
• Arquitetura Escalável: Uma implementação multi-GPU com armazenamento out-of-core e um projetor de baixo posto que permite a construção de Hamiltonianos transcorrelacionados para sistemas de até 1200 orbitais (base cc-pCV5Z).
• Validação: Demonstração do método na cadeia linear de hidrogênio e no benzeno, mostrando convergência sistemática da aproximação ISDF e alta precisão nos limites termodinâmico e de base completa (CBS).

Resultos

• Convergência ISDF: Para o etileno (cc-pVTZ), o erro de truncamento ISDF na energia total cai abaixo de 1 mHa com uma razão de posto $N_{ch}/N_{orb} \approx 6$, e converge para $<0.01$ mHa em $N_{ch} = 10 N_{orb}$. A configuração de produção ($N_{ch} = 12 N_{orb}$) está bem convergida.
• Cadeia de Hidrogênio (Limite Termodinâmico):
  • O método ISDF-xTC-CCSD ao nível cc-pVTZ supera o CCSD canônico ao nível cc-pV5Z em precisão.
  • No limite termodinâmico, os resultados concordam com referências de estado da arte AFQMC+$\Delta$DMRG dentro de 1 mHa/átomo em uma ampla gama de comprimentos de ligação ($r = 1.4–2.8$ bohr).
  • O método alcança essa precisão mantendo o escalonamento $O(N^6)$ do CCSD, evitando o custo $O(N^7)$ de triplos perturbativos ou amostragem estocástica.
• Benzeno (Estado Fundamental):
  • Cálculos foram realizados até a base cc-pCV5Z ($N_{orb} = 1200$).
  • O método recupera uma energia de correlação CBS de $\approx -1.418$ Eh, situando-se entre o CCSD canônico ($-1.379$ Eh) e o CCSD(T) ($-1.440$ Eh).
  • Crucialmente, a extrapolação de base (CBS) para o método transcorrelacionado é marcadamente mais robusta do que para o CCSD convencional. A diferença entre as extrapolações exponencial e de potência inversa é de apenas 1.1 mHa para o xTC-CCSD, comparado a 7.6 mHa para CCSD e 8.0 mHa para CCSD(T).
• Desempenho: O tempo de configuração para o benzeno escala empiricamente como $T \propto N_{orb}^{1.76}$, com o cálculo cc-pCV5Z (1200 orbitais) completando aproximadamente 2.3 horas em uma única GPU B200 (excluindo iterações SCF/VMC e CCSD).

Significância e Alegações
O artigo afirma que este trabalho torna o método transcorrelacionado prático para grandes sistemas e correlatores flexíveis. Ao comprimir integrais avaliadas em grade via ISDF e utilizar diferenciação automática, os autores superam os gargalos de armazenamento e integração que anteriormente restringiam os métodos TC a sistemas pequenos.

A significância reside em alcançar precisão de estado da arte (dentro de ~1 mHa/átomo de referências de alto nível) com o escalonamento formal de CCSD ($O(N^6)$) sem exigir triplos perturbativos ou redes neurais explicitamente correlacionadas. Além disso, o método fornece um caminho mais robusto para o limite CBS, reduzindo a sensibilidade dos resultados de extrapolação à escolha da forma de ajuste. Essa combinação de compressão de baixo posto, correlatores diferenciáveis e aceleração por GPU estabelece a transcorrelação como uma rota viável para alta precisão em bases compactas para uma ampla gama de solvers subsequentes.