Why periodic dipoles need more than one branch
The central technical obstacle is not the potential energy surface but the dipole. In a periodic system, the modern theory of polarization defines the dipole only up to a quantum of polarization, e|a| along each lattice vector, so the dipole along a lattice direction can be written as a branch index times the lattice vector length. If a model is forced to return the same dipole value at periodically equivalent atomic positions, the learned function becomes discontinuous: it must jump between branches whenever an atom crosses a cell boundary.
That discontinuity matters because the model is autodifferentiable. Forces in an electric field depend on the derivative of the dipole with respect to nuclear positions, which is the Born effective charge tensor. A model trained to satisfy periodic single-valuedness will bend its slope near the boundaries to close the loop, so its Born effective charges, and therefore its field-dependent forces, deviate from the correct values once configurations move far from equilibrium. The authors note that adding more out-of-equilibrium data does not fix this, because the single-valued function cannot represent the underlying physics.
The fix is to let the dipole be multi-valued while keeping it smooth. The authors modify the MACE equivariant message-passing network so the predicted dipole equals a usual environment-dependent term plus a sum over atoms of an integer branch index times the atomic position. The integer indices are the atomic oxidation numbers, obtained from a DFT calculation that displaces an atom across a periodic boundary and measures the change in dipole. Because the extra term is linear in positions, its derivative contributes a constant to the Born effective charges, and the acoustic sum rule is preserved automatically.
Training on dipole alone, then differentiating for forces
The framework couples two neural networks: a MACE interatomic potential that supplies energies and ordinary Born-Oppenheimer forces, and a MACE dipole model that supplies the dipole and, through autodifferentiation, the Born effective charges needed for the field term. Training only on the dipole avoids the cost of computing Born effective charges directly at every first-principles step, and the authors argue it also makes data generation easier across electronic-structure methods.
The reported test errors set the accuracy scale. For liquid water the dipole error is 3.12 mD per atom, with energy and force errors of 1.10 meV per atom and 31.19 meV per angstrom; the asterisk in the paper marks the energy and force values as training-set errors, while the dipole value is a test-set error. For LiNbO3 the corresponding values are 11.38 mD per atom, 0.85 meV per atom, and 27.51 meV per angstrom, all on the test set. These are the numbers a reader should use to judge whether the model is accurate enough for a given observable, and they are not interchangeable across the two systems.
The LiNbO3 training set came from three ab initio NVT trajectories of 20 ps at 500, 1000, and 1500 K in a 30-atom cell, with 5000 structures selected and polarizations computed by the Berry phase formalism and then branch-matched. The water model used the MACE architecture with a 6 angstrom cutoff, two layers, 64 embedding channels, and a maximum spherical harmonic of order three. Molecular dynamics was driven through i-PI, which communicated with the models to obtain all force components.
Water's dielectric function spans twelve frequency decades
For liquid water at room temperature and standard density, the authors compute the frequency-dependent dielectric function over 0.001 to 135 THz, equivalent to 0.03 to 4500 inverse centimeters. This range is the practical payoff of nanosecond trajectories: reaching the low-frequency end requires sampling far longer than conventional ab initio molecular dynamics can afford. The paper compares the calculated real and imaginary permittivity against experimental data at 20 degrees Celsius, shown as reference points in the figure.
The same simulations produce infrared absorption spectra at 300 K under several applied field strengths. The relevant distinction for readers is that the dielectric function is an equilibrium response property computed from fluctuations, whereas the field-dependent spectra and polarization curves come from simulations with an explicit applied field. They answer different questions and should not be treated as the same measurement.
The electrofreezing transition is the second water result. The authors report that nuclear quantum effects enhance this phenomenon at larger field intensities. The enhancement is described as seemingly small in the polarization behavior itself, but the discussion argues that such changes can move a system to very different points in a phase transition, which is why the authors flag superionic water phases under fields and nanoconfinement as a natural next target.
LiNbO3 crosses to paraelectric but never fully switches
For the ferroelectric perovskite LiNbO3, the authors simulate the ferroelectric-to-paraelectric transition using a 4x4x4 supercell of 640 atoms, with ten independent NVT runs per temperature and a Langevin thermostat. The dipole at each step is evaluated with the trained dipole model. This is the equilibrium side of the LiNbO3 work: it establishes the transition behavior in a cell large enough to contain depolarization fields.
The non-equilibrium side applies a time-dependent monochromatic pulse at 18 THz, close to resonance with an 18.4 THz infrared-active mode, with a Gaussian envelope of 188 fs full width at half maximum and peak fields from 0.05 to 0.30 V per angstrom. Sixteen independent trajectories were run per pulse from geometries equilibrated at 300 K, in the NVE ensemble with a 0.1 fs time step. The proposed polarization-reversal mode lies at 7.4 THz, and the simulations show it accumulating energy with a delay of some hundreds of femtoseconds through non-linear coupling to the driven mode. Two additional A1-symmetry modes at 8.2 and 10.0 THz also become excited, and the 8.2 THz mode contributes to polarization reversal.
The polarization response splits into three field regimes. At 0.05 V per angstrom the polarization is only slightly perturbed and the material stays ferroelectric. At 0.15 and 0.20 V per angstrom the system is transiently driven to zero polarization during the pulse but returns to the original ferroelectric state with a relaxation time of about 300 fs, retaining coherent dynamics. At 0.25 and 0.30 V per angstrom the system transiently reaches higher reversed polarization but never fully switches; after the pulse it settles into the paraelectric state with incoherent dynamics because the absorbed energy heats it above the Curie temperature.
Full-dimensional driving explains a partial experimental switch
The authors state plainly that a full polarization switch is not achieved in their simulations. They read this as consistent with experiments that could not fully reverse polarization with this kind of phonon driving, and they attribute the incomplete reversal to depolarization fields, which the large supercells used here can capture. Earlier theoretical treatments of this effect relied on potential energy surfaces containing only two or a few phonon modes and lowest-order anharmonic couplings; the machine-learned model removes that restriction.
The paper's own interpretive proposal goes one step further. The authors suggest that coherent driving to the paraelectric state within the pulse duration, at sufficiently intense fields, is likely the explanation for the dip in the second-harmonic generation signal observed during field application in the referenced experiment. This is presented as a proposed explanation rather than a demonstrated causal link, and readers should treat it accordingly.
The methodological point is that the multi-valued dipole treatment is described as essential for this LiNbO3 example. Strongly non-equilibrium phonon driving is exactly the regime where a single-valued dipole model would produce wrong Born effective charges and forces, so the physical conclusion about partial switching depends on getting that term right.
What the framework covers and what it does not
The authors frame the method as generally applicable to molecules, solids, liquids, and disordered systems without system-specific assumptions, retaining quantum-mechanical accuracy for electronic interactions and allowing nuclear quantum effects to be included. The water simulations used a PIGLET thermostat with a 0.25 fs time step, which is how nuclear quantum effects enter the electrofreezing result. The claim of generality rests on the formulation rather than on an exhaustive survey: the demonstrations are liquid water and LiNbO3.
The electric-dipole approximation itself sets a boundary. The Hamiltonian used here adds a term minus the field dotted with the dipole, which the paper introduces for fields small compared with internal fields within the linear-response regime. Readers considering intense or strongly inhomogeneous fields should check whether that approximation still holds for their case.
The dipole model also depends on oxidation numbers extracted from DFT for each atomic species, and the LiNbO3 dipoles required branch-matching of the training data. These are per-system setup steps, not automatic outputs, so transferring the approach to a new material means repeating them. The reported errors are also system-specific: 3.12 mD per atom for water and 11.38 mD per atom for LiNbO3 should not be read as a general accuracy guarantee.
Field intensity, not pulse presence, decides the LiNbO3 outcome
The most decision-relevant comparison in the paper is the LiNbO3 field scan. The same 18 THz, 188 fs pulse produces three qualitatively different outcomes depending only on peak intensity: a slightly perturbed ferroelectric at 0.05 V per angstrom, a transient paraelectric excursion that recovers in about 300 fs at 0.15 to 0.20 V per angstrom, and a permanent transition to the paraelectric state above the Curie temperature at 0.25 to 0.30 V per angstrom, with no full reversal in any case.
That split is the practical takeaway for anyone reasoning about phonon-driven polarization control in this material. Reaching a reversed polarization is not the same as reaching a switched one, and the simulations indicate that pushing the field harder buys heating rather than complete switching. The authors connect the transient paraelectric window to the experimentally observed second-harmonic generation dip, which makes the intermediate intensity range the most interesting regime for coherent, recoverable control.
For water, the analogous priority is the low-frequency end of the dielectric response and the field dependence of electrofreezing, where nuclear quantum effects shift the outcome at larger field intensities. In both materials, the value of the framework is that it lets a reader resolve field- and frequency-dependent behavior that equilibrium or short-trajectory methods cannot reach, while the multi-valued dipole treatment is what keeps the forces trustworthy once the system is driven away from equilibrium.
Sources used in this answer
Electric-Field Driven Nuclear Dynamics of Liquids and Solids from a Multi-Valued Machine-Learned Dipolar Model
A multi-valued, autodifferentiable machine-learned dipole model combined with a MACE interatomic potential enables nanosecond, quantum-accurate electric-field-driven dynamics; in LiNbO3 no full polarization switch occurs, and field intensity alone separates recovery, transient paraelectric behavior, and heating above the Curie temperature.
