Interpolative Separable Density-Fitting for Transcorrelated Hamiltonians

Technische Samenvatting: Interpolatieve Separabele Density-Fitting voor Transgecorreleerde Hamiltoniaanse Operatoren

Probleemstelling
De nauwkeurige behandeling van elektronencorrelatie in ab initio elektronische structuurtheorie wordt gehinderd door de trage convergentie van gecorreleerde energieën met betrekking tot de omvang van de één-deeltjes basisset. Standaard orbitaalproductexpansies convergeren als $O(X^{-3})$ (waarbij $X$ het cardinaliteit van de basisset is) omdat antisymmetrische gladde orbitalen de elektron-elektron cusp niet kunnen reproduceren. Hoewel expliciet gecorreleerde methoden (R12/F12) deze convergentie versnellen, vereisen zij gespecialiseerde integralen en zijn ze vaak afgestemd op specifieke implementaties.

De Transgecorreleerde (TC) methode biedt een alternatief door een Jastrow-correlator in de Hamiltonian te vouwen via een gelijkenistransformatie ($\bar{H} = F^{-1}\hat{H}F$). Dit bouwt de cusp direct in de effectieve interactie, wat de basisset-convergentie versnelt. De TC-Hamiltoniaan introduceert echter een genuïne drie-lichaam operator en een effectieve twee-lichaam term. Hoewel het effectieve twee-lichaam (xTC) schema de drie-lichaam operator contracteert naar een twee-lichaam correctie met behulp van een referentie-densiteitsmatrix, blijft de evaluatie van de resulterende integralen een knelpunt.

• Analytische schema's zijn beperkt tot specifieke correlatoren en worden onhandelbaar voor flexibele, multi-centrische vormen (bijv. Boys–Handy) die nodig zijn voor gelijktijdige elektron-elektron, elektron-kern, en drie-lichaam correlaties.
• Numerieke roosterintegratie is algemeen maar evenredig duur, waarbij de kosten schalen als $O(N_g N_{orb}^4)$ voor de assemblage van effectieve twee-lichaam correcties, waarbij $N_g$ het aantal roosterpunten is. Deze kosten verhinderen de toepassing van TC-methoden op grote systemen of grote basissets (bijv. cc-pV5Z).

Methodologie
De auteurs stellen een raamwerk voor dat de Interpolatieve Separabele Density-Fitting (ISDF) benadering combineert met de effectieve transgecorreleerde (xTC) behandeling en automatische differentiatie.

1. ISDF-factorisatie: De methode comprimeert producten van orbitalen en orbitaalgradiënten op een real-space rooster in een laag-rang representatie verankerd bij een kleine set interpolatiepunten (pivots).

   • Pivot-selectie: In plaats van QR of k-means gebruiken de auteurs een iteratieve gepivoteerde Cholesky-decompositie op de orbitaal overlap densiteit ($\rho_{pq} = \phi_p \phi_q$) en de orbitaal-gradiënt densiteit. Deze selectie is onafhankelijk van de specifieke Jastrow-vorm.
   • Fusie: Scalaire en gradiënt pivot-sets worden gefuseerd om een enkele hulp-dimensie te waarborgen voor alle daaropvolgende contracties.
   • Laag-rang Representatie: De transgecorreleerde twee-lichaam integralen ($K$) en de effectieve twee-lichaam correctie ($\Delta U$) blijken deze laag-rang structuur te erven, wat de opslag- en integratiekosten met ordes van grootte vermindert.

2. Jastrow Optimalisatie: De methode maakt gebruik van een flexibele multi-centrische Boys–Handy (BH) correlator. Parameters worden geoptimaliseerd via Variational Monte Carlo (VMC) met een referentie-variantie objectief (het minimaliseren van de variantie van de lokale energie van de Slater–Jastrow golffunctie) in plaats van energie-minimalisatie.

   • Automatische Differentiatie: Gradiënten van de Jastrow-factor die vereist zijn voor de integraalkernen worden verkregen via forward-mode automatische differentiatie, wat de noodzaak voor handmatig gecodeerde, correlator-specifieke integraalroutines elimineert.

3. Implementatie en Schaling:

   • Multi-GPU Pipeline: Een producer/consumer pipeline met out-of-core opslag handelt de massieve intermediaire tensoren af (bijv. de $X_{qs\mu}$ intermediair voor $\Delta U$) die het GPU-geheugen overschrijden.
   • Laag-rang Projector: Een cruciale algoritmische innovatie past een laag-rang projector toe op de densiteitsmatrix-contractie binnen de exchange kernel, waardoor de schaling van specifieke termen wordt gereduceerd van $O(N_g N_\mu^2)$ naar $O(N_g N_\mu N_{orb})$.
   • Solver Integratie: De resulterende niet-Hermitische integralen worden doorgegeven aan een beperkte CCSD solver die is aangepast voor transgecorreleerde Hamiltoniaanse operatoren.

Belangrijkste Bijdragen

• ISDF-xTC Factorisatie: Een ISDF-factorisatie van de transgecorreleerde $K$ en $\Delta U$ integralen voor algemene rooster-geëvalueerde correlatoren, gebruikmakend van gepivoteerde-Cholesky selectie die onafhankelijk is van de Jastrow-vorm.
• Differentieerbare Workflow: Een GPU-compatibele workflow die referentie-variantie VMC optimalisatie combineert met automatische differentiatie, waardoor de noodzaak voor analytische integraalafleidingen vervalt.
• Schaalbare Architectuur: Een multi-GPU implementatie met out-of-core opslag en een laag-rang projector die de constructie van transgecorreleerde Hamiltoniaanse operatoren mogelijk maakt voor systemen met tot 1200 orbitalen (cc-pCV5Z basisset).
• Validatie: Demonstratie van de methode op de lineaire waterstofketen en benzeen, waarbij een systematische convergentie van de ISDF-benadering en een hoge nauwkeurigheid in de thermodynamische en complete-basisset (CBS) limieten wordt getoond.

Resultaten

• ISDF Convergentie: Voor ethyleen (cc-pVTZ) valt de ISDF-truncatiefout in de totale energie onder 1 mHa bij een rangratio van $N_{ch}/N_{orb} \approx 6$ en is deze geconvergeerd naar $<0.01$ mHa bij $N_{ch} = 10 N_{orb}$. De productiesetting ($N_{ch} = 12 N_{orb}$) is goed geconvergeerd.
• Waterstofketen (Thermodynamische Limiet):
  • De ISDF-xTC-CCSD methode op cc-pVTZ niveau overtreft de canonieke CCSD op cc-pV5Z niveau qua nauwkeurigheid.
  • In de thermodynamische limiet komen de resultaten overeen met state-of-the-art AFQMC+$\Delta$DMRG referenties tot binnen 1 mHa/atoom over een breed scala aan bindingslengtes ($r = 1.4–2.8$ bohr).
  • De methode bereikt deze nauwkeurigheid terwijl de $O(N^6)$ schaling van CCSD behouden blijft, waarbij de $O(N^7)$ kosten van perturbatieve triples of stochastische bemonstering worden vermeden.
• Benzeen (Grondtoestand):
  • Berekeningen werden uitgevoerd tot de cc-pCV5Z basis ($N_{orb} = 1200$).
  • De methode herstelt een CBS correlatie-energie van $\approx -1.418$ Eh, gelegen tussen canonieke CCSD ($-1.379$ Eh) en CCSD(T) ($-1.440$ Eh).
  • Cruciaal is dat de basisset-extrapolatie voor de transgecorreleerde methode merkbaar robuuster is dan voor conventionele CCSD. Het verschil tussen exponentiële en inverse-power extrapolaties is slechts 1.1 mHa voor xTC-CCSD, vergeleken met 7.6 mHa voor CCSD en 8.0 mHa voor CCSD(T).
• Prestaties: De setup tijd voor benzeen schaalt empirisch als $T \propto N_{orb}^{1.76}$, waarbij de cc-pCV5Z berekening (1200 orbitalen) ongeveer 2.3 uur duurt op een enkele B200 GPU (exclusief SCF/VMC en CCSD iteraties).

Betekenis en Claims
Het artikel claimt dat dit werk de transgecorreleerde methode praktisch maakt voor grote systemen en flexibele correlatoren. Door rooster-geëvalueerde integralen te comprimeren via ISDF en automatische differentiatie te gebruiken, hebben de auteurs de opslag- en integratie-bottlenecks overwonnen die eerder TC-methoden beperkten tot kleine systemen.

De betekenis ligt in het bereiken van state-of-the-art nauwkeurigheid (binnen ~1 mHa/atoom van hoog-niveau referenties) tegen de formele schaling van CCSD ($O(N^6)$) zonder dat perturbatieve triples of expliciet gecorreleerde neurale netwerken nodig zijn. Bovendien biedt de methode een robuustere weg naar de CBS-limiet, waarbij de gevoeligheid van extrapolatie-resultaten voor de keuze van de fit-vorm wordt verminderen. Deze combinatie van laag-rang compressie, differentieerbare correlatoren en GPU-acceleratie vestigt transcorrelatie als een levensvatbare route naar hoge nauwkeurigheid in compacte basissets voor een breed scala aan downstream solvers.