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

Technische Samenvatting: Scalaire Padgetrouwheid in Kort-Bereik Equivariante Interatomaire Potentialen

Probleemstelling
Kort-bereik equivariante interatomaire potentialen (MLIP's), zoals MACE, Allegro en NequIP, hebben het vakgebied vooruitgeholpen door lokale atomaire geometrie te representeren via $O(3)$-equivariante tensor-features. Deze architecturen vangen succesvol hoog-body-orde geometrische correlaties op. Een kritieke bottleneck blijft echter: de uiteindelijke, voor deployment relevante voorspellingen (energie, krachten en spanning) worden afgeleid door deze rijke tensor-representaties te projecteren op invariante scalaire ($\ell=0$) kanalen.

De auteurs identificeren dat deze "scalaire pad" vaak een bron van fysieke onnauwkeurigheid is vanwege twee systemische beperkingen:

1. Uniforme Aggregatie: Scalaire berichten worden doorgaans geaggregeerd via ongewogen, permutatie-invariante sommen. Dit mist expliciete adaptieve amplitude-gating die geconditioneerd is op lokale coördinatie, dichtheid of geometrische distortie, wat leidt tot een gebrek aan resolutie in heterogene omgevingen (bijv. oppervlakken, defecten, gespannen regio's).
2. Spectrale Bias: Scalaire nonlineariteiten (bijv. SiLU) vertonen een sterke laag-frequentie spectrale bias. Dit onderdrukt hoog-frequente structuren in de potentiaalenergie-oppervlakte (PES), wat resulteert in fysieke fouten zoals verzachte kort-bereik afstoting, roodverschoven optische fonon-modi en energie-drift tijdens microcanonische integratie met eindige tijdstappen.

Huidige benaderingen behandelen de scalaire readout vaak als een vast implementatiedetail in plaats van een controleerbare ontwerpdimensie. Het artikel stelt dat residuele fysieke fouten in kort-bereik MLIP's aanzienlijk kunnen voortkomen uit deze scalaire compressiefase, in plaats van enkel uit de equivariante tensor-backbone.

Methodologie
Om deze beperkingen aan te pakken, introduceren de auteurs twee lichtgewicht, symmetrie-behoudende modules die uitsluitend werken op de $\ell=0$ scalaire kanalen, waarbij de equivariante tensorproduct-backbone ongewijzigd blijft. Deze modules dienen als gecontroleerde probes van de scalaire interface:

1. Physics-Aware Neighborhood (PAN) Pooling:

   • Mechanisme: Vervangt uniforme scalaire aggregatie door amplitude-modulatie met sigmoid gates.
   • Input: De gates worden afgeleid van invariante paar-descriptors, inclusief atoomnummers, een gladde coördinatie-descriptor ($\rho_i$), en een cutoff-gewogen cosinus-hoek variantie ($\eta_i$).
   • Functie: Dit maakt de scalaire aggregatie gevoelig voor lokale geometrische distortie, coördinatieverlies en dichtheidsvariaties zonder de permutatie-invariantie te verbreken.

2. Physics-Guided Spectral (PGS) Mixers:

   • Mechanisme: Augmenteert scalaire features met fysisch gemotiveerde spectrale bases om de kort-bereik spectrale resolutie te verbeteren.
   • Implementatie:
     • Edge Level: Injecteert een Fourier–Bessel expansie op paar-afstanden.
     • Readout Level: Injecteert een getaperde Exponential-of-Semicircle (ES) expansie op per-atoom invariante latents.
   • Functie: Deze bases zijn ontworpen om steile kort-bereik kromming en hoog-frequente modi beter te representeren die standaard nonlineariteiten afvlakken.

Experimenteel Ontwerp
De auteurs gebruiken primair MACE als een hoog-body-orde mechanistisch scaffold om deze interventies te testen. Het experimentele protocol waarborgt dat:

• De equivariante backbone, interactiediepte, cutoff en trainingsbudgetten identiek blijven tussen de baseline en de aangepaste modellen.
• Vergelijkingen worden gemaakt over vijf onafhankelijke trainings-seeds.
• De modules worden getest op diverse systemen: metallisch Ag, covalent Si, een kort-bereik ionisch LiF/Li–F subset, en de MD17/rMD17 moleculaire benchmarks.
• Portabiliteitstests: De PAN+PGS modules worden ingevoegd in Allegro en NequIP op architecturaal analoge locaties om te verifiëren of de voordelen overdraagbaar zijn naar verschillende kort-bereik equivariante backbones.

Belangrijkste Resultaten

1. Nauwkeurigheidsverbeteringen:

   • Over Ag, Si en LiF/Li–F verminderde het gecombineerde MACE+PAN+PGS model de kracht Mean Absolute Error (MAE) met 22–27% en de energie MAE met 19–22% vergeleken met de baseline MACE.
   • Op systemen met stress-labels daalden de stress-fouten met 27–28%.
   • In moleculaire benchmarks (rMD17) werden de krachtfouten gemiddeld met 13,9% verminderd, waarbij de grootste winsten werden waargenomen in flexibele, anharmonische moleculen.
   • Capaciteits-gematchte controles (het verbreden/verdiepen van de backbone zonder PAN/PGS) leverden slechts 4–7% verbetering op, wat aantoont dat de winsten niet louter toe te schrijven zijn aan een groter aantal parameters.

2. Portabiliteit:

   • Het invoegen van PAN+PGS in Allegro en NequIP reduceerde de krachtfouten met 17–19%, wat bevestigt dat de correctie portabel is over verschillende kort-bereik equivariante backbones, hoewel de mate van verbetering architectuur-afhankelijk is (grootst in MACE).

3. Fysische Observabelen:
   De correctie van de scalaire pad specifiek de vier verschillende vormen van fysieke falingsmodi die de auteurs voorspelden:

   • Anharmonische Torsies: Verminderde de Kullback–Leibler divergentie in aspirine torsie-distributies met ~3x, waardoor "barrier leakage" werd gecorrigeerd.
   • Vibratie-modi: Verminderde de roodverschuiving van hoog-frequente optische fononen in Si, waardoor de fout in de piekpositie van de vibratie dichtheidsverdeling (VDOS) met een factor ~4 werd verbeterd.
   • Integratie-drift: Verminderde de seculaire energie-drift in 500 ps NVE simulaties van bulk Ag met een factor ~7.
   • Vloeistofstructuur: Attenueerde spurieuze sub-shell pieken in de radiale distributiefunctie (RDF) van vloeibaar Ag, wat onfysische atomaire nadering corrigeert.

4. Computationele Kosten:
   De verbeteringen werden bereikt met ongeveer 5% extra inference FLOPs, waardoor de methode zich in een laag-fout, laag-overhead regio van de kosten-nauwkeurigheid frontier bevindt.

Betekenis en Claims
Het artikel claimt de identificatie van scalaire padgetrouwheid als een praktische, orthogonale ontwerpdimensie voor kort-bereik equivariante interatomaire potentialen. De auteurs stellen dat:

• Tensor-expressiviteit en scalaire getrouwheid niet vervangbaar zijn; een expressieve equivariante backbone kan nog steeds fysieke details verliezen als de downstream scalaire operaties onvoldoende adaptief of spectraal beperkt zijn.
• De scalaire pad niet louter een implementatiedetail is, maar een kritieke interface waar geometrische informatie wordt omgezet in een energielandschap.
• PAN en PGS specifieke, onderscheidende vormen van informatieverlies aanpakken (geometrische onder-resolutie en spectrale bias) zonder expliciete fysieke termen (zoals lang-bereik elektrostatica) te vereisen of de equivariante backbone te vervangen.
• De methode complementair is aan bestaande uitbreidingen (bijv. voor lang-bereik interacties of reactiviteit), specifiek gericht op de residuele kort-bereik fout die overblijft nadat de lokale representatie is opgebouwd.

De auteurs concluderen dat toekomstige kort-bereik MLIP's scalaire aggregatie en spectrale mixing als eersteklas ontwerpcomponenten moeten behandelen om te garanderen dat de fysieke nauwkeurigheid van de equivariante representatie behouden blijft wanneer deze wordt omgezet in een inzetbare energierelatie.