fig8
Figure 8. The mean absolute error (MAE) of replacing MLPs in various components of the Allegro model with KANs on the HfO2 dataset. Replacing the MLP in the output block of Allegro significantly improves prediction accuracy for both energies and forces. MLPs: Multi-layer perceptrons; KANs: Kolmogorov-Arnold Networks.