Wonder if that can be avoided by modifying the training approach. Ideas offhand: group by topic, train a subset of weights per node; figure out which layers have the most divergence and reduce lr on those only.

A provable way to recover convergence is to calculate the hessian. It’s computationally expensive but there are approximation methods.