pith. sign in

arxiv: 2606.19133 · v1 · pith:QC3QMKTRnew · submitted 2026-06-17 · ⚛️ physics.optics · cond-mat.mtrl-sci· cs.AI

Equivariant Graph Neural Networks Improve Optical Spectra Prediction for Materials Screening

Pith reviewed 2026-06-26 19:58 UTC · model grok-4.3

classification ⚛️ physics.optics cond-mat.mtrl-scics.AI
keywords equivariant graph neural networksoptical spectra predictionmaterials screeningrandom phase approximationstatic permittivityhigh-throughput screeningGotenNetthin-film optics
0
0 comments X

The pith

Equivariant graph neural networks outperform prior models when predicting optical spectra of materials.

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The paper establishes that adapting an equivariant graph neural network improves accuracy over existing surrogate models for optical spectra prediction. These predictions support high-throughput screening of materials for optoelectronic uses such as solar cells, where full first-principles calculations remain too expensive. Gains appear largest in the 0-8 eV range and for the static real permittivity, quantities directly relevant to thin-film optics. The model processes atomic structures while preserving rotational symmetries, unlike earlier rotation-invariant scalar-feature approaches. Evaluation covers multiple datasets, one containing 10,533 structures with spectra computed at the random phase approximation level.

Core claim

The paper claims that an equivariant graph neural network adapted from GotenNet outperforms the current state of the art on optical spectra prediction tasks. Performance advantages are reported on several datasets, including a collection of 10,533 structures with random phase approximation spectra, and are largest in the 0-8 eV interval and for the static real permittivity.

What carries the argument

Equivariant graph neural network (adapted GotenNet) that encodes material structures while respecting rotational symmetries to predict full optical spectra.

If this is right

  • More reliable high-throughput screening becomes feasible for optoelectronic materials such as solar-cell candidates.
  • Predictions improve most where they matter for thin-film device design, namely low-energy spectra and static permittivity.
  • Rotationally aware models can replace scalar-feature surrogates without loss of scalability.
  • Screening pipelines can incorporate these predictions directly into property filters for experimental follow-up.

Where Pith is reading between the lines

These are editorial extensions of the paper, not claims the author makes directly.

  • The same architecture may transfer to other geometry-dependent tensor properties such as elastic or piezoelectric responses.
  • Combining the model with active learning on experimental spectra could close the gap between theory and measurement.
  • Equivariant training might reduce data requirements when generalizing across crystal families.
  • The approach suggests a route to parameter-free surrogate models for spectra that respect physical symmetries by construction.

Load-bearing premise

Performance differences arise mainly from the equivariant architecture rather than from dataset tuning, training details, or target choices.

What would settle it

An ablation that removes equivariance while keeping all other model and training elements fixed and shows no accuracy change on the same RPA dataset.

Figures

Figures reproduced from arXiv: 2606.19133 by Fran\c{c}ois R J Cornet, Kasper Helverskov Petersen, Kristian S. Thygesen, Martin Ovesen, Mikkel Jordahn, Mikkel N. Schmidt.

Figure 1
Figure 1. Figure 1: Architecture of GotenNetOpt. (a) The model follows the MPNN framework: an embedding layer initializes atomic and bond features from elemental information and 3D coordinates, T message-passing rounds update the representations, and a readout layer predicts the optical spectra. (b) The message block. (c) Scalar edge feature update. (d) Node feature update. The summation P l P m is over l = 1, . . . , Lmax an… view at source ↗
Figure 2
Figure 2. Figure 2: Density histograms of SC (top), MSE (middle), and MAE (bottom) across the RPA test samples for the two models. MSE and MAE are shown on logarithmic x-axes to accommodate their heavy-tailed distributions. The difference in distributions is most pronounced for MAE and MSE, particularly the latter, as evidenced by the per￾formance summary in [PITH_FULL_IMAGE:figures/full_fig_p006_2.png] view at source ↗
Figure 3
Figure 3. Figure 3: Predicted (dashed, red) vs. target (solid, blue) dielectric spectra at the 10th, 40th, 60th, and 90th percentiles of the RPA test set ranked by SC. Each row shows Im(¯εRPA) and Re(¯εRPA) components for a single structure (chemical formula shown above) [PITH_FULL_IMAGE:figures/full_fig_p007_3.png] view at source ↗
Figure 4
Figure 4. Figure 4: confirms this explicitly, having a strong-per sample correlation between SC on Re(¯ε) and Im(¯ε). −0.2 0.0 0.2 0.4 0.6 0.8 1.0 SC Im(¯εRPA) −0.2 0.0 0.2 0.4 0.6 0.8 1.0 SC Re(¯εRPA) Test sample Kr y = x [PITH_FULL_IMAGE:figures/full_fig_p007_4.png] view at source ↗
Figure 5
Figure 5. Figure 5: Median SC for each compound in the IPA test set that contains a given element. Elements that are not present in any compounds are colored white. 101 102 103 Training set count (# compounds containing element) 3 × 10−1 4 × 10−1 6 × 10−1 Median SC – IPA Cr Mn Kr Mo Ru Te Re Element [PITH_FULL_IMAGE:figures/full_fig_p008_5.png] view at source ↗
Figure 6
Figure 6. Figure 6: Log-log plot of median SC for all compounds in the test set that contain a given element as a function of how often that element occurs in the training set for the IPA data. 5. Discussion This work suggests that equivariant GNNs can meaning￾fully improve optical spectra prediction, with the most pro￾nounced gains in the 0–8 eV range and for prediction of the static real permittivity. Gains are also proport… view at source ↗
Figure 7
Figure 7. Figure 7: Predicted of GotenNetOpt (dashed, red) and OptiMate3B (dashed, green) vs. target (solid, blue) dielectric spectra for the top 5 wins of GotenNetOpt over OptiMate3B as measured by ∆SC between the two models on the RPA test set. 16 [PITH_FULL_IMAGE:figures/full_fig_p016_7.png] view at source ↗
Figure 8
Figure 8. Figure 8: Predicted spectra of GotenNetOpt (dashed, red) and OptiMate3B (dashed, green) vs. target (solid, blue) dielectric spectra for the top 5 wins of OptiMate3B over GotenNetOpt as measured by ∆SC on the RPA test set. 17 [PITH_FULL_IMAGE:figures/full_fig_p017_8.png] view at source ↗
Figure 9
Figure 9. Figure 9: Predicted spectra of GotenNetOpt (dashed, red) and OptiMate3B (dashed, green) vs. target (solid, blue) for the 5 samples with the lowest average SC across both models (SC < 0.80) on the RPA test set, illustrating failure cases shared by both models. 18 [PITH_FULL_IMAGE:figures/full_fig_p018_9.png] view at source ↗
Figure 10
Figure 10. Figure 10: Predicted spectra of GotenNetOpt (dashed, red) and OptiMate3B (dashed, green) vs. target (solid, blue) for the 5 samples with the highest average SC across both models (SC > 0.95) on the RPA test set, illustrating representative cases where both models achieve high fidelity. 19 [PITH_FULL_IMAGE:figures/full_fig_p019_10.png] view at source ↗
read the original abstract

Scalable prediction of optical spectra is a critical component of high-throughput materials screening for optoelectronic applications such as solar cells. Existing surrogate models are trained on spectra computed from lower levels of theory or rely on rotation-invariant scalar features, limiting their geometric expressiveness. We explore the use of equivariant graph neural networks for optical spectra prediction, adapting GotenNet to this task and evaluating it on multiple datasets including a recently published collection of 10,533 structures with spectra computed at the level of the random phase approximation (RPA). The proposed model outperforms the current state of the art, with the largest gains in the 0-8 eV range and on predicting the static real permittivity, both of particular relevance for thin-film optics.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit. Tearing a paper down is the easy half of reading it; the pith above is the substance, this is the friction.

Referee Report

0 major / 1 minor

Summary. The manuscript adapts the equivariant graph neural network GotenNet for predicting optical spectra of materials computed at the random phase approximation (RPA) level. It evaluates the model on multiple datasets, including a collection of 10,533 structures, and reports that the proposed model outperforms prior state-of-the-art approaches, with the largest gains in the 0-8 eV range and for the static real permittivity.

Significance. If the reported performance gains hold under the provided dataset splits, training protocols, and baseline comparisons, the work would be significant for high-throughput materials screening in optoelectronics. The manuscript supplies the necessary dataset descriptions, model architecture details, training protocol, and quantitative comparisons, addressing the initial concern that the abstract alone was insufficient to assess the central claim.

minor comments (1)
  1. [Abstract] Abstract: the claim of outperformance would be more immediately verifiable if the abstract briefly named the primary baselines and reported a key metric (e.g., MAE on the 0-8 eV window) rather than leaving all quantitative detail to the main text.

Simulated Author's Rebuttal

0 responses · 0 unresolved

We thank the referee for the positive assessment of our manuscript and the recommendation for minor revision. The report raises no major comments requiring point-by-point rebuttal.

Circularity Check

0 steps flagged

No significant circularity

full rationale

The paper is a standard supervised ML benchmarking study that trains an adapted equivariant GNN on external RPA-computed spectra datasets and reports test-set performance gains versus published baselines. No derivation chain exists; claims rest on empirical comparisons with independent data splits and external references rather than any self-definitional mapping, fitted parameter renamed as prediction, or load-bearing self-citation. The architecture, training protocol, and metrics are fully specified without reducing to their own inputs by construction.

Axiom & Free-Parameter Ledger

0 free parameters · 0 axioms · 0 invented entities

Abstract-only review provides no information on free parameters, axioms, or invented entities; full manuscript required.

pith-pipeline@v0.9.1-grok · 5684 in / 976 out tokens · 14474 ms · 2026-06-26T19:58:54.805385+00:00 · methodology

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Reference graph

Works this paper leans on

32 extracted references · 6 canonical work pages

  1. [1]

    Applied Physics Letters , volume =

    Grunert, Malte and Großmann, Max and Runge, Erich , title =. Applied Physics Letters , volume =. 2026 , month =. doi:10.1063/5.0305586 , url =

  2. [2]

    ChemRxiv , volume =

    Anthony Wang and Steven Kauwe and Ryan Murdock and Taylor Sparks , title =. ChemRxiv , volume =. 2021 , doi =

  3. [3]

    The Thirteenth International Conference on Learning Representations , year =

    GotenNet: Rethinking Efficient 3D Equivariant Graph Neural Networks , author =. The Thirteenth International Conference on Learning Representations , year =

  4. [4]

    2024 , eprint =

    Ensemble-Embedding Graph Neural Network for Direct Prediction of Optical Spectra from Crystal Structure , author =. 2024 , eprint =

  5. [5]

    2025 , eprint =

    Accelerated prediction of dielectric functions in solar cell materials with graph neural networks , author =. 2025 , eprint =

  6. [6]

    2024 , eprint =

    Prediction of Frequency-Dependent Optical Spectrum for Solid Materials: A Multi-Output & Multi-Fidelity Machine Learning Approach , author =. 2024 , eprint =

  7. [7]

    Computational Materials Science , volume =

    Python Materials Genomics (pymatgen): A Robust, Open-Source Python Library for Materials Analysis , author =. Computational Materials Science , volume =. 2013 , doi =

  8. [8]

    Schoenholz and Patrick F

    Justin Gilmer and Samuel S. Schoenholz and Patrick F. Riley and Oriol Vinyals and George E. Dahl , title =. CoRR , volume =. 2017 , url =. 1704.01212 , timestamp =

  9. [9]

    Inhomogeneous Electron Gas , author =. Phys. Rev. , volume =. 1964 , month =. doi:10.1103/PhysRev.136.B864 , url =

  10. [10]

    Electronic excitations: density-functional versus many-body Green's-function approaches , author =. Rev. Mod. Phys. , volume =. 2002 , month =. doi:10.1103/RevModPhys.74.601 , url =

  11. [11]

    2019 , eprint =

    Molecular Property Prediction: A Multilevel Quantum Interactions Modeling Perspective , author =. 2019 , eprint =

  12. [12]

    2017 , eprint =

    SchNet: A continuous-filter convolutional neural network for modeling quantum interactions , author =. 2017 , eprint =

  13. [13]

    Margraf and Stephan G

    Johannes Klicpera and Shankari Giri and Johannes T. Margraf and Stephan G. Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Molecules , journal =. 2020 , url =. 2011.14115 , timestamp =

  14. [14]

    CoRR , volume =

    Yi Liu and Limei Wang and Meng Liu and Xuan Zhang and Bora Oztekin and Shuiwang Ji , title =. CoRR , volume =. 2021 , url =. 2102.05013 , timestamp =

  15. [15]

    CoRR , volume =

    Victor Garcia Satorras and Emiel Hoogeboom and Max Welling , title =. CoRR , volume =. 2021 , url =. 2102.09844 , timestamp =

  16. [16]

    Kristof T. Sch. Equivariant message passing for the prediction of tensorial properties and molecular spectra , journal =. 2021 , url =. 2102.03150 , timestamp =

  17. [17]

    2019 , eprint =

    Cormorant: Covariant Molecular Neural Networks , author =. 2019 , eprint =

  18. [18]

    Bekkers and Max Welling , title =

    Johannes Brandstetter and Rob Hesselink and Elise van der Pol and Erik J. Bekkers and Max Welling , title =. CoRR , volume =. 2021 , url =. 2110.02905 , timestamp =

  19. [19]

    2020 , eprint =

    Generalizing Convolutional Neural Networks for Equivariance to Lie Groups on Arbitrary Continuous Data , author =. 2020 , eprint =

  20. [20]

    2023 , eprint =

    MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fields , author =. 2023 , eprint =

  21. [21]

    2022 , eprint =

    Equivariant Graph Attention Networks for Molecular Property Prediction , author =. 2022 , eprint =

  22. [22]

    TorchMD-NET: Equivariant Transformers for Neural Network based Molecular Potentials , journal =

    Philipp Th. TorchMD-NET: Equivariant Transformers for Neural Network based Molecular Potentials , journal =. 2022 , url =. 2202.02541 , timestamp =

  23. [23]

    2023 , eprint =

    TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials , author =. 2023 , eprint =

  24. [24]

    2022 , eprint =

    Learning Local Equivariant Representations for Large-Scale Atomistic Dynamics , author =. 2022 , eprint =

  25. [25]

    2023 , eprint =

    Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs , author =. 2023 , eprint =

  26. [26]

    The Thirteenth International Conference on Learning Representations , journal =

    Liao, Yi-Lun and Wood, Brandon and Das, Abhishek and Smidt, Tess , title =. The Thirteenth International Conference on Learning Representations , journal =. 2024 , doi =

  27. [27]

    Physical review , volume =

    Self-consistent equations including exchange and correlation effects , author =. Physical review , volume =. 1965 , publisher =

  28. [28]

    Physical review , volume =

    Causality and the dispersion relation: logical foundations , author =. Physical review , volume =. 1956 , publisher =

  29. [29]

    Data for evaluating OptiMate3B

    Malte Grunert and Max Großmann and Erich Runge. Data for evaluating OptiMate3B. 2026. doi:10.6084/m9.figshare.30257551.v1

  30. [30]

    Dielectric Function of Semiconductors and Insulators under the Independent Particle Approximation and the Random Phase Approximation

    Malte Grunert and Max Großmann and Erich Runge. Dielectric Function of Semiconductors and Insulators under the Independent Particle Approximation and the Random Phase Approximation. 2026. doi:10.6084/m9.figshare.30257689.v1

  31. [31]

    2022 , eprint=

    High-throughput optical absorption spectra for inorganic semiconductors , author=. 2022 , eprint=

  32. [32]

    , title =

    Jain, Anubhav and Ong, Shyue Ping and Hautier, Geoffroy and Chen, Wei and Richards, William Davidson and Dacek, Stephen and Cholia, Shreyas and Gunter, Dan and Skinner, David and Ceder, Gerbrand and Persson, Kristin A. , title =. APL Materials , volume =. 2013 , month =. doi:10.1063/1.4812323 , url =