Multi-dimensional domain generalization with low-rank structures