Wide and Cross GNNs: Cross‐Interactions and Parallel Scaling for Robust Chemical Property Prediction

ABSTRACT Graph neural networks (GNNs) have emerged as a powerful tool for molecular property prediction by representing chemical structures as graphs, yet they frequently struggle with oversmoothing and the inefficient capture of high‐order feature interactions. This paper proposes the wide and cross graph isomorphism network (WCGIN), an architecture designed to address these limitations through the integration of explicit cross‐feature interaction layers and a soft mixture‐of‐experts (MoE) parallel scaling paradigm. By incorporating matrix‐based cross‐layers into the graph isomorphism network (GIN) aggregation process, the model explicitly forms nonlinear feature relationships that might otherwise be lost during neighborhood processing. Simultaneously, the parallel MoE structure facilitates scaling by allowing multiple expert GNN modules to learn distinct representations, which are then aggregated via a gating network to mitigate the representational collapse typical of deep sequential stacking. The performance of WCGIN was evaluated across seven molecular benchmarks covering physical chemistry and biology, including four regression and three classification tasks. Experimental results indicate that adding even a single cross‐layer boosts performance in most cases, and the parallel scaling approach significantly outperforms standard stacked configurations for models with fewer than 10 layers. Furthermore, the study demonstrates that specialized mechanisms like load‐balancing loss are not required for these compact models to achieve robust predictive gains. These findings suggest that combining explicit feature interactions with parallel processing offers an efficient and effective pathway for scaling GNNs in chemical property prediction.

Authors

Institutions

Publication Details

Journal
Journal of Chemometrics
Published
2026-05-19
DOI
https://doi.org/10.1002/cem.70136
Primary Topic
Machine Learning in Materials Science
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Wide and Cross GNNs: Cross‐Interactions and Parallel Scaling for Robust Chemical Property Prediction

Kamil Paduszyński, Adrian Racki
Journal of Chemometrics
Machine Learning in Materials Science
article

Wide and Cross GNNs: Cross‐Interactions and Parallel Scaling for Robust Chemical Property Prediction

Kamil Paduszyński, Adrian Racki
article en

Abstract

ABSTRACT Graph neural networks (GNNs) have emerged as a powerful tool for molecular property prediction by representing chemical structures as graphs, yet they frequently struggle with oversmoothing and the inefficient capture of high‐order feature interactions. This paper proposes the wide and cross graph isomorphism network (WCGIN), an architecture designed to address these limitations through the integration of explicit cross‐feature interaction layers and a soft mixture‐of‐experts (MoE) parallel scaling paradigm. By incorporating matrix‐based cross‐layers into the graph isomorphism network (GIN) aggregation process, the model explicitly forms nonlinear feature relationships that might otherwise be lost during neighborhood processing. Simultaneously, the parallel MoE structure facilitates scaling by allowing multiple expert GNN modules to learn distinct representations, which are then aggregated via a gating network to mitigate the representational collapse typical of deep sequential stacking. The performance of WCGIN was evaluated across seven molecular benchmarks covering physical chemistry and biology, including four regression and three classification tasks. Experimental results indicate that adding even a single cross‐layer boosts performance in most cases, and the parallel scaling approach significantly outperforms standard stacked configurations for models with fewer than 10 layers. Furthermore, the study demonstrates that specialized mechanisms like load‐balancing loss are not required for these compact models to achieve robust predictive gains. These findings suggest that combining explicit feature interactions with parallel processing offers an efficient and effective pathway for scaling GNNs in chemical property prediction.

Journal of ChemometricsVol. 40(6)
Warsaw University of Technology (PL)
Openalex Percentile: Top 10%
Machine Learning in Materials Science
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.