REVIEW 2 major objections 6 minor 34 references
A small per-layer adapter lets any equivariant foundation force field respond continuously to charge without rebuilding the model.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · grok-4.5
2026-07-11 05:48 UTC pith:RKTHPUUN
load-bearing objection Clean, equivariance-preserving FiLM adapter that turns charge conditioning into a few-thousand-frame fine-tune; water results are solid, generality language is a bit ahead of the tests. the 2 major comments →
EquiFiLM: Charge-Conditioned Equivariant Force Fields via Feature-wise Linear Modulation
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
EquiFiLM shows that continuous external conditioning of an equivariant foundation force field can be realized by a per-layer FiLM block that modulates only scalar interaction-layer channels from a single per-graph scalar; the construction preserves E(3)-equivariance exactly, adds negligible inference cost, and, when applied to MACE-MatPES on charged liquid water, yields a single set of weights that matches specialist and charge-aware baselines on training charges, generalizes across held-out charges, and supports stable, energy-conserving molecular dynamics whose structural response matches the expected charge-dependent pair-distribution-function shift.
What carries the argument
ChargeFiLMBlock: two small MLPs map the conditioning scalar c to per-channel scale γ(c) and shift β(c) that act only on the scalar (ℓ=0) slice of every message tensor via m′ = (1+γ)⊙m + β, leaving all higher-rank equivariant channels unchanged; zero-initialized so the adapter begins as the identity.
Load-bearing premise
Held-out charge accuracy is measured mainly by re-labeling geometries taken from nearby trained trajectories rather than by fully re-equilibrating nuclei at the new electronic state, so the reported interpolation and extrapolation errors may not capture the true charged potential-energy surface.
What would settle it
Run independent AIMD at a held-out charge (for example q=12e or 18e), recompute forces and energies with the same DFT settings, and check whether E-MACE force RMSE remains inside the claimed 18–61 meV/Å band and whether NVE trajectories stay energy-conserving without special seeding.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript introduces EquiFiLM, a lightweight per-layer Feature-wise Linear Modulation adapter that injects a continuous per-graph scalar (here total charge, entered as c=q/N) into equivariant foundation MLFFs by scaling and shifting only scalar interaction-layer channels (Eq. 2), thereby preserving E(3)-equivariance exactly. Instantiated on MACE-MatPES for charged liquid water (E-MACE) and trained on ~6400 r2SCAN AIMD frames at four charges, the model reports ~3.1× lower force RMSE and ~61× lower per-atom energy RMSE than an unconditioned fine-tune on the same data, force accuracy competitive with fine-tuned MACE-POLAR-1-M at ~3× lower inference cost, usable held-out interpolation/extrapolation force and energy errors, stable energy-conserving MD, and a charge-dependent first-shell ΔG(r;q) prediction relevant to UED. Ablations (Table 2), a per-state specialist matrix (Appendix A), data-efficiency curves, a GPAW/PBE cross-pipeline force-response check, and a 2×2×2 supercell structural check support the design. Code, data, and a checkpoint are released.
Significance. If the results hold, EquiFiLM is a practically important contribution: it reframes external conditioning of foundation MLFFs as a parameter-efficient adapter problem rather than a from-scratch charge-aware foundation (~10^8 structures) or a bank of per-state specialists. Exact equivariance preservation by scalar-only modulation, zero-init drop-in identity, and negligible inference overhead are clean design points. Strengths that raise confidence include reproducible code and Zenodo data, systematic ablations (β-only, concat embedding, width, init), NVE energy conservation at interpolation and extrapolation charges, falsifiable ΔG(r;q) structural predictions, supercell transfer, and an independent DFT-engine force-difference check. The recipe is of clear interest for electrochemistry, photoinjection, and other driven atomistic processes where foundations currently lack a conditioning axis.
major comments (2)
- Section 3.1 and Appendix C (Table 6): held-out force/energy RMSE is measured as vertical electronic response—geometries sampled from AIMD at a nearby training charge and only re-labeled at the target NELECT—not on nuclei re-equilibrated at the new charge. Full charge-plus-geometry generalization is deferred to MD stability and structure. This protocol is stated in the text but is load-bearing for the abstract claim that the model “generalizes” across seven held-out charges and that “one set of weights is usable at any charge.” Please elevate this distinction into the abstract and the opening of §3.1 (e.g., “vertical force response at fixed near-equilibrium nuclei”), and either (i) add a limited set of re-equilibrated single-points or short AIMD-relabeled checks at one interpolation and one extrapolation charge, or (ii) explicitly bound the RMSE claim so readers do not over-read Figure 3
- §2.2–§4 and the abstract claim the recipe is backbone- and conditioning-agnostic (“any equivariant MLFF with scalar interaction-layer channels”; temperature, pressure, doping). Empirically only MACE-MatPES + total charge on liquid water is shown. The architectural argument (scalar-only FiLM preserves equivariance) is sound, but the generality claim is stronger than the evidence. Either add a second backbone (e.g., NequIP/Allegro-style or another MACE variant) and/or a second continuous scalar on a small corpus, or rephrase abstract/conclusions to “architecturally applicable; demonstrated on MACE-MatPES with charge,” matching the honest caveats already in §4 about unvalidated γ-dominated axes and multi-axis conditioning.
minor comments (6)
- Figure 1 caption and §1: “reaches the accuracy of … MACE-POLAR-1-M fine-tuned” is force-centric; energy comparisons in Figure 3 use offset correction that the text correctly calls generous to POLAR. State “on forces” in the Figure 1 caption to avoid over-reading.
- Table 1 vs Appendix B: inference is reported as µs/atom-step with overlapping 1σ bars for E-MACE and fine-tuned MatPES; good. Briefly note in the main text that the pure-functional cat rewrite (Appendix B) is required to avoid IndexPut slowdown, so “indistinguishable cost” assumes the reference implementation.
- Appendix C: the jellium-compensated, delocalized excess-electron regime (and the explicit disclaimer against localized hydrated-electron polarons) is important for experimental interpretation of ΔG(r;q). A one-sentence pointer in §3.4.2 would help non-DFT readers.
- Notation: c = q/N is introduced in §2.2; some later MD sections switch between q and q_cell without restating that c is held fixed under supercell tiling. A short reminder in Appendix H would reduce confusion.
- Related work: SpookyNet and the charge-equilibration lineage are covered; a brief pointer to other conditional/adapter uses of FiLM outside atomistics is already present—fine. Consider citing any concurrent charge-conditioned MACE/UMA adapters if known at revision time.
- Typos/clarity: “≈10 8” spacing in abstract/intro; “r 2SCAN” vs “r2SCAN” inconsistency; Figure 3 bottom panel “offset-corrected energy vs. fine-tuned MACE-POLAR-1-M” y-axis label could state units explicitly in the figure.
Circularity Check
No significant circularity: EquiFiLM is an empirical adapter whose accuracy claims rest on external DFT labels and independent baselines, not on definitions or self-citation chains that force the reported results.
full rationale
The paper’s load-bearing claims are architectural (scalar-only FiLM preserves E(3)-equivariance by the standard definition of equivariance under O(3)-invariant per-graph gates) and empirical (force/energy RMSE vs no-FiLM, concat embedding, per-state specialists, and fine-tuned MACE-POLAR-1-M; held-out charges; NVE conservation; ΔG(r;q) from MD). Training and evaluation labels are external VASP r2SCAN (with a GPAW/PBE cross-pipeline check). FiLM γ/β are learned parameters, zero-initialized to the identity, not defined to equal the target RMSE or PDF. Ablations and the specialist cross-matrix show the gain comes from per-layer modulation rather than mere access to q. Citations (FiLM, MACE, charge-aware foundations) are external prior art, not author uniqueness theorems that forbid alternatives. The vertical-relabel protocol for held-out charges is a scope limit on what is measured, not a construction that forces the reported errors. No equation or self-citation reduces a claimed prediction to its inputs by definition.
Axiom & Free-Parameter Ledger
free parameters (4)
- FiLM MLP hidden width h =
h=128 (headline); h=64 (ablations)
- Adapter MLP weights (γ, β networks) =
~0.10 M parameters
- Training optimizer and schedule (lr, SWA, loss weights) =
lr=0.01; SWA 150–250; wF=100
- Conditioning representation c=q/N =
c = q/N
axioms (5)
- standard math E(3)-equivariance is preserved if only scalar (ℓ=0) channels are scaled/shifted by an O(3)-invariant function of a per-graph scalar.
- domain assumption The foundation backbone already covers the chemistry of interest at near-DFT accuracy on the unconditioned baseline.
- domain assumption External conditioning (here total charge) acts as a smooth continuous scalar effect on the PES over the studied range.
- domain assumption Jellium-compensated delocalized excess electrons (NELECT = N_neutral + q) are an adequate model of the charge axis for the intended photoinjection/electron-transfer regime.
- ad hoc to paper Geometries from nearby-charge AIMD re-labeled at target charge adequately probe charge generalization of forces/energies.
invented entities (2)
-
EquiFiLM / ChargeFiLMBlock
independent evidence
-
E-MACE
independent evidence
read the original abstract
Foundation machine learning force fields (MLFFs) such as MACE-MP-0 and UMA cover broad chemical space at near density functional theory (DFT) accuracy. However, they assume equilibrium ground-state physics and do not natively handle externally induced changes to the electronic state, such as charging, applied fields, or electronic excitation, which limits their use for driven processes such as photoexcitation and charge injection. We propose EquiFiLM, a lightweight extension that adds continuous external conditioning to any equivariant foundation MLFF via a per-layer Feature-wise Linear Modulation (FiLM) block, learning externally driven changes to the potential energy surface from minimal training data. The block modulates only scalar channels and preserves E(3)-equivariance exactly. We demonstrate the recipe on charged liquid water with the foundation model MACE-MatPES as the backbone, yielding E-MACE. On the four training charges, E-MACE delivers a $3.1\times$ reduction in force RMSE ($21.3$ to $6.96$ meV/$\mathring{A}$) and a $61\times$ reduction in per-atom energy RMSE ($6.1$ to $0.1$ meV/atom) over a baseline without EquiFiLM trained on the same data, at indistinguishable inference cost. Across seven held-out interpolation and extrapolation charges, force RMSE stays within $18-61$ meV/$\mathring{A}$ and energy RMSE within $0.7-5.4$ meV/atom. The model runs stable molecular dynamics across the full range tested and predicts the charge-dependent first-shell response of the reduced pair distribution function probed by ultrafast electron diffraction. Adding this conditioning axis to the foundation requires only a few thousand DFT-labeled frames, against the $\approx 10^8$ structures of a charge-aware foundation trained from scratch. The recipe is backbone- and conditioning-agnostic: it applies without architectural change to any equivariant MLFF with scalar interaction-layer channels.
Figures
Reference graph
Works this paper leans on
-
[1]
A foundation model for atomistic materials chemistry.The Journal of Chemical Physics, 2025
Ilyes Batatia, Philipp Benner, Yuan Chiang, Gábor Csányi, et al. A foundation model for atomistic materials chemistry.The Journal of Chemical Physics, 2025
2025
-
[2]
Brandon M. Wood, Misko Dzamba, Xiang Fu, Meng Gao, Muhammed Shuaibi, Luis Barroso- Luque, Kareem Abdelmaqsoud, Vahe Gharakhanyan, John R. Kitchin, Daniel S. Levine, Kyle Michel, Anuroop Sriram, Taco Cohen, Abhishek Das, Ammar Rizvi, Sushree Jagriti Sahoo, Zachary W. Ulissi, and C. Lawrence Zitnick. UMA: A family of universal models for atoms. arXiv prepri...
arXiv 2025
-
[3]
Kaplan, Runze Liu, Ji Qi, Tsz Wai Ko, Bowen Deng, Janosh Riebesell, Gerbrand Ceder, Kristin A
Aaron D. Kaplan, Runze Liu, Ji Qi, Tsz Wai Ko, Bowen Deng, Janosh Riebesell, Gerbrand Ceder, Kristin A. Persson, and Shyue Ping Ong. A foundational potential energy surface dataset for materials.arXiv preprint arXiv:2503.04070, 2025
arXiv 2025
-
[4]
Yi-Lun Liao, Alexander J. Hoffman, Sabrina C. Shen, Alexandre Duval, Sam Walton Norwood, and Tess Smidt. EquiformerV3: Scaling efficient, expressive, and general SE(3)-equivariant graph attention transformers.arXiv preprint arXiv:2604.09130, 2026
Pith/arXiv arXiv 2026
-
[5]
Wood, Luis Barroso-Luque, Daniel S
Xiang Fu, Brandon M. Wood, Luis Barroso-Luque, Daniel S. Levine, Meng Gao, Misko Dzamba, and C. Lawrence Zitnick. Learning smooth and expressive interatomic potentials for physical property prediction.arXiv preprint arXiv:2502.12147, 2025
Pith/arXiv arXiv 2025
-
[6]
Jaesun Kim, Jinmu You, Yutack Park, Yunsung Lim, Yujin Kang, Jisu Kim, Haekwan Jeon, Suyeon Ju, Deokgi Hong, Seung Yul Lee, Saerom Choi, Yongdeok Kim, Jae W. Lee, and Seungwu Han. Optimizing cross-domain transfer for universal machine learning interatomic potentials.arXiv preprint arXiv:2510.11241, 2025
arXiv 2025
-
[7]
Orb-v3: atomistic simulation at scale.arXiv preprint arXiv:2504.06231, 2025
Benjamin Rhodes, Sander Vandenhaute, Vaidotas Šimkus, James Gin, Jonathan Godwin, Tim Duignan, and Mark Neumann. Orb-v3: atomistic simulation at scale.arXiv preprint arXiv:2504.06231, 2025
Pith/arXiv arXiv 2025
-
[8]
PET-MAD as a lightweight universal interatomic potential for advanced materials modeling.Nature Communications, 16:10653, 2025
Arslan Mazitov, Filippo Bigi, Matthias Kellner, Paolo Pegolo, Davide Tisi, Guillaume Fraux, Sergey Pozdnyakov, Philip Loche, and Michele Ceriotti. PET-MAD as a lightweight universal interatomic potential for advanced materials modeling.Nature Communications, 16:10653, 2025
2025
-
[9]
Baldwin, Domantas Kuryla, Joseph Hart, Elliott Kasoar, Alin M
Ilyes Batatia, William J. Baldwin, Domantas Kuryla, Joseph Hart, Elliott Kasoar, Alin M. Elena, Harry Moore, Mikołaj J. Gawkowski, Benjamin X. Shi, Venkat Kapil, Panagiotis Kourtis, Ioan- Bogdan Magd˘au, and Gábor Csányi. MACE-POLAR-1: A polarisable electrostatic foundation model for molecular chemistry.arXiv preprint arXiv:2602.19411, 2026
arXiv 2026
-
[10]
Finkler, Stefan Goedecker, and Jörg Behler
Tsz Wai Ko, Jonas A. Finkler, Stefan Goedecker, and Jörg Behler. A fourth-generation high- dimensional neural network potential with accurate electrostatics including non-local charge transfer.Nature Communications, 12:398, 2021
2021
-
[11]
Anstine, Maike Bergeler, V olker Settels, Con- rad Stork, Sebastian Spicher, and Olexandr Isayev
Bhupalee Kalita, Roman Zubatyuk, Dylan M. Anstine, Maike Bergeler, V olker Settels, Con- rad Stork, Sebastian Spicher, and Olexandr Isayev. AIMNet2-NSE: A transferable reac- tive neural network potential for open-shell chemistry.ChemRxiv, 2025. doi: 10.26434/ chemrxiv-2025-kdg6n-v2
2025
-
[12]
Panagiotopoulos, Roberto Car, and Weinan E
Linfeng Zhang, Han Wang, Maria Carolina Muniz, Athanassios Z. Panagiotopoulos, Roberto Car, and Weinan E. A deep potential model with long-range electrostatic interactions.Journal of Chemical Physics, 156:124107, 2022
2022
-
[13]
Unke, Stefan Chmiela, Michael Gastegger, Kristof T
Oliver T. Unke, Stefan Chmiela, Michael Gastegger, Kristof T. Schütt, Huziel E. Sauceda, and Klaus-Robert Müller. SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects.Nature Communications, 12:7273, 2021
2021
-
[14]
FiLM: Visual reasoning with a general conditioning layer
Ethan Perez, Florian Strub, Harm de Vries, Vincent Dumoulin, and Aaron Courville. FiLM: Visual reasoning with a general conditioning layer. InAAAI Conference on Artificial Intelligence, 2018. 11
2018
-
[15]
Ilyes Batatia, Dávid Péter Kovács, Gregor N. C. Simm, Christoph Ortner, and Gábor Csányi. MACE: Higher order equivariant message passing neural networks for fast and accurate force fields. InAdvances in Neural Information Processing Systems, 2022
2022
-
[16]
Atomic cluster expansion for accurate and transferable interatomic potentials
Ralf Drautz. Atomic cluster expansion for accurate and transferable interatomic potentials. Physical Review B, 99:014104, 2019
2019
-
[17]
Kresse and J
G. Kresse and J. Furthmüller. Efficient iterative schemes forab initiototal-energy calculations using a plane-wave basis set.Physical Review B, 54:11169–11186, 1996
1996
-
[18]
Furness, Aaron D
James W. Furness, Aaron D. Kaplan, Jinliang Ning, John P. Perdew, and Jianwei Sun. Accurate and numerically efficient r2SCAN meta-generalized gradient approximation.Journal of Physical Chemistry Letters, 11:8208–8215, 2020
2020
-
[19]
Pavel Izmailov, Dmitrii Podoprikhin, Timur Garipov, Dmitry Vetrov, and Andrew Gordon Wilson. Averaging weights leads to wider optima and better generalization.arXiv preprint arXiv:1803.05407, 2018
Pith/arXiv arXiv 2018
-
[20]
Ming-Fu Lin, Nirmal Singh, Senmiao Liang, Mianzhen Mo, J. P. F. Nunes, Kathryn Ledbetter, Jie Yang, Michael Kozina, Stephen Weathersby, Xiaozhe Shen, et al. Imaging the short-lived hydroxyl-hydronium pair in ionized liquid water.Science, 374:92–95, 2021
2021
-
[21]
E(3)-equivariant graph neural networks for data-efficient and accurate inter- atomic potentials.Nature Communications, 13:2453, 2022
Simon Batzner. E(3)-equivariant graph neural networks for data-efficient and accurate inter- atomic potentials.Nature Communications, 13:2453, 2022
2022
-
[22]
Owen, Mordechai Kornbluth, and Boris Kozinsky
Albert Musaelian, Simon Batzner, Anders Johansson, Lixin Sun, Cameron J. Owen, Mordechai Kornbluth, and Boris Kozinsky. Learning local equivariant representations for large-scale atomistic dynamics.Nature Communications, 14:579, 2023
2023
-
[23]
Courville
Harm de Vries, Florian Strub, Jérémie Mary, Hugo Larochelle, Olivier Pietquin, and Aaron C. Courville. Modulating early visual processing by language. InAdvances in Neural Information Processing Systems, 2017
2017
-
[24]
Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen
Edward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. LoRA: Low-rank adaptation of large language models. In International Conference on Learning Representations, 2022
2022
-
[25]
Parameter-efficient transfer learning for NLP
Neil Houlsby, Andrei Giurgiu, Stanisław Jastrz˛ ebski, Bruna Morrone, Quentin de Laroussilhe, Andrea Gesmundo, Mona Attariyan, and Sylvain Gelly. Parameter-efficient transfer learning for NLP. InInternational Conference on Machine Learning, 2019
2019
-
[26]
David Ha, Andrew Dai, and Quoc V . Le. HyperNetworks. InInternational Conference on Learning Representations, 2017
2017
-
[27]
Smith, Benjamin T
Roman Zubatyuk, Justin S. Smith, Benjamin T. Nebgen, Sergei Tretiak, and Olexandr Isayev. Teaching a neural network to attach and detach electrons from molecules.Nature Communica- tions, 12:4870, 2021
2021
-
[28]
Probing the self-ionization of liquid water with ab initio deep potential molecular dynamics.Proceedings of the National Academy of Sciences, 120:e2302468120, 2023
Marcos Calegari Andrade, Roberto Car, and Annabella Selloni. Probing the self-ionization of liquid water with ab initio deep potential molecular dynamics.Proceedings of the National Academy of Sciences, 120:e2302468120, 2023
2023
-
[29]
Enhanced deep potential model for fast and accurate molecular dynamics: application to the hydrated electron.Physical Chemistry Chemical Physics, 26:23080–23088, 2024
Ruiqi Gao, Yifan Li, and Roberto Car. Enhanced deep potential model for fast and accurate molecular dynamics: application to the hydrated electron.Physical Chemistry Chemical Physics, 26:23080–23088, 2024
2024
-
[30]
A foundation machine learning potential with polarizable long-range interactions for materials modelling.Nature Communications, 16:10484, 2025
Rongzhi Gao, ChiYung Yam, Jianjun Mao, Shuguang Chen, GuanHua Chen, and Ziyang Hu. A foundation machine learning potential with polarizable long-range interactions for materials modelling.Nature Communications, 16:10484, 2025
2025
-
[31]
Decoupled weight decay regularization.arXiv preprint arXiv:1711.05101, 2017
Ilya Loshchilov and Frank Hutter. Decoupled weight decay regularization.arXiv preprint arXiv:1711.05101, 2017
Pith/arXiv arXiv 2017
-
[32]
Reddi, Satyen Kale, and Sanjiv Kumar
Sashank J. Reddi, Satyen Kale, and Sanjiv Kumar. On the convergence of Adam and beyond. arXiv preprint arXiv:1904.09237, 2019. 12
Pith/arXiv arXiv 1904
-
[33]
excess electron
Jens Enkovaara, C. Rostgaard, Jens J. Mortensen, J. Chen, M. Dułak, L. Ferrighi, J. Gavnholt, C. Glinsvad, V . Haikola, H. A. Hansen, et al. Electronic structure calculations with GPAW: a real-space implementation of the projector augmented-wave method.Journal of Physics: Condensed Matter, 22:253202, 2010. 13 Appendix A Per-state cross-evaluation matrix T...
2010
-
[34]
Odd-q configurations would require open-shell treatment and are out of scope. In a periodic cell the added electrons are compensated by a uniform neutralizing background (the standard convention for charged supercells), so the excess charge is delocalized across the cell rather than localized as a solvated electron with an explicit counter-cation. The con...
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.