Group Averaging for Physics Applications: Accuracy Improvements at Zero Training Cost

TL;DR

Applying inexpensive group averaging at test time can significantly improve the accuracy of machine learning models in physics applications, especially for differential equations, without additional training or model modifications.

Contribution

Demonstrates that test-time group averaging enhances model accuracy in physics ML tasks, with practical benefits and no training overhead.

Findings

Up to 37% reduction in evaluation loss

Visually improved predictions of continuous dynamics

Always decreases average evaluation loss in experiments

Abstract

Many machine learning tasks in the natural sciences are precisely equivariant to particular symmetries. Nonetheless, equivariant methods are often not employed, perhaps because training is perceived to be challenging, or the symmetry is expected to be learned, or equivariant implementations are seen as hard to build. Group averaging is an available technique for these situations. It happens at test time; it can make any trained model precisely equivariant at a (often small) cost proportional to the size of the group; it places no requirements on model structure or training. It is known that, under mild conditions, the group-averaged model will have a provably better prediction accuracy than the original model. Here we show that an inexpensive group averaging can improve accuracy in practice. We take well-established benchmark machine learning models of differential equations in which certain symmetries ought to be obeyed. At evaluation time, we average the models over a small group of symmetries. Our experiments show that this procedure always decreases the average evaluation loss, with improvements of up to 37\% in terms of the VRMSE. The averaging produces visually better predictions for continuous dynamics. This short paper shows that, under certain common circumstances, there are no disadvantages to imposing exact symmetries; the ML4PS community should consider group averaging as a cheap and simple way to improve model accuracy.